# -*- 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 django.utils.translation import ugettext as _ # 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 [] flags = ['catalog', 'translator', 'participation', 'remote', 'tickets', 'visit', 'buildstand'] 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=_(u'Дата начала')) data_end = models.DateTimeField(verbose_name=_(u'Дата окончания')) #relations country = models.ForeignKey('country.Country', verbose_name=_(u'Страна'), on_delete=models.PROTECT) city = models.ForeignKey('city.City', verbose_name=_(u'Город'), on_delete=models.PROTECT) address = LocationField(verbose_name='Адрес', blank=True) theme = models.ManyToManyField('theme.Theme', verbose_name=_(u'Тематики'), related_name='seminar_themes') tag = models.ManyToManyField('theme.Tag', verbose_name=_(u'Теги'), blank=True, null=True, related_name='seminar_tags') organiser = models.ManyToManyField('organiser.Organiser', verbose_name=_(u'Организатор'), blank=True, null=True, related_name='seminar_organisers') company = models.ManyToManyField('company.Company', verbose_name=_(u'Компании'), blank=True, null=True, related_name='seminar_companies') users = models.ManyToManyField('accounts.User', verbose_name=_(u'Посетители выставки'), 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=_(u'Вебсайт'), max_length=255, blank=True) link = models.CharField(verbose_name=_(u'Линк на регистрацию'), max_length=255, blank=True) discount = models.PositiveIntegerField(verbose_name=_(u'Скидка'), blank=True, null=True) # currency = EnumField(values=CURRENCY, default='RUB') tax = models.BooleanField(verbose_name=_(u'Налог'), default=1) min_price = models.PositiveIntegerField(verbose_name=_(u'Минимальная цена'), blank=True, null=True) max_price = models.PositiveIntegerField(verbose_name=_(u'Максимальная цена'), 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=_(u'Год основания'), blank=True, null=True) visitors = models.PositiveIntegerField(verbose_name=_(u'Посетитеил'), blank=True, null=True) members = models.PositiveIntegerField(verbose_name=_(u'Участники'), blank=True, null=True) #translated fields translations = TranslatedFields( name = models.CharField(verbose_name=_(u'Название'), max_length=255), main_title = models.TextField(verbose_name=_(u'Краткое описание'), blank=True), programm = models.TextField(verbose_name=_(u'Описание'), blank=True), discount_description = models.TextField(verbose_name=_(u'Условия и Скидки'), 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=_(u'Год')) members = models.PositiveIntegerField(verbose_name=_(u'Посетители')) visitors = models.PositiveIntegerField(verbose_name=_(u'Участники')) # -*- 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)