XH_Digital_Management/application/ops_tbl/tests.py

32 lines
1.1 KiB
Python

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()