TrainingPlatform_Django/datasets/models.py

from django.db import models
import uuid
class CreatorProfile(models.Model):
    """达人信息模型，用于筛选功能"""
    # 基本信息
    name = models.CharField(max_length=255, verbose_name="达人名称")
    # 修改为支持本地图片上传，同时保持URL兼容性
    avatar = models.ImageField(upload_to='avatars/', blank=True, null=True, verbose_name="头像图片")
    avatar_url = models.TextField(blank=True, null=True, verbose_name="头像URL")
    platform_id = models.CharField(max_length=255, blank=True, null=True, verbose_name="Platform ID")

    # 新增联系方式
    email = models.EmailField(max_length=255, blank=True, null=True, verbose_name="电子邮箱")
    instagram = models.CharField(max_length=255, blank=True, null=True, verbose_name="Instagram账号")
    location = models.CharField(max_length=100, blank=True, null=True, verbose_name="位置")
    live_schedule = models.CharField(max_length=255, blank=True, null=True, verbose_name="直播时间表")

    # 新增达人平台字段
    PROFILE_CHOICES = [
        ('tiktok', 'TikTok'),
        ('instagram', 'Instagram'),
        ('youtube', 'YouTube'),
        ('other', '其他平台'),
    ]
    profile = models.CharField(max_length=20, choices=PROFILE_CHOICES, default='tiktok', verbose_name="达人平台")

    # 新增hashtag和trend字段
    hashtags = models.TextField(blank=True, null=True, verbose_name="标签", help_text="以#分隔的标签，例如:#fashion#beauty#lifestyle")
    trends = models.TextField(blank=True, null=True, verbose_name="趋势", help_text="创作者相关的趋势关键词")

    # 新增地区字段
    region = models.CharField(max_length=100, blank=True, null=True, verbose_name="地区")

    # 新增主页链接字段
    tiktok_link = models.URLField(max_length=500, blank=True, null=True, verbose_name="主页链接")

    # 新增对标美区达人等级
    us_creator_level = models.CharField(max_length=50, blank=True, null=True, verbose_name="对标美区达人等级")

    # 新增报价字段
    price_gbp = models.DecimalField(max_digits=10, decimal_places=2, blank=True, null=True, verbose_name="报价(英镑)")
    price_usd = models.DecimalField(max_digits=10, decimal_places=2, blank=True, null=True, verbose_name="报价(美金)")

    # 新增GMV(英镑)字段
    gmv_gbp = models.DecimalField(max_digits=12, decimal_places=2, blank=True, null=True, verbose_name="GMV(英镑)")

    # 新增链接字段
    link = models.URLField(max_length=500, blank=True, null=True, verbose_name="链接")

    # 类别 - Category
    CATEGORY_CHOICES = [
        ('Phones & Electronics', '手机与电子产品'),
        ('Homes Supplies', '家居用品'),
        ('Kitchenware', '厨房用品'),
        ('Textiles & Soft Furnishings', '纺织品和软装'),
        ('Household Appliances', '家用电器'),
        ('Womenswear & Underwear', '女装和内衣'),
        ('Muslim Fashion', '穆斯林时尚'),
        ('Shoes', '鞋类'),
        ('Beauty & Personal Care', '美容和个人护理'),
        ('Computers & Office Equipment', '电脑和办公设备'),
        ('Pet Supplies', '宠物用品'),
        ('Baby & Maternity', '婴儿和孕妇用品'),
        ('Sports & Outdoor', '运动和户外'),
        ('Toys', '玩具'),
        ('Furniture', '家具'),
        ('Tools & Hardware', '工具和硬件'),
        ('Home Improvement', '家居装修'),
        ('Automotive & Motorcycle', '汽车和摩托车'),
        ('Fashion Accessories', '时尚配饰'),
        ('Food & Beverages', '食品和饮料'),
        ('Health', '健康'),
        ('Books, Magazines & Audio', '书籍、杂志和音频'),
        ('Kids Fashion', '儿童时尚'),
        ('Menswear & Underwear', '男装和内衣'),
        ('Luggage & Bags', '行李和包'),
        ('Pre-Owned Collections', '二手收藏'),
        ('Jewellery Accessories & Derivatives', '珠宝配饰及衍生品'),
    ]
    category = models.CharField(max_length=100, choices=CATEGORY_CHOICES, blank=True, null=True, verbose_name="类别")

    # 电商等级 - E-commerce Level (L1-L7)
    E_COMMERCE_LEVEL_CHOICES = [
        (1, 'L1'),
        (2, 'L2'),
        (3, 'L3'),
        (4, 'L4'),
        (5, 'L5'),
        (6, 'L6'),
        (7, 'L7'),
    ]
    e_commerce_level = models.IntegerField(choices=E_COMMERCE_LEVEL_CHOICES, blank=True, null=True,
                                           verbose_name="电商能力等级")

    # 曝光等级 - Exposure Level (KOC-1, KOC-2, KOL-1, KOL-2, KOL-3)
    EXPOSURE_LEVEL_CHOICES = [
        ('KOC-1', 'KOC-1'),
        ('KOC-2', 'KOC-2'),
        ('KOL-1', 'KOL-1'),
        ('KOL-2', 'KOL-2'),
        ('KOL-3', 'KOL-3'),
    ]
    exposure_level = models.CharField(max_length=10, choices=EXPOSURE_LEVEL_CHOICES, blank=True, null=True,
                                      verbose_name="曝光等级")

    # 粉丝数 - Followers
    followers = models.IntegerField(default=0, verbose_name="粉丝数")

    # GMV - Gross Merchandise Value (in thousands of dollars)
    gmv = models.DecimalField(max_digits=12, decimal_places=2, blank=True, null=True, verbose_name="GMV(千美元)")
    items_sold = models.DecimalField(max_digits=12, decimal_places=2, blank=True, null=True,
                                     verbose_name="售出商品数量")

    # 视频数据 - Video Views
    avg_video_views = models.IntegerField(default=0, blank=True, null=True, verbose_name="平均视频浏览量")

    # 价格信息 - Pricing
    pricing = models.DecimalField(max_digits=10, decimal_places=2, blank=True, null=True, verbose_name="个人定价")
    pricing_package = models.CharField(max_length=100, blank=True, null=True, verbose_name="套餐定价")

    # 合作信息 - Collaboration
    collab_count = models.IntegerField(default=0, blank=True, null=True, verbose_name="合作次数")
    latest_collab = models.CharField(max_length=100, blank=True, null=True, verbose_name="最新合作")

    # 电商平台 - E-commerce platforms (存储为JSON数组，如["SUNLINK", "ARZOPA", "BELIFE"])
    e_commerce_platforms = models.JSONField(blank=True, null=True, verbose_name="电商平台")

    # 分析数据 - Analytics (JSON格式存储销售渠道和类别分布)
    gmv_by_channel = models.JSONField(blank=True, null=True, verbose_name="GMV按渠道分布")
    gmv_by_category = models.JSONField(blank=True, null=True, verbose_name="GMV按类别分布")

    # MCN机构
    mcn = models.CharField(max_length=255, blank=True, null=True, verbose_name="MCN机构")

    # 时间戳
    create_time = models.DateTimeField(auto_now_add=True, verbose_name="创建时间")
    update_time = models.DateTimeField(auto_now=True, verbose_name="更新时间")

    def get_avatar_url(self):
        """获取头像URL，优先返回本地图片，其次返回外部URL"""
        if self.avatar:
            return self.avatar.url
        elif self.avatar_url:
            return self.avatar_url
        return None

    class Meta:
        verbose_name = "达人信息"
        verbose_name_plural = verbose_name
        db_table = "creator_profiles"

    def __str__(self):
        return f"{self.name}"

