# -*- coding: utf-8 -*- from django.db import models from django.contrib.contenttypes import generic from django.db.models.signals import post_save, pre_save from django.utils.translation import ugettext as _ # from hvad.models import TranslatableModel, TranslatedFields, TranslationManager # from functions.custom_fields import EnumField, LocationField from functions.signal_handlers import post_save_handler, pre_save_handler from functions.models_methods import ExpoManager import copy from exposition.models import Exposition from place_conference.models import PlaceConference EXPOSITION_TYPE = (('Exposition complex', u'Выставочный комплекс'), ('Convention centre', u'Конгрессно-выставочный центр'), ('Exposition centre', u'Выставочный центр'),) class PlaceExposition(TranslatableModel): """ Create PlaceConference model Uses hvad.TranslatableModel which is child of django.db.models class """ #set manager of this model objects = ExpoManager() url = models.SlugField(unique=True, max_length=255) country = models.ForeignKey('country.Country', on_delete=models.PROTECT) city = models.ForeignKey('city.City', on_delete=models.PROTECT) #type uses EnumField for creating Enum type field in Mysql database type = EnumField(values = [item1 for item1, item2 in EXPOSITION_TYPE]) #information address = LocationField(verbose_name='Адресс') phone = models.BigIntegerField(blank=True, null=True) fax = models.BigIntegerField(blank=True, null=True) web_page = models.URLField(blank=True) email = models.EmailField(blank=True) # foundation_year = models.PositiveIntegerField(blank=True, null=True) event_in_year = models.PositiveIntegerField(blank=True, null=True) total_area = models.PositiveIntegerField(blank=True, null=True) closed_area = models.PositiveIntegerField(blank=True, null=True) open_area = models.PositiveIntegerField(blank=True, null=True) total_pavilions = models.PositiveIntegerField(blank=True, null=True) total_halls = models.PositiveIntegerField(blank=True, null=True) video_link = models.CharField(max_length=255, blank=True) virtual_tour = models.URLField(blank=True) # wifi = models.NullBooleanField() bank = models.NullBooleanField() children_room = models.NullBooleanField() disabled_service = models.NullBooleanField() conference_centre = models.NullBooleanField() #change name here, in form, in template business_centre = models.NullBooleanField() online_registration = models.NullBooleanField() cafe = models.NullBooleanField() terminals = models.NullBooleanField() parking = models.NullBooleanField() press_centre = models.NullBooleanField() mobile_application = models.NullBooleanField() # files = generic.GenericRelation('file.FileModel', content_type_field='content_type', object_id_field='object_id') photos = generic.GenericRelation('file.Photo', content_type_field='content_type', object_id_field='object_id') #translations is translated fields translations = TranslatedFields( name = models.CharField(max_length=100), main_title = models.TextField(blank=True), description = models.TextField(blank=True), adress = models.TextField(blank=True), #-----meta data title = models.CharField(max_length=250), descriptions = models.CharField(max_length=250), keywords = models.CharField(max_length=250), # total_year_action = models.TextField(blank=True), ) #fields saves information about creating and changing model created = models.DateTimeField(auto_now_add=True) modified = models.DateTimeField(auto_now=True) views = models.PositiveIntegerField(default=0) is_published = models.BooleanField(default=1) class Meta: ordering = ['translations__name'] def get_index_text(self): names = [tr.name for tr in self.translations.all()] return names def __unicode__(self): return self.lazy_translation_getter('name', unicode(self.pk)) def get_object_type(self): return _(u'Места') def get_catalog_url(self): return '/places/' def get_permanent_url(self): url = '%splace-%s'%(self.get_catalog_url(), self.url) return url def get_logo(self): logo = self.files.filter(purpose='logo') if logo: return logo[0] return logo def get_photos(self): photos = self.photos.all() return photos def get_scheme(self): scheme = self.files.filter(purpose='scheme teritory') return scheme def events(self): events = Exposition.objects.filter(place=self)[:6] return events def get_nearest_places(self): pl_exp = PlaceExposition.objects.language().filter(city=self.city).exclude(id=self.id)[:3] pl_conf = PlaceConference.objects.language().filter(city=self.city)[:3] #return pl_exp[] return list(pl_exp)+ list(pl_conf)#PlaceExposition.objects.filter(city=self.city).exclude(id=self.id) def get_type(self): type = {'Convention centre': _(u'Конгессо-выставочный центр'), 'Exposition centre': _(u'Выставочный центр'), 'Exposition complex': _(u'Выставочный комплекс')} return type.get(self.type) 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 PlaceExposition.objects.safe_get(url=duplicate.url): #already has copy this instance return 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 Halls. # copy halls halls = Hall.objects.filter(place_exposition=getattr(self, 'id')) for hall in halls: duplicate_hall = copy.copy(hall) duplicate_hall.exposition = duplicate duplicate_hall.save() return duplicate def get_events_number(self): exp = Exposition.objects.filter(place=self) return len(exp) class Hall(models.Model): """ Create Hall model which saves information about halls in PlaceExposition """ place_exposition = models.ForeignKey(PlaceExposition, related_name='halls') name = models.CharField(max_length=100, blank=True) number = models.PositiveIntegerField(blank=True, null=True) capacity = models.PositiveIntegerField(blank=True, null=True) #test pre_save.connect(pre_save_handler, sender=PlaceExposition) post_save.connect(post_save_handler, sender=PlaceExposition)