# -*- coding: utf-8 -*- from django.db import models from hvad.models import TranslatableModel, TranslatedFields, TranslationManager from django.contrib.contenttypes import generic import copy from bitfield import BitField from service.models import Service from functions.db import db_table_exists # from functions.custom_fields import EnumField from functions.custom_fields import LocationField from functions.models_methods import ExpoManager from functions.model_mixin import EventMixin, ExpoMixin # check if table exist and create flags if true flags = [str(item.id) for item in Service.objects.all()] if db_table_exists('service_service') else [] CURRENCY = ('RUB', 'USD', 'EUR') class Seminar(TranslatableModel, EventMixin, ExpoMixin): """ Create Seminar model Uses hvad.TranslatableModel which is child of django.db.models class """ catalog = '/seminar/' # type of event event_type = 'seminar' #set manager of this model objects = ExpoManager() services = BitField(flags=flags) url = models.SlugField(unique=True) data_begin = models.DateTimeField(verbose_name='Дата начала') data_end = models.DateTimeField(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) address = LocationField(verbose_name='Адрес', blank=True) theme = models.ManyToManyField('theme.Theme', verbose_name='Тематики', related_name='seminar_themes') tag = models.ManyToManyField('theme.Tag', verbose_name='Теги', blank=True, null=True, related_name='seminar_tags') organiser = models.ManyToManyField('organiser.Organiser', verbose_name='Организатор', blank=True, null=True, related_name='seminar_organisers') company = models.ManyToManyField('company.Company', verbose_name='Компании', blank=True, null=True, related_name='seminar_companies') users = models.ManyToManyField('accounts.User', verbose_name='Посетители выставки', blank=True, null=True, related_name='seminar_users') #!service has bitfield uncomment when country data will be filled #service = models.ManyToManyField('service.Service', verbose_name='Услуги', blank=True, null=True) #periodic = models.FloatField(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) discount = 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) expohit = models.BooleanField(verbose_name='Expohit', default=0) #administrator can cancel seminar canceled_by_administrator = models.BooleanField(default=0) quality_label = BitField(flags=['ufi', 'rsva', 'exporating']) #can publish not immediately is_published = models.BooleanField(default=0) canceled = models.BooleanField(default=0) moved = models.BooleanField(default=0) files = generic.GenericRelation('file.FileModel', content_type_field='content_type', object_id_field='object_id') # statistic foundation_year = models.PositiveIntegerField(verbose_name='Год основания', blank=True, null=True) visitors = models.PositiveIntegerField(verbose_name='Посетитеил', blank=True, null=True) members = models.PositiveIntegerField(verbose_name='Участники', blank=True, null=True) #translated fields translations = TranslatedFields( name = models.CharField(verbose_name='Название', max_length=255), main_title = models.TextField(verbose_name='Краткое описание', blank=True), programm = models.TextField(verbose_name='Описание', blank=True), discount_description = 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) main_page = models.PositiveIntegerField(default=0, db_index=True) views = models.PositiveIntegerField(default=0) def __unicode__(self): return self.lazy_translation_getter('name', unicode(self.pk)) def get_nearest_events(self): seminars = Seminar.objects.all()[:5] return seminars def get_calendar_url(self): return '/seminar-add-calendar/%s/'%self.id def get_visit_url(self): return '/seminar-visit/%s/'%self.id def get_catalog_url(self): return '/seminars/' 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 Seminar.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 ignore_fields = ['id', 'master', 'language_code'] duplicate.translate('ru') tr = self._meta.translations_model.objects.get(language_code = 'ru',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() # 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) return duplicate class Statistic(models.Model): seminar = models.ForeignKey(Seminar, related_name='statistic') year = models.PositiveIntegerField(verbose_name='Год') members = models.PositiveIntegerField(verbose_name='Посетители') visitors = models.PositiveIntegerField(verbose_name='Участники') # -*- coding: utf-8 -*- from django.db.models.signals import post_save from functions.signal_handlers import post_save_handler post_save.connect(post_save_handler, sender=Seminar)