import os
import random
import sys

import django

# 配置 Django 环境
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "server.settings")  # 修改为你的 settings 路径
django.setup()

# 2. 导入 Faker 和 模型
from faker import Faker
from news.models import News, NewsSection
from users.models import User

# 初始化 Faker (zh_CN 表示生成中文数据)
fake = Faker('zh_CN')


def seed_news(count=20):
    print(f"🚀 开始生成 {count} 条新闻数据...")

    # --- 1. 获取关联数据 ---
    # 获取所有板块
    sections = list(NewsSection.objects.all())
    # 获取所有用户
    users = list(User.objects.all())

    # 检查基础数据是否存在
    if not sections:
        print("❌ 错误：数据库中没有找到任何板块 (NewsSection)！请先添加板块。")
        return
    if not users:
        print("⚠️ 警告：数据库中没有用户 (User)，作者字段将设为空。")

    # --- 2. 生成数据 ---
    created_count = 0
    for i in range(count):
        # 随机选择一个板块
        section = random.choice(sections)
        # 随机选择一个作者 (如果有用户)
        author = random.choice(users) if users else None

        # 生成富文本内容 (模拟 CKEditor 输出的 HTML)
        # paragrahs 生成多段文本，join 拼接
        paragraphs = fake.paragraphs(nb=random.randint(3, 6))
        content_html = "\n".join([f"<p>{p}</p>" for p in paragraphs])

        # 创建新闻对象
        news = News.objects.create(
            title=fake.sentence(nb_words=random.randint(4, 8))[:-1],  # 生成标题并去掉末尾句号
            section=section,
            author=author,
            content=content_html,
            is_published=True,  # 默认已发布
            # 封面图留空，因为自动生成图片文件比较复杂，建议手动上传测试图
        )

        created_count += 1
        print(f" - [{section.name}] {news.title}")

    print(f"✅ 成功生成 {created_count} 条新闻数据！")


if __name__ == "__main__":
    seed_news(30)