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


class CompanyManager(TranslationManager):
    def safe_get(self, **kwargs):
        model = self.model
        try:
            return model.objects.get(**kwargs)
        except:
            return None


class Company(TranslatableModel):
    """
    Create Company model

    Uses hvad.TranslatableModel which is child of django.db.models class

    """
    objects = CompanyManager()

    url = models.SlugField()
    #relations
    creator = models.ForeignKey('accounts.User', verbose_name='Создатель', related_name='created_company')
    theme = models.ManyToManyField('theme.Theme', verbose_name='Отрасль',
                                   blank=True, null=True, related_name='companies')
    tag = models.ManyToManyField('theme.Tag', verbose_name='Теги', blank=True, null=True, related_name='companies')
    country = models.ForeignKey('country.Country', verbose_name='Страна', blank=True, null=True,
                                on_delete=models.PROTECT, related_name='companies')
    city = models.ForeignKey('city.City', verbose_name='Город', blank=True, null=True,
                                on_delete=models.PROTECT, related_name='companies')
    #address. uses LocationField. saves data in json format
    address = LocationField(verbose_name='Адрес', blank=True)

    staff_number = models.CharField(verbose_name='Количество сотрудников', max_length=50, blank=True)
    phone = models.FloatField(verbose_name='Телефон', blank=True, null=True)
    fax = models.FloatField(verbose_name='Факс', blank=True, null=True)
    web_page = models.CharField(verbose_name='Веб-сайт',max_length=255, blank=True)
    email = models.EmailField(verbose_name='Email', blank=True)
    social = models.TextField(verbose_name='Социальные страници', blank=True)
    foundation = models.PositiveIntegerField(verbose_name='Год основания', blank=True, null=True)
    #translation fields
    translation = TranslatedFields(
        name = models.CharField(verbose_name='Название компании', max_length=255),
        specialization  = models.CharField(verbose_name='Специализация', max_length=255, blank=True),
        description = models.TextField(verbose_name='О компании', blank=True),
        address_inf = models.TextField(verbose_name='Доп инф по адресу', 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)