import os
import django
import random
from datetime import datetime

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

from application.fin_tbl.models import DepartmentOperationalData


def generate_test_data():
    # 生成 DepartmentOperationalData 测试数据
    for year in range(2018, 2024):
        for month in range(1, 13):
            year_month = f"{year}-{str(month).zfill(2)}"
            data = DepartmentOperationalData.objects.create(
                department=f"部门{random.randint(1, 10)}",
                year_month=year_month,
                current_month_revenue=random.uniform(100000, 1000000),
                current_month_new_revenue=random.uniform(5000, 100000),
                current_month_existing_revenue=random.uniform(50000, 500000),
                current_month_cost=random.uniform(50000, 700000),
                gross_profit=random.uniform(10000, 300000),
                taxes_and_surcharges=random.uniform(5000, 50000),
                expenses=random.uniform(5000, 50000),
                operating_profit=random.uniform(10000, 200000),
                operational_cash_flow=random.uniform(10000, 200000)
            )
            print(f"Created DepartmentOperationalData: {data}")


if __name__ == "__main__":
    generate_test_data()