# -*- coding: utf-8 -*- from django.db import models from django.db.models.signals import post_save, pre_save from django.dispatch import receiver from hvad.models import TranslatableModel, TranslatedFields from bitfield import BitField # my models from directories.models import Iata from service.models import Service from settings.models import Setting # custom dunctions from functions.db import db_table_exists from functions.form_check import translit_with_separator from functions.signal_additional_func import fill_missing_languages, fill_meta_information #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 [] class City(TranslatableModel): """ Create City model Uses hvad.TranslatableModel which is child of django.db.models class """ services = BitField(flags=flags) url = models.SlugField(unique=True) #relations country = models.ForeignKey('country.Country', null=True, on_delete=models.PROTECT, related_name='cities') code_IATA = models.ForeignKey(Iata, blank=True, null=True) population = models.PositiveIntegerField(blank=True, null=True) phone_code = models.PositiveIntegerField(blank=True, null=True) #translated fields translations = TranslatedFields( name = models.CharField(max_length=50), transport = models.TextField(blank=True), description = models.TextField(blank=True), famous_places = models.TextField(blank=True), shoping = models.TextField(blank=True), #-----meta title = models.CharField(max_length=255), descriptions = models.CharField(max_length=255), keywords = models.CharField(max_length=255), ) #fields saves information about creating and changing model created = models.DateTimeField(auto_now_add=True) modified = models.DateTimeField(auto_now=True) def __unicode__(self): return self.lazy_translation_getter('name', self.pk) @receiver(post_save) def city_post_save_handler(sender, **kwargs): """ objects saves two times: - first time Country object - second time CountryTranslation object object must have at least one Translation """ obj = kwargs['instance'] if isinstance(obj, City): fill_missing_languages(obj) fill_meta_information(obj) """ if isinstance(obj, CityTranslation): # object is Translation - set url if obj.language_code == 'ru': city = City.objects.get(id=obj.master_id) country = city.country city.url = '%s-%s'%(translit_with_separator(country.name), translit_with_separator(obj.name)) city.save() """