class Brand(models.Model):
    """品牌模型"""
    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    name = models.CharField(max_length=100, unique=True, verbose_name='品牌名称')
    description = models.TextField(blank=True, null=True, verbose_name='品牌描述')
    logo_url = models.CharField(max_length=255, blank=True, null=True, verbose_name='品牌Logo')
    category = models.CharField(max_length=100, blank=True, null=True, verbose_name='品牌分类')
    source = models.CharField(max_length=100, blank=True, null=True, verbose_name='来源')
    collab_count = models.IntegerField(default=0, verbose_name='合作数量')
    creators_count = models.IntegerField(default=0, verbose_name='创作者数量')
    campaign_id = models.CharField(max_length=100, blank=True, null=True, verbose_name='活动ID')
    
    # 添加数据统计字段
    total_gmv_achieved = models.DecimalField(max_digits=12, decimal_places=2, default=0, verbose_name='总GMV')
    total_views_achieved = models.DecimalField(max_digits=12, decimal_places=2, default=0, verbose_name='总浏览量')
    shop_overall_rating = models.DecimalField(max_digits=3, decimal_places=1, default=0.0, verbose_name='店铺评分')
    
    # 存储关联到此品牌的所有产品和活动知识库ID列表
    dataset_id_list = models.JSONField(default=list, blank=True, verbose_name='知识库ID列表',
                                      help_text='所有关联的知识库ID列表')
    created_at = models.DateTimeField(auto_now_add=True, verbose_name='创建时间')
    updated_at = models.DateTimeField(auto_now=True, verbose_name='更新时间')
    is_active = models.BooleanField(default=True, verbose_name='是否激活')
    
    class Meta:
        verbose_name = '品牌'
        verbose_name_plural = '品牌'
    
    def __str__(self):
        return self.name

