import autoslug
from django.db import models
from django.contrib.auth.models import User


class Company(models.Model):
    user = models.ForeignKey(User, related_name='company')
    name = models.CharField(max_length=300, verbose_name='Имя компании')
    inn = models.CharField(max_length=14, unique=True, verbose_name='ИНН')
    address = models.TextField(verbose_name='Адрес')
    phone = models.CharField(max_length=12, verbose_name='Телефон')
    mail = models.CharField(max_length=300, verbose_name='Почта')
    logo = models.ImageField(upload_to='general/static/logos/')
    balance = models.BigIntegerField(default=0, verbose_name='Баланс')
    date = models.DateTimeField()
    slug = autoslug.fields.AutoSlugField(populate_from='name', db_index=True,
                                         always_update=True, sep='-', unique=True, unique_with='date')
    active = models.IntegerField(default=0, verbose_name='Активность')

    def __str__(self):
        return self.name

    class Meta:
        verbose_name = 'Компания'
        verbose_name_plural = 'Компании'


class Category(models.Model):
    name = models.CharField(max_length=2000, verbose_name='Название категории')
    slug = autoslug.AutoSlugField(populate_from='name', always_update=True, db_index=True, sep='-')
    image = models.ImageField(upload_to='static/category/', verbose_name='Фото')

    def __str__(self):
        return self.name

    class Meta:
        verbose_name = 'Категория'
        verbose_name_plural = 'Категории'


class Product(models.Model):
    name = models.CharField(max_length=2000, verbose_name='Название продукта')
    number = models.CharField(max_length=500, verbose_name='Каталожный номер')
    slug = autoslug.fields.AutoSlugField(populate_from='name', always_update=True, db_index=True, sep='-')
    image = models.ImageField(upload_to='static/products/', verbose_name='Фото')
    category = models.ForeignKey(Category)

    def __str__(self):
        return self.name

    class Meta:
        verbose_name = 'Товар'
        verbose_name_plural = 'Товары'


class Offer(models.Model):
    product = models.ForeignKey(Product, verbose_name='Продукт')
    company = models.ForeignKey(Company, verbose_name='Компания')
    price = models.IntegerField(verbose_name='Цена')
    active = models.IntegerField(default=0, verbose_name='Активность')

    def __str__(self):
        return self.product

    class Meta():
        verbose_name = 'Предложение'
        verbose_name_plural = 'Предложения'


class Attribute(models.Model):
    key = models.CharField(max_length=200, verbose_name='Имя атрибута')
    value = models.CharField(max_length=200, verbose_name='Значение атрибута')
    product = models.ForeignKey(Product, verbose_name='Товар')

    def __str__(self):
        return self.product

    class Meta:
        verbose_name = 'Атрибут'
        verbose_name_plural = 'Атрибуты'