import os
import sys
import django

# 设置Django环境
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'danmu_activity.settings')
django.setup()

from activity.models import Danmu

def add_test_danmus():
    """添加测试弹幕数据"""
    test_danmus = [
        {
            'name': '测试用户1',
            'content': '这是第一条测试弹幕！',
            'is_approved': True
        },
        {
            'name': '测试用户2',
            'content': '这是第二条测试弹幕，祝活动圆满成功！',
            'is_approved': True
        },
        {
            'name': '测试用户3',
            'content': '这是第三条测试弹幕，大家玩得开心！',
            'is_approved': True
        },
        {
            'name': '测试用户4',
            'content': '这是第四条测试弹幕，感谢组织者！',
            'is_approved': True
        },
        {
            'name': '测试用户5',
            'content': '这是第五条测试弹幕，祝大家身体健康！',
            'is_approved': True
        }
    ]
    
    print("开始添加测试弹幕...")
    for danmu_data in test_danmus:
        Danmu.objects.create(**danmu_data)
    
    print(f"成功添加 {len(test_danmus)} 条测试弹幕！")

if __name__ == '__main__':
    add_test_danmus()