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.ops_tbl.models import DepartmentExpenseData


def generate_test_data():
    # 生成 DepartmentExpenseData 测试数据
    for year in range(2018, 2024):
        for month in range(1, 13):
            year_month = f"{year}-{str(month).zfill(2)}"
            data = DepartmentExpenseData.objects.create(
                primary_department="测试部门",
                year_month=year_month,
                expense_type=random.choice(["营销", "行政", "研发", "销售"]),
                expense_detail=random.choice(["办公耗材", "差旅费", "培训费", "福利费"]),
                budget_limit=random.uniform(100, 1000),
                reimbursed_amount=random.uniform(50, 500),
                available_budget=random.uniform(50, 500)
            )
            print(f"Created DepartmentExpenseData: {data}")


if __name__ == "__main__":
    generate_test_data()