expomap/webinar/models.py

# -*- coding: utf-8 -*-
from django.db import models
from hvad.models import TranslatableModel, TranslatedFields, TranslationManager
import copy
from bitfield import BitField
from service.models import Service
from functions.db import db_table_exists

from django.contrib.contenttypes import generic
#
from functions.custom_fields import EnumField
from functions.models_methods import ExpoManager
from functions.model_mixin import EventMixin

# 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 Webinar(TranslatableModel, EventMixin):
    """
    Create Webinar model

    Uses hvad.TranslatableModel which is child of django.db.models class
    """
    #set manager of this model
    objects = ExpoManager()
    services = BitField(flags=flags)

    url = models.SlugField(unique=True)
    data_begin = models.DateTimeField(verbose_name='Дата начала', blank=True)
    #relations
    theme = models.ManyToManyField('theme.Theme', verbose_name='Тематики',
                                   related_name='webinar_themes')
    tag = models.ManyToManyField('theme.Tag', verbose_name='Теги',
                                   blank=True, null=True, related_name='webinar_tags')
    organiser = models.ManyToManyField('organiser.Organiser', verbose_name='Организатор',
                                   blank=True, null=True, related_name='webinar_organisers')
    company = models.ManyToManyField('company.Company', verbose_name='Компании',
                                   blank=True, null=True, related_name='webinar_companies')
    users = models.ManyToManyField('accounts.User', verbose_name='Посетители выставки',
                                   blank=True, null=True, related_name='webinar_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)
    quality_label = BitField(flags=['ufi', 'rsva', 'exporating'])
    #administrator can cancel webinar
    canceled_by_administrator = models.BooleanField(default=0)
    #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):
        webinars = Webinar.objects.all()[:5]
        return webinars

    def get_calendar_url(self):
        return '/webinar-add-calendar/%s/'%self.id

    def get_visit_url(self):
        return '/webinar-visit/%s/'%self.id

    def get_catalog_url(self):
        return '/webinars/'

    def duration_days(self):
        return 1


    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 Webinar.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):
    webinar = models.ForeignKey(Webinar, related_name='statistic')
    year = models.PositiveIntegerField(verbose_name='Год')
    members = models.PositiveIntegerField(verbose_name='Посетители')
    visitors = models.PositiveIntegerField(verbose_name='Участники')

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

post_save.connect(post_save_handler, sender=Webinar)