# -*- coding: utf-8 -*- from django.db import models from hvad.models import TranslatableModel, TranslatedFields, TranslationManager import copy # from functions.custom_fields import EnumField AUDIENCE1 = ((None,'Не выбрано'), ('experts', 'Специалисты'), ('experts and consumers', 'Специалисты и потребители'), ('general public', 'Широкая публика')) AUDIENCE = (None,'experts', 'experts and consumers', 'general public') CURRENCY = ('RUB', 'USD', 'EUR') class ExpositionManager(TranslationManager): def safe_get(self, **kwargs): model = self.model try: return model.objects.get(**kwargs) except: return None class Exposition(TranslatableModel): """ Create Exposition model Uses hvad.TranslatableModel which is child of django.db.models class """ #set manager of this model objects = ExpositionManager() 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_exposition.PlaceExposition', verbose_name='Место проведения', blank=True, null=True, on_delete=models.PROTECT, related_name='exposition_place') theme = models.ManyToManyField('theme.Theme', verbose_name='Тематики', related_name='exposition_themes') tag = models.ManyToManyField('theme.Tag', verbose_name='Теги', blank=True, null=True, related_name='exposition_tags') organiser = models.ManyToManyField('organiser.Organiser', verbose_name='Организатор', blank=True, null=True, related_name='exposition_organisers') company = models.ManyToManyField('company.Company', verbose_name='Компании', blank=True, null=True, related_name='exposition_compamies') users = models.ManyToManyField('accounts.User', verbose_name='Посетители выставки', blank=True, null=True, related_name='exposition_users') #service = models.ManyToManyField('service.Service', verbose_name='Услуги', # blank=True, null=True, related_name='exposition_services') periodic = models.FloatField(verbose_name='Переодичность', blank=True, null=True) audience = EnumField(values=AUDIENCE, blank=True) web_page = models.CharField(verbose_name='Вебсайт', max_length=255, blank=True) foundation_year = models.PositiveIntegerField(verbose_name='Год основания', blank=True, null=True) min_area = models.PositiveIntegerField(verbose_name='Минимальная площадь', blank=True, null=True) # currency = EnumField(values=CURRENCY, default='RUB') tax = models.BooleanField(verbose_name='Налог', default=1) min_closed_area = models.PositiveIntegerField(verbose_name='Минимальная цена закрытой НЕ оборудованной площади', blank=True, null=True) max_closed_area = models.PositiveIntegerField(verbose_name='Максимальная цена закрытой НЕ оборудованной площади', blank=True, null=True) min_closed_equipped_area = models.PositiveIntegerField(verbose_name='Минимальная цена закрытой оборудованной площади', blank=True, null=True) max_closed_equipped_area = models.PositiveIntegerField(verbose_name='Максимальная цена закрытой оборудованной площади', blank=True, null=True) min_open_area = models.PositiveIntegerField(verbose_name='Минимальная цена открытой площади', blank=True, null=True) max_open_area = models.PositiveIntegerField(verbose_name='Максимальная цена открытой площади', blank=True, null=True) registration_payment = models.PositiveIntegerField(verbose_name='Регистрационный взнос', blank=True, null=True) #administrator can cancel exposition canceled_by_administrator = models.BooleanField(default=0) #can publish not immediately is_published = models.BooleanField(default=0) 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), products = models.TextField(verbose_name='Экспонируемые продукты', blank=True), time = 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), ) #field saves information about creating and changing model created = models.DateTimeField(auto_now_add=True) modified = models.DateTimeField(auto_now=True) #mark def __unicode__(self): return self.lazy_translation_getter('name', unicode(self.pk)) def cancel(self): self.canceled_by_administrator = True 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 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 Exposition.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(Exposition, 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 exposition_post_save_handler(sender, **kwargs): """ objects saves two times: - first time Exposition object - second time ExpositionTranslation object object must have at least one Translation """ obj = kwargs['instance'] if isinstance(obj, Exposition): # object is Exposition # 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] # 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() if isinstance(obj, ExpositionTranslation): # object is Translation print('trans') if obj.language_code == 'ru': exp = Exposition.objects.get(id=obj.master_id) exp.url = translit_with_separator(exp.name) exp.save()