import os
import django
import random
from datetime import datetime, timedelta

# 设置 Django 环境
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'XH_Digital_Management.settings')
django.setup()

from application.opa_mgnt.models import (
    SocialMediaAccountRegistration, AccountOperationManagement,
    WebsiteRegistration, WebsiteMaintenanceRecord, WebsiteExpirationModificationRecord
)


def generate_test_data():
    # 创建新媒体账号登记
    social_media_accounts = []
    for i in range(5):
        account = SocialMediaAccountRegistration.objects.create(
            platform=f"平台{i}",
            account_name=f"账号名称{i}",
            primary_department=f"一级部门{i}",
            secondary_department=f"二级部门{i}",
            operation_account=f"运营账号{i}",
            password=f"密码{i}",
            operator=f"运营者{i}"
        )
        social_media_accounts.append(account)
        print(f"Created SocialMediaAccountRegistration: {account}")

    # 创建账号运营管理记录
    for account in social_media_accounts:
        for i in range(3):
            operation_record = AccountOperationManagement.objects.create(
                platform=account.platform,
                account_name=account.account_name,
                content_title=f"发布内容标题{i}",
                publication_time=datetime.now() - timedelta(days=random.randint(1, 30)),
                views=random.randint(100, 10000),
                favorites=random.randint(10, 1000),
                comments=random.randint(1, 500),
                shares=random.randint(1, 200),
                likes=random.randint(1, 5000),
                followers=random.randint(1000, 50000),
                update_time=datetime.now()
            )
            print(f"Created AccountOperationManagement: {operation_record}")

    # 创建网站登记
    websites = []
    for i in range(5):
        website = WebsiteRegistration.objects.create(
            website_name=f"官网名称{i}",
            primary_department=f"一级部门{i}",
            secondary_department=f"二级部门{i}",
            account=f"账号{i}",
            password=f"密码{i}",
            login_url=f"登录入口{i}",
            tech_support=f"技术支持{i}",
            domain=f"域名{i}.com",
            expiration_date=datetime.now().date() + timedelta(days=random.randint(365, 1825)),
            operator=f"运营者{i}"
        )
        websites.append(website)
        print(f"Created WebsiteRegistration: {website}")

    # 创建网站运维记录
    for website in websites:
        for i in range(3):
            maintenance_record = WebsiteMaintenanceRecord.objects.create(
                website_name=website.website_name,
                maintenance_content=f"维护内容{i}",
                maintenance_details=f"维护详情{i}",
                maintenance_time=datetime.now() - timedelta(days=random.randint(1, 30)),
                maintainer=f"维护者{i}"
            )
            print(f"Created WebsiteMaintenanceRecord: {maintenance_record}")

    # 创建网站到期时间修改记录表
    for website in websites:
        modification_record = WebsiteExpirationModificationRecord.objects.create(
            website_registration=website,
            original_expiration_date=website.expiration_date,
            new_expiration_date=website.expiration_date + timedelta(days=random.randint(365, 730)),
            modification_date=datetime.now(),
            cost=random.uniform(1000, 10000),
            modified_by=f"修改人{i}"
        )
        print(f"Created WebsiteExpirationModificationRecord: {modification_record}")


if __name__ == "__main__":
    generate_test_data()