from django.db import models import uuid from django.utils import timezone 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, 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='是否激活') 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 Campaign(models.Model): """活动模型 - 作为一个知识库""" id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False) brand = models.ForeignKey(Brand, on_delete=models.CASCADE, related_name='campaigns', 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='活动图片') # 活动相关字段 service = models.CharField(max_length=100, blank=True, null=True, verbose_name='服务类型') creator_type = models.CharField(max_length=100, blank=True, null=True, verbose_name='创作者类型') creator_level = models.CharField(max_length=100, blank=True, null=True, verbose_name='创作者等级') creator_category = models.CharField(max_length=100, blank=True, null=True, verbose_name='创作者分类') creators_count = models.IntegerField(default=0, verbose_name='创作者数量') gmv = models.CharField(max_length=100, blank=True, null=True, verbose_name='GMV范围') followers = models.CharField(max_length=100, blank=True, null=True, verbose_name='粉丝数范围') views = models.CharField(max_length=100, blank=True, null=True, verbose_name='浏览量范围') budget = models.CharField(max_length=100, blank=True, null=True, verbose_name='预算范围') link_product = models.ManyToManyField(Product, blank=True, related_name='campaigns', verbose_name='关联产品') # 时间信息 start_date = models.DateTimeField(blank=True, null=True, verbose_name='开始日期') end_date = models.DateTimeField(blank=True, null=True, verbose_name='结束日期') # 知识库信息 dataset_id = models.CharField(max_length=100, 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='是否激活') 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=['start_date']), models.Index(fields=['end_date']), ] 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 BrandChatSession(models.Model): """品牌聊天会话模型""" id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False) brand = models.ForeignKey(Brand, on_delete=models.CASCADE, related_name='chat_sessions', verbose_name='品牌') session_id = models.CharField(max_length=100, unique=True, verbose_name='会话ID') title = models.CharField(max_length=200, default='新对话', verbose_name='会话标题') # 存储此次会话使用的所有知识库ID dataset_id_list = models.JSONField(default=list, blank=True, verbose_name='知识库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 = '品牌聊天会话' indexes = [ models.Index(fields=['brand']), models.Index(fields=['session_id']), models.Index(fields=['created_at']), ] def __str__(self): return f"{self.brand.name} - {self.title}"