from django.db import models


COOPERATION_MODE_CHOICES = [
    ('direct', '厂家直供'),
    ('authorized', '授权代理商'),
]


class Factory(models.Model):
    """
    工厂模型
    """
    dealer_name = models.CharField(max_length=255, blank=True, null=True, verbose_name='经销商名称')
    product_category = models.CharField(max_length=255, blank=True, null=True, verbose_name='产品分类')
    factory_name = models.CharField(max_length=255, verbose_name='生产工厂全称')
    short_name = models.CharField(max_length=100, unique=True, verbose_name='供应商简称')
    unified_social_credit_code = models.CharField(max_length=18, blank=True, null=True, verbose_name='统一社会信用代码')
    cooperation_mode = models.CharField(max_length=20, choices=COOPERATION_MODE_CHOICES, blank=True, null=True, verbose_name='合作模式')
    interaction_capability = models.TextField(blank=True, null=True, verbose_name='交互能力')
    province = models.CharField(max_length=50, verbose_name='省')
    city = models.CharField(max_length=50, verbose_name='市')
    district = models.CharField(max_length=50, blank=True, null=True, verbose_name='区')
    address = models.TextField(blank=True, null=True, verbose_name='详细地址')
    website = models.URLField(blank=True, null=True, verbose_name='官网链接')
    created_at = models.DateTimeField(auto_now_add=True, verbose_name='创建时间')
    updated_at = models.DateTimeField(auto_now=True, verbose_name='更新时间')

    class Meta:
        verbose_name = '工厂'
        verbose_name_plural = '工厂'
        db_table = 'factory'
        ordering = ['id']

    def __str__(self):
        return self.factory_name