class Product(models.Model):
    """产品模型 - 作为一个知识库"""
    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    brand = models.ForeignKey(Brand, on_delete=models.CASCADE, related_name='products', verbose_name='所属品牌')
    name = models.CharField(max_length=100, verbose_name='产品名称')
    description = models.TextField(blank=True, null=True, verbose_name='产品描述')
    image_url = models.CharField(max_length=255, blank=True, null=True, verbose_name='产品图片')
    
    # 添加产品详情字段
    pid = models.CharField(max_length=100, blank=True, null=True, verbose_name='产品ID')
    commission_rate = models.DecimalField(max_digits=5, decimal_places=2, default=0, verbose_name='佣金率')
    open_collab = models.DecimalField(max_digits=5, decimal_places=2, default=0, verbose_name='开放合作率')
    available_samples = models.IntegerField(default=0, verbose_name='可用样品数')
    sales_price_min = models.DecimalField(max_digits=10, decimal_places=2, default=0, verbose_name='最低销售价')
    sales_price_max = models.DecimalField(max_digits=10, decimal_places=2, default=0, verbose_name='最高销售价')
    stock = models.IntegerField(default=0, verbose_name='库存')
    items_sold = models.IntegerField(default=0, verbose_name='已售数量')
    product_rating = models.DecimalField(max_digits=3, decimal_places=1, default=0, verbose_name='产品评分')
    reviews_count = models.IntegerField(default=0, verbose_name='评价数量')
    collab_creators = models.IntegerField(default=0, verbose_name='合作创作者数')
    tiktok_shop = models.BooleanField(default=False, verbose_name='是否TikTok商店')
    
    dataset_id = models.CharField(max_length=100, blank=True, null=True, verbose_name='知识库ID', 
                                help_text='外部知识库系统中的ID')
    external_id = models.CharField(max_length=100, blank=True, null=True, verbose_name='外部ID',
                                 help_text='外部系统中的唯一标识')
    created_at = models.DateTimeField(auto_now_add=True, verbose_name='创建时间')
    updated_at = models.DateTimeField(auto_now=True, verbose_name='更新时间')
    is_active = models.BooleanField(default=True, verbose_name='是否激活')
    
    creators = models.ManyToManyField(
        'CreatorProfile',
        through='CreatorProductMatch',
        related_name='matched_products',
        verbose_name='匹配达人'
    )
    
    class Meta:
        verbose_name = '产品'
        verbose_name_plural = '产品'
        unique_together = ['brand', 'name']
        indexes = [
            models.Index(fields=['brand']),
            models.Index(fields=['dataset_id']),
            models.Index(fields=['is_active']),
            models.Index(fields=['pid']),
        ]
    
    def __str__(self):
        return f"{self.brand.name} - {self.name}"
    
    def save(self, *args, **kwargs):
        """重写save方法，更新品牌的dataset_id_list"""
        is_new = self.pk is None
        super().save(*args, **kwargs)
        
        # 刷新品牌的dataset_id_list
        if is_new and self.is_active and self.dataset_id:
            brand = self.brand
            if self.dataset_id not in brand.dataset_id_list:
                brand.dataset_id_list.append(self.dataset_id)
                brand.save(update_fields=['dataset_id_list', 'updated_at'])

class CreatorProductMatch(models.Model):
    creator = models.ForeignKey('CreatorProfile', on_delete=models.CASCADE, verbose_name='达人')
    product = models.ForeignKey('Product', on_delete=models.CASCADE, verbose_name='产品')
    is_matched = models.BooleanField(default=False, verbose_name='是否匹配')
    match_score = models.FloatField(blank=True, null=True, verbose_name='匹配度分数')

    class Meta:
        unique_together = ('creator', 'product')
        verbose_name = '达人-产品匹配'
        verbose_name_plural = verbose_name

    def __str__(self):
        return f"{self.creator} - {self.product} 匹配: {self.is_matched} 分数: {self.match_score}"








class Dataset(models.Model):
    # 自增主键字段，Django 默认为每个模型提供一个 id 字段
    # 但是如果你需要自定义主键字段，可以用 `primary_key=True` 显式声明
    id = models.AutoField(primary_key=True)  # 自增主键

    # 数据集名称，最长255个字符，不能为空
    name = models.CharField(max_length=255)

    # 创建该数据集的用户，最长100个字符，不能为空
    user = models.CharField(max_length=100)

    # 任务类型，最长50个字符，不能为空
    task_type = models.CharField(max_length=16)

    # 数据集大小，使用BigIntegerField来表示较大的整数
    size = models.CharField(max_length=64)

    # 数据集的总图片数
    number = models.IntegerField()

    #数据集的描述
    description = models.TextField(default="")

    # 创建时间，默认为当前时间戳
    create_time = models.DateTimeField(auto_now_add=True)  # 自动设置为当前时间（只在创建时）

    # 数据集的类别，使用 JSONField 存储类别信息
    categories = models.JSONField(default=list)  # 默认值为一个空列表

    # 该数据集是否上传至minio的一个标识
    is_upload = models.BooleanField(default=False)  # 表示数据集是否已上传，默认为 False

    def update_is_upload(self, upload_status):
        """更新数据集的上传状态"""
        self.is_upload = upload_status
        self.save()  # 保存更改到数据库

    def __str__(self):
        return self.name  # 返回数据集名称作为对象的字符串表示
    

class Project(models.Model):
    id = models.AutoField(primary_key=True)
    name = models.CharField(max_length=255)
    user = models.CharField(max_length=100)
    task_type = models.CharField(max_length=16)
    description = models.TextField(default="")
    create_time = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.name