expomap/meta/models.py

# -*- coding: utf-8 -*-
from django.db import models
from django.utils import translation
from django.db.models.signals import post_save
from django.utils.translation import ugettext_lazy as _
from django.conf import settings
from django.core.urlresolvers import reverse_lazy
from django.core.cache import cache
from hvad.models import TranslatableModel, TranslatedFields, TranslationManager
from pymorphy.django_conf import default_morph as morph
from functions.signal_handlers import post_save_handler
import copy



# additional funcs
MONTHES = {'jan': _(u'январе'), 'feb': _(u'феврале'), 'mar': _(u'марте'), 'apr': _(u'апреле'),
               'may': _(u'мае'), 'jun': _(u'июне'), u'jul': _(u'июле'), 'aug': _(u'августе'),
               'sep': _(u'сентябре'), 'oct': _(u'октябре'), 'nov': _(u'ноябре'), 'dec': _(u'декабре'),}
def get_month_inflect(value, key):
    return {key: MONTHES.get(value, '') }

def get_obj_inflect(obj, key):
    if translation.get_language() == 'ru':
        result = {key: getattr(obj, 'inflect', '')}
    else:
        result = {key: getattr(obj, 'name', '')}
    return result

def get_theme_inflect(obj, key):
    if translation.get_language() == 'ru':
        result = {'theme_inflect': getattr(obj, 'inflect', ''),
                  'theme_name': getattr(obj, 'name', '')}
    else:
        result = {'theme_inflect': getattr(obj, 'name', ''),
                  'theme_name': getattr(obj, 'name', '')}
    return result

def get_tag_inflect(obj, key):
    if translation.get_language() == 'ru':
        result = {'tag_inflect': getattr(obj, 'inflect', ''),
                  'tag_name': getattr(obj, 'name', '')}
    else:
        result = {'tag_inflect': getattr(obj, 'name', ''),
                  'tag_name': getattr(obj, 'name', '')}
    return result


class MetaSetting(TranslatableModel):
    name = models.CharField(max_length=100, unique=True)
    translations = TranslatedFields(
        title = models.CharField(max_length=255, blank=True),
        description = models.CharField(max_length=255, blank=True),
        keywords = models.CharField(max_length=255, blank=True),
        h1 = models.CharField(max_length=255, blank=True),
    )

    object_params = {'object_name': 'name', 'object_title': 'main_title', 'city': 'city', 'country': 'country'}

    params = {'month': get_month_inflect, 'country': get_obj_inflect, 'city': get_obj_inflect,
              'theme': get_theme_inflect, 'tag': get_tag_inflect}
    monthes = {'jan': _(u'январе'), 'feb': _(u'феврале'), 'mar': _(u'марте'), 'apr': _(u'апреле'),
               'may': _(u'мае'), 'jun': _(u'июне'), u'jul': _(u'июле'), 'aug': _(u'августе'),
               'sep': _(u'сентябре'), 'oct': _(u'октябре'), 'nov': _(u'ноябре'), 'dec': _(u'декабре'),}


    def __unicode__(self):
        return self.name


    def generate_meta(self, params, obj=None):
        """
        obj must be in current language
        """

        lang = translation.get_language()
        updates = {}
        if obj:
            for key, value in self.object_params.iteritems():
                updates.update({key: getattr(obj, value, '')})

        for key, value in params.iteritems():
            if key in self.params:
                updates.update(self.params[key](value, key))
        params.update(updates)

        tr = self.translations.get(language_code=lang)
        title = tr.title.format(**params)
        description = tr.description.format(**params)
        keywords = []#tr.keywords.format(**params)
        h1 = tr.h1.format(**params)

        return {'title': title, 'description': description, 'keywords': keywords, 'h1': h1}

    def get_param(self, obj, field):
        param = self.params.get(field)
        if not param:
            return ''
        s = getattr(obj, param['name'], '')
        if param.get('inflected'):
            s = morph.inflect_ru(s, u'пр')
        return s


    def get_title(self):
        title = self.title
        return title

    def get_h1(self):
        h1 = self.h1
        return h1

    def get_description(self):
        description = self.description
        return description

    def get_keywords(self):
        keywords = self.keywords
        return []


post_save.connect(post_save_handler, sender=MetaSetting)


#           SEO - tests             #

class SeoTextManager(TranslationManager):
    cache_time = 120

    def cache_get(self, *args, **kwargs):
        url = kwargs.get('url')
        lang = kwargs.get('lang')[:2] or translation.get_language()[:2]
        key = 'seo_text_cache'
        result = cache.get(key)
        if result:
            return result.get("%s_%s" % (lang, url))

        qs = list(SeoText.objects.language('all'))
        value_dict = {obj.language_code+'_'+obj.url: obj for obj in qs}
        cache.set(key, value_dict, self.cache_time)
        return value_dict.get("%s_%s" % (lang, url))


class SeoText(TranslatableModel):
    url = models.CharField(max_length=50, unique=True, verbose_name=u"URL: www.expomap.ru")

    translations = TranslatedFields(
        title=models.CharField(max_length=255, verbose_name=u"Заголовок"),
        page_title=models.CharField(max_length=255, verbose_name=u"Тайтл страницы"),
        description=models.CharField(max_length=1000, verbose_name=u"Дескрипшн"),
        body=models.TextField(verbose_name=u"Текст")
    )

    objects = SeoTextManager()

    def __init__(self, *args, **kwargs):
        super(SeoText, self).__init__(*args, **kwargs)
        self.cache_fields = ['title', 'body', 'page_title', 'description']
        self.is_new = True

    def get_absolute_url(self):
        return self.url

    def save(self, *args, **kwargs):
        super(SeoText, self).save(*args, **kwargs)
        self.initial_language = 'ru'

        new_values = {field: getattr(self, field) for field in self.cache_fields}
        langs = [code for code, _ in settings.LANGUAGES]
        if self.is_new:
            for lang in langs:
                if lang not in self.get_available_languages():
                    self.translate(lang)
                    for key, value in new_values.items():
                        setattr(self, key, value)
                    self.save_translations(self)
        else:
            translations = {obj.language_code:obj for obj in list(self.translations.all())}
            for lang in langs:
                if lang is not self.initial_language:
                    tr = translations[lang]
                    for key, value in new_values.items():
                        #if u'%s' % getattr(self, key) is u'' or getattr(self, key) is u'%s' % self.var_cache[key]:
                        setattr(tr, key, value)
                    tr.save()

        self.lazy_translation_getter(self.initial_language)
        self.var_cache = {var: copy.copy(getattr(self, var)) for var in self.cache_fields}
        self.is_new = False
        return self

    def __unicode__(self):
        return self.url