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