expomap/service/models.py

# -*- coding: utf-8 -*-
from django.db import models
from hvad.models import TranslatableModel, TranslatedFields, TranslationManager
from functions.custom_fields import EnumField
from bitfield import BitField


CURENCIES = ('', 'USD', 'RUB', 'EUR')





class Service(TranslatableModel):

    url = models.SlugField(unique=True)
    currency = EnumField(values=CURENCIES, blank=True, default='USD')
    price = models.PositiveIntegerField(blank=True, null=True)
    params = models.CharField(max_length=255, blank=True)
    template = models.CharField(max_length=255, blank=True)
    type = BitField(flags=['expo', 'conference'])
    #translated fields
    translations = TranslatedFields(
        name = models.CharField(max_length=50),
        main_title = models.TextField(),
        description = models.TextField(),
        advantage = models.TextField(blank=True),
        #meta fields
        title = models.CharField(max_length=255, blank=True),
        descriptions = models.CharField(max_length=255, blank=True),
        keywords = models.CharField(max_length=255, blank=True),
    )
    sort = models.PositiveIntegerField(default=0, db_index=True)
    main_page = models.PositiveIntegerField(default=0, db_index=True)
    meta_id = models.IntegerField(null=True)


    def __unicode__(self):
        return self.lazy_translation_getter('name', self.pk)

    def get_form(self):
        pass

    def get_permanent_url(self):
        return '/service/%s/'%self.url

    def get_price(self):
        pr = self.price

    def get_current_state(self):
        """
        uses for control form
        :return:
        """
        from country.models import Country
        country_all = False
        country = []
        region = []
        expositions = []
        conferences = []
        service = self.url
        event_type = [key for key, value in self.type.iteritems() if value]
        if not event_type:
            return {'event_type': event_type,
                    'region': region,
                    'country': country,
                    'country_all': country_all,
                    'expositions': expositions,
                    'conferences': conferences}

        count1 = Country.objects.filter().count()
        count2 = Country.objects.filter(services=getattr(Country.services, service)).count()
        country_all = count1 == count2
        if not country_all:
            from exposition.models import Exposition
            from conference.models import Conference
            from country.models import Area
            from django.utils.translation import get_language
            lang = get_language()

            countries = list(Country.objects.language(lang).filter(services=getattr(Country.services, service)))

            expositions = [(item.id, item.name) for item in Exposition.enable.upcoming().exclude(country__in=countries, services=getattr(Exposition.services, service))]
            conferences = [(item.id, item.name) for item in Conference.enable.upcoming().exclude(country__in=countries, services=getattr(Conference.services, service))]

            region = []
            countries = set(countries)
            for item in list(Area.objects.language(lang).all()):
                print(item)
                area_countries = item.countries()
                if set(area_countries).issubset(countries):
                    region.append((item.id, item.name))
                    countries = countries - set(area_countries)

            country = [(item.id, item.name) for item in list(countries)]

        state = {'event_type': event_type,
                 'region': region,
                 'country': country,
                 'country_all': country_all,
                 'expositions': expositions,
                 'conferences': conferences}
        return state

from django.db.models.signals import post_save
from functions.signal_handlers import post_save_handler

post_save.connect(post_save_handler, sender=Service)

class AbstractOrder(models.Model):
    person = models.ForeignKey('accounts.User')
    phone = models.BigIntegerField()
    city = models.CharField(max_length=255, blank=True, default='')
    country = models.CharField(max_length=255, blank=True, default='')
    person_inf = models.CharField(max_length=255)
    price = models.IntegerField(blank=True, null=True)
    currency = EnumField(values=CURENCIES, blank=True)
    event = models.CharField(max_length=255, blank=True, null=True)
    exposition = models.ForeignKey('exposition.Exposition', null=True)
    conference = models.ForeignKey('conference.Conference', null=True)
    seminar = models.ForeignKey('seminar.Seminar', null=True)
    created = models.DateTimeField(auto_now_add=True)
    viewed = models.DateTimeField(null=True, blank=True)

    class Meta:
        abstract = True
        ordering = ['-created']



class Catalog(AbstractOrder):
    pass

class BuildStand(AbstractOrder):
    pass

class Tickets(AbstractOrder):
    days = models.SmallIntegerField()
    people = models.SmallIntegerField()

class Remote(AbstractOrder):
    interested_participants = models.TextField(blank=True)
    additional = models.CharField(max_length=255, blank=True)


class Participation(AbstractOrder):
    area = models.IntegerField(null=True)
    area_type = models.CharField(max_length=255)
    company_inf = models.TextField()

class Translation(AbstractOrder):
    days = models.IntegerField(null=True,blank=True)
    hours = models.IntegerField(null=True, blank=True)
    fr = models.DateField(blank=True)
    to = models.DateField(blank=True)
    languages = models.TextField(blank=True)
    themes = models.TextField(blank=True)


class Visit(AbstractOrder):
    fr = models.DateField()
    to = models.DateField()
    people = models.IntegerField(blank=True, null=True)
    hotel_additional = models.TextField(blank=True)
    city_from = models.CharField(max_length=255, blank=True)
    hotel_stars = models.CharField(max_length=30, blank=True)
    hotel_location = models.CharField(max_length=255, blank=True)
    hotel_budget = models.CharField(max_length=255, blank=True)
    hotel_budget_add = models.CharField(max_length=255, blank=True)

    avia_tickets = models.BooleanField()
    avia_type = models.CharField(max_length=255)
    visa = models.BooleanField()
    transfer = models.BooleanField()
    event_tickets = models.BooleanField()
    #
    translator = models.BooleanField()
    languages = models.CharField(max_length=255, blank=True)
    themes = models.CharField(max_length=255, blank=True)
    tr_employment_days = models.PositiveIntegerField(blank=True, null=True)
    tr_employment_hours = models.PositiveIntegerField(blank=True, null=True)

    excursion = models.BooleanField()
    notes = models.TextField(blank=True)

ADVERTYSE_TYPES = ((u'Рекламировать участника', u'Рекламировать участника'),
                  (u'Баннерная реклама', u'Баннерная реклама'),
                  (u'Приоритетное размещение событий в каталоге',u'Приоритетное размещение событий в каталоге'),
                  (u'Персональная визитка Вашей компании',u'Персональная визитка Вашей компании'))
class Advertising(AbstractOrder):
    type = models.CharField(max_length=255, blank=True, null=True)


class CallBack(models.Model):
    phone = models.CharField(max_length=30)
    person_inf = models.CharField(max_length=255)
    created = models.DateTimeField(auto_now_add=True)
    viewed = models.DateTimeField(null=True, blank=True)

    class Meta:
        ordering = ['-created']