from django.core.management.base import BaseCommand from django.utils import timezone from apps.daren_detail.models import ( CollaborationMetrics, VideoMetrics, LiveMetrics, CreatorProfile, CreatorCampaign, BrandCampaign, CreatorVideo, FollowerMetrics, TrendMetrics, PublicCreatorPool, PrivateCreatorPool, PrivateCreatorRelation ) from apps.expertproducts.models import Product as ExpertProduct, Creator as ExpertCreator, Negotiation, Message from apps.discovery.models import SearchSession, Creator as DiscoveryCreator from apps.brands.models import Brand, Product as BrandProduct, Campaign, BrandChatSession from apps.template.models import TemplateCategory, Template from apps.user.models import User import random from datetime import datetime, timedelta import uuid class Command(BaseCommand): help = '填充测试数据到所有相关模型' def handle(self, *args, **kwargs): self.stdout.write('开始填充数据...') # 创建用户（避免重复） user, created = User.objects.get_or_create( email='test@example.com', defaults={ 'password': 'testpassword', 'company': '测试公司', 'name': '测试用户', 'is_first_login': False, 'last_login': timezone.now() } ) # 创建品牌（避免重复） brands = [] brand_names = ['U品牌', 'R品牌', 'X品牌', 'Q品牌', 'A品牌', 'M品牌'] for name in brand_names: brand, _ = Brand.objects.get_or_create( name=name, defaults={ 'description': f'{name}的描述信息', 'logo_url': f'https://example.com/logos/{name}.png', 'category': '电子产品', 'source': '内部', 'collab_count': random.randint(5, 20), 'creators_count': random.randint(10, 50), 'campaign_id': str(uuid.uuid4()), 'total_gmv_achieved': random.randint(10000, 100000), 'total_views_achieved': random.randint(100000, 1000000), 'shop_overall_rating': round(random.uniform(3.5, 5.0), 1), 'dataset_id_list': [str(uuid.uuid4()) for _ in range(3)] } ) brands.append(brand) # 创建品牌产品（避免重复） products = [] for brand in brands: for i in range(3): product, _ = BrandProduct.objects.get_or_create( brand=brand, name=f'{brand.name}产品{i+1}', defaults={ 'description': f'{brand.name}产品{i+1}的详细描述', 'image_url': f'https://example.com/products/{brand.name}_{i+1}.jpg', 'pid': f'PID_{uuid.uuid4().hex[:8]}', 'commission_rate': random.uniform(5, 20), 'open_collab': random.uniform(10, 30), 'available_samples': random.randint(10, 100), 'sales_price_min': random.uniform(100, 500), 'sales_price_max': random.uniform(500, 2000), 'stock': random.randint(100, 1000), 'items_sold': random.randint(50, 500), 'product_rating': round(random.uniform(3.5, 5.0), 1), 'reviews_count': random.randint(10, 100), 'collab_creators': random.randint(5, 20), 'tiktok_shop': random.choice([True, False]), 'dataset_id': str(uuid.uuid4()) } ) products.append(product) # 创建活动（避免重复） campaigns = [] for brand in brands: campaign, _ = Campaign.objects.get_or_create( brand=brand, name=f'{brand.name}活动', defaults={ 'description': f'{brand.name}的活动描述', 'image_url': f'https://example.com/campaigns/{brand.name}.jpg', 'service': '直播带货', 'creator_type': 'KOL', 'creator_level': 'L3', 'creator_category': '电子产品', 'creators_count': random.randint(5, 20), 'gmv': '10000-50000', 'followers': '10000-100000', 'views': '100000-1000000', 'budget': '5000-20000', 'start_date': timezone.now(), 'end_date': timezone.now() + timedelta(days=30), 'dataset_id': str(uuid.uuid4()), 'status': 'in_progress', 'gmv_achieved': '25000', 'views_achieved': '500000', 'video_link': 'https://example.com/videos/campaign.mp4' } ) campaign.link_product.set(random.sample(products, 2)) campaigns.append(campaign) # 创建创作者档案 creators = [] for i in range(10): creator = CreatorProfile.objects.create( name=f'创作者{i+1}', avatar_url=f'https://example.com/avatars/creator_{i+1}.jpg', email=f'creator{i+1}@example.com', instagram=f'creator{i+1}', tiktok_link=f'https://tiktok.com/@{i+1}', location='上海', live_schedule='每周一、三、五 20:00-22:00', category=random.choice([c[0] for c in CreatorProfile.CATEGORY_CHOICES]), e_commerce_level=random.randint(1, 7), exposure_level=random.choice([e[0] for e in CreatorProfile.EXPOSURE_LEVEL_CHOICES]), followers=random.randint(10000, 1000000), gmv=random.uniform(10000, 1000000), items_sold=random.uniform(1000, 10000), avg_video_views=random.randint(10000, 100000), pricing_min=random.uniform(1000, 5000), pricing_max=random.uniform(5000, 20000), pricing_package='基础套餐', collab_count=random.randint(5, 50), latest_collab='最新合作项目', e_commerce_platforms=['TikTok', 'Instagram'], gmv_by_channel={'TikTok': 60, 'Instagram': 40}, gmv_by_category={'电子产品': 70, '服装': 30}, mcn='知名MCN机构' ) creators.append(creator) # 创建协作指标 for creator in creators: CollaborationMetrics.objects.create( avg_commission_rate=random.uniform(5, 20), products_count=random.randint(10, 50), brand_collaborations=random.randint(5, 20), min_product_price=random.uniform(100, 500), max_product_price=random.uniform(500, 2000), start_date=timezone.now().date(), end_date=timezone.now().date() + timedelta(days=30), creator=creator ) # 创建视频指标 for creator in creators: for video_type in ['regular', 'shoppable']: VideoMetrics.objects.create( video_type=video_type, gpm=random.uniform(100, 1000), videos_count=random.randint(10, 50), avg_views=random.uniform(10000, 100000), avg_engagement=random.uniform(1, 10), avg_likes=random.randint(1000, 10000), start_date=timezone.now().date(), end_date=timezone.now().date() + timedelta(days=30), creator=creator ) # 创建直播指标 for creator in creators: for live_type in ['regular', 'shoppable']: LiveMetrics.objects.create( live_type=live_type, gpm=random.uniform(100, 1000), lives_count=random.randint(5, 20), avg_views=random.uniform(10000, 100000), avg_engagement=random.uniform(1, 10), avg_likes=random.randint(1000, 10000), start_date=timezone.now().date(), end_date=timezone.now().date() + timedelta(days=30), creator=creator ) # 创建创作者视频 for creator in creators: for i in range(3): CreatorVideo.objects.create( creator=creator, title=f'视频标题{i+1}', description=f'视频描述{i+1}', thumbnail_url=f'https://example.com/thumbnails/video_{i+1}.jpg', video_url=f'https://example.com/videos/video_{i+1}.mp4', video_id=f'VID_{uuid.uuid4().hex[:8]}', video_type=random.choice(['regular', 'product']), badge=random.choice(['red', 'gold']), view_count=random.randint(10000, 100000), like_count=random.randint(1000, 10000), comment_count=random.randint(100, 1000), has_product=random.choice([True, False]), product_name=f'产品{i+1}' if random.choice([True, False]) else None, product_url=f'https://example.com/products/{i+1}' if random.choice([True, False]) else None, release_date=timezone.now().date() ) # 创建粉丝指标 for creator in creators: FollowerMetrics.objects.create( creator=creator, start_date=timezone.now().date(), end_date=timezone.now().date() + timedelta(days=30), female_percentage=random.uniform(40, 60), male_percentage=random.uniform(40, 60), age_18_24_percentage=random.uniform(20, 40), age_25_34_percentage=random.uniform(30, 50), age_35_44_percentage=random.uniform(10, 30), age_45_54_percentage=random.uniform(5, 15), age_55_plus_percentage=random.uniform(1, 10), location_data={ '上海': random.uniform(10, 30), '北京': random.uniform(10, 30), '广州': random.uniform(5, 20), '深圳': random.uniform(5, 20), '杭州': random.uniform(5, 15) } ) # 创建趋势指标 for creator in creators: for i in range(10): TrendMetrics.objects.create( creator=creator, date=timezone.now().date() - timedelta(days=i), gmv=random.uniform(1000, 10000), items_sold=random.randint(100, 1000), followers_count=random.randint(10000, 100000), video_views=random.randint(10000, 100000), engagement_rate=random.uniform(1, 10) ) # 创建公有达人库 for creator in creators: PublicCreatorPool.objects.create( creator=creator, category=creator.category, remark=f'备注信息：{creator.name}的详细信息' ) # 创建私有达人库（避免重复） private_pool, _ = PrivateCreatorPool.objects.get_or_create( user=user, name='我的收藏', defaults={ 'description': '收藏的达人列表', 'is_default': True } ) # 创建私有达人关联 for creator in random.sample(creators, 5): PrivateCreatorRelation.objects.create( private_pool=private_pool, creator=creator, added_from_public=True, notes=f'关于{creator.name}的笔记', status=random.choice(['active', 'archived', 'favorite']) ) # 创建专家产品 expert_products = [] for i in range(10): p = ExpertProduct.objects.create( name=f'专家产品{i+1}', category='电子产品', max_price=random.uniform(1000, 5000), min_price=random.uniform(500, 1000), description=f'专家产品{i+1}的详细描述' ) expert_products.append(p) # 创建专家创作者 expert_creators = [] for i in range(10): c = ExpertCreator.objects.create( name=f'专家创作者{i+1}', sex=random.choice(['男', '女']), age=random.randint(18, 45), category='带货类', followers=random.randint(10000, 1000000) ) expert_creators.append(c) # 创建谈判记录 for i in range(10): negotiation = Negotiation.objects.create( creator=expert_creators[i], product=expert_products[i], status=random.choice(['brand_review', 'price_negotiation', 'contract_review']), current_round=random.randint(1, 5), context={'current_price': random.uniform(500, 2000)} ) # 为每个谈判创建消息 for j in range(3): Message.objects.create( negotiation=negotiation, role=random.choice(['user', 'assistant']), content=f'谈判消息{j+1}', stage=negotiation.status ) # 创建搜索会话 for i in range(10): session = SearchSession.objects.create( session_number=i+1, creator_count=random.randint(5, 20), shoppable_creators=random.randint(2, 10), avg_followers=random.uniform(10000, 100000), avg_gmv=random.uniform(10000, 100000), avg_video_views=random.uniform(10000, 100000) ) # 为每个会话创建创作者 for j in range(5): DiscoveryCreator.objects.create( session=session, name=f'发现创作者{j+1}', avatar=f'https://example.com/avatars/discovery_{j+1}.jpg', category=random.choice([c[0] for c in DiscoveryCreator.CATEGORIES]), ecommerce_level=random.choice([l[0] for l in DiscoveryCreator.ECOMMERCE_LEVELS]), exposure_level=random.choice([l[0] for l in DiscoveryCreator.EXPOSURE_LEVELS]), followers=random.uniform(10000, 1000000), gmv=random.uniform(10000, 100000), items_sold=random.uniform(1000, 10000), avg_video_views=random.uniform(10000, 100000), has_ecommerce=random.choice([True, False]), tiktok_url=f'https://tiktok.com/@{j+1}' ) # 创建模板分类 categories = [] for i in range(5): category = TemplateCategory.objects.create( name=f'模板分类{i+1}', description=f'模板分类{i+1}的描述' ) categories.append(category) # 创建模板 for category in categories: for i in range(2): Template.objects.create( title=f'模板{i+1}', content=f'这是模板{i+1}的详细内容，包含了很多有用的信息。', category=category, mission=random.choice([m[0] for m in Template.MISSION_CHOICES]), platform=random.choice([p[0] for p in Template.PLATFORM_CHOICES]), collaboration_type=random.choice([c[0] for c in Template.COLLABORATION_CHOICES]), service=random.choice([s[0] for s in Template.SERVICE_CHOICES]), is_public=random.choice([True, False]) ) self.stdout.write(self.style.SUCCESS('数据填充完成！'))