expomap/conference/models.py

# -*- coding: utf-8 -*-
from django.db import models
from hvad.models import TranslatableModel, TranslatedFields, TranslationManager
import copy
#custom functions
from functions.custom_fields import EnumField



class ConferenceManager(TranslationManager):
    def safe_get(self, **kwargs):
        model = self.model
        try:
            return model.objects.get(**kwargs)
        except:
            return None



CURRENCY = ('RUB', 'USD', 'EUR')

class Conference(TranslatableModel):
    """
    Create Conference model

    Uses hvad.TranslatableModel which is child of django.db.models class
    """
    #set manager of this model
    objects = ConferenceManager()

    url = models.SlugField(unique=True)
    data_begin = models.DateField(verbose_name='Дата начала')
    data_end = models.DateField(verbose_name='Дата окончания')
    #relations
    country = models.ForeignKey('country.Country', verbose_name='Страна', on_delete=models.PROTECT)
    city = models.ForeignKey('city.City', verbose_name='Город', on_delete=models.PROTECT)
    place = models.ForeignKey('place_conference.PlaceConference', verbose_name='Место проведения',
                              blank=True, null=True, on_delete=models.PROTECT, related_name='conference_place')
    theme = models.ManyToManyField('theme.Theme', verbose_name='Тематики',
                              related_name='conference_themes')
    tag = models.ManyToManyField('theme.Tag', verbose_name='Теги',
                              blank=True, null=True, related_name='conference_tags')
    organiser = models.ManyToManyField('organiser.Organiser', verbose_name='Организатор',
                              blank=True, null=True, related_name='conference_organisers')
    company = models.ManyToManyField('company.Company', verbose_name='Компании',
                              blank=True, null=True, related_name='conference_compamies')
    users = models.ManyToManyField('accounts.User', verbose_name='Посетители выставки',
                              blank=True, null=True, related_name='conference_users')
    #!service has bitfield uncomment when country data will be filled
    #service = models.ManyToManyField('service.Service', verbose_name='Услуги', blank=True, null=True)
    web_page = models.CharField(verbose_name='Вебсайт', max_length=255, blank=True)
    link = models.CharField(verbose_name='Линк на регистрацию', max_length=255, blank=True)
    foundation_year = models.PositiveIntegerField(verbose_name='Год основания', blank=True, null=True)
    #
    currency = EnumField(values=CURRENCY, default='RUB')
    tax = models.BooleanField(verbose_name='Налог', default=1)
    min_price = models.PositiveIntegerField(verbose_name='Минимальная цена', blank=True, null=True)
    max_price = models.PositiveIntegerField(verbose_name='Максимальная цена', blank=True, null=True)
    #administrator can cancel conference
    canceled_by_administrator = models.BooleanField(default=0)
    #can publish not immediately
    is_published = models.BooleanField(default=0)
    #translated fields
    translations = TranslatedFields(
        name = models.CharField(verbose_name='Название', max_length=255),
        main_title=models.TextField(verbose_name='Краткое описание', blank=True),
        description=models.TextField(verbose_name='Описание', blank=True),
        main_themes=models.TextField(verbose_name='Основные темы', blank=True),
        time=models.TextField(verbose_name='Время работы', blank=True),
        discount=models.TextField(verbose_name='Условия и Скидки', blank=True),
        #-----meta data
        title=models.CharField(max_length=250),
        descriptions=models.CharField(max_length=250),
        keywords=models.CharField(max_length=250),
    )

    #fields saves information about creating and changing model
    created = models.DateTimeField(auto_now_add=True)
    modified = models.DateTimeField(auto_now=True)

    #audience = EnumField(values=AUDIENCE)
    #mark

    def __unicode__(self):
        return self.lazy_translation_getter('name', unicode(self.pk))

    def on(self):
        self.is_published = True
        self.canceled_by_administrator = False
        self.save()

    def off(self):
        self.is_published = False
        self.canceled_by_administrator = True
        self.save()


    def cancel(self):
        self.canceled_by_administrator = True

    def clone(self):
        """
        Return an identical copy of the instance with a new ID.
        """
        if not self.pk:
            raise ValueError('Instance must be saved before it can be cloned.')

        duplicate = copy.copy(self)
        # Setting pk to None. Django thinking this is a new object.
        duplicate.pk = None
        # url must be unique
        duplicate.url += '_copy'
        if Conference.objects.safe_get(url=duplicate.url):
            #already has copy this instance
            return

        # duplicate should not be published
        duplicate.is_published = False
        duplicate.cancel_by_administrator = False

        duplicate.save()
        # but lost all ManyToMany relations and Translations.

        # copy relations
        for field in self._meta.many_to_many:
            source = getattr(self, field.attname)
            destination = getattr(duplicate, field.attname)
            for item in source.all():
                destination.add(item)

        # copy translations
        languages = self.get_available_languages()
        ignore_fields = ['id', 'master', 'language_code']

        for code in languages:
            duplicate.translate(code)
            tr = self._meta.translations_model.objects.get(language_code = code,master__id=self.pk)
            for field in duplicate._translated_field_names:

                if field in ignore_fields:
                    continue

                setattr(duplicate, field, getattr(tr, field))

            duplicate.save()

        return duplicate


class TimeTable(TranslatableModel):
    """
    TimeTable for business program

    """
    exposition = models.ForeignKey(Conference, related_name='business_program')
    begin = models.DateTimeField(verbose_name='Начало')
    end = models.DateTimeField(verbose_name='Конец')
    #translated fields
    translations = TranslatedFields(
        name = models.CharField(verbose_name='Название', max_length=255)
    )

from django.db.models.signals import post_save
from django.dispatch import receiver
from django.conf import settings
from functions.form_check import translit_with_separator
from functions.translate import get_translated_fields

@receiver(post_save)
def conference_post_save_handler(sender, **kwargs):
    """
    objects saves two times:
        - first time Conference object
        - second time ConferenceTranslation object
    object must have at least one Translation
    """

    obj = kwargs['instance']
    if isinstance(obj, Conference):
    # object is Exposition
        if kwargs['created'] and obj.url == '':
        # if url doesn't set in new object set it
            obj.url = translit_with_separator(obj.name)

        # what languages must be
        all_langs = [code for code, lang in settings.LANGUAGES]
        # what languages are
        obj_langs = obj.get_available_languages()
        #
        missing_languages = list(set(all_langs) - set(obj_langs))
        # get first Translation object
        translation = obj.translations.all()[0]
        # translated fields
        fields = get_translated_fields(obj)

        for code in missing_languages:
            # translate
            obj.translate(code)
            # go through all fields and set value
            for field in fields:
                setattr(obj, field, getattr(translation, field))

            obj.save()