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

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

from application.cpc_mgnt.models import (
    SealUsageRegistry, SealAndDocumentBorrowingRegistry, ContractManagementLedger,
    CompanyPolicies
)

def generate_test_data():
    # 创建用印登记表数据
    for i in range(10):
        seal_usage = SealUsageRegistry.objects.create(
            applicant=f"申请人{i}",
            seal_date=date.today() - timedelta(days=random.randint(1, 365)),
            seal_entity=f"用印主体{i}",
            seal_type=f"用印类型{i}",
            document_name=f"文件名称{i}",
            number_of_copies=random.randint(1, 10),
            handler=f"经办人{i}"
        )
        print(f"Created SealUsageRegistry: {seal_usage}")

    # 创建印章证件借用登记表数据
    for i in range(10):
        borrowing_registry = SealAndDocumentBorrowingRegistry.objects.create(
            entity_name=f"主体名称{i}",
            item_type=f"借章/证类型{i}",
            purpose=f"用途{i}",
            borrower=i,
            borrowing_time=datetime.now() - timedelta(days=random.randint(1, 365)),
            expected_return_date=date.today() + timedelta(days=random.randint(1, 365)),
            lender=i,
            returnee=i,
            return_date=date.today() + timedelta(days=random.randint(1, 365)),
            verifier=i
        )
        print(f"Created SealAndDocumentBorrowingRegistry: {borrowing_registry}")

    # 创建合同管理台账数据
    for i in range(10):
        contract_ledger = ContractManagementLedger.objects.create(
            project_name=f"项目名称{i}",
            contract_number=f"合同编号{i}",
            contract_name=f"合同名称{i}",
            primary_department=f"一级部门{i}",
            project_leader=f"项目负责人{i}",
            project_progress=f"项目进度{i}",
            contract_status=random.choice(['已生效未履行', '履行中', '履行中止', '协商变更', '期满终止', '协商解除', '未生效']),
            contract_type=random.choice(['业务类', '采购类', '用工类', '租赁类', '其他类']),
            business_type=f"业务类型{i}",
            number_of_copies=random.randint(1, 10),
            cost_agreement=f"费用约定{i}",
            review_date=date.today() - timedelta(days=random.randint(1, 365)),
            signing_date=date.today() - timedelta(days=random.randint(1, 365)),
            effective_date=date.today() - timedelta(days=random.randint(1, 365)),
            expiration_date=date.today() + timedelta(days=random.randint(1, 365)),
            our_side=f"我方单位{i}",
            other_side=f"对方单位{i}",
            submitter_for_review=f"提交审核人{i}",
            custodian_of_original=f"合同原件保管人{i}",
            payment_agreement_needed=random.choice([True, False]),
            signed=random.choice([True, False]),
            backup_location=f"备份地点{i}",
            notes=f"备注{i}"
        )
        print(f"Created ContractManagementLedger: {contract_ledger}")

    # 创建公司制度表数据
    for i in range(10):
        company_policy = CompanyPolicies.objects.create(
            issuing_unit=f"发布单位{i}",
            policy_category=random.choice(['公司治理', '人力资源', '财务']),
            policy_name=f"制度名称{i}",
            version=f"版本{i}",
            publication_date=date.today() - timedelta(days=random.randint(1, 365)),
            effective_date=date.today() - timedelta(days=random.randint(1, 365)),
            status=random.choice(['有效', '待生效', '废止'])
        )
        print(f"Created CompanyPolicies: {company_policy}")

if __name__ == "__main__":
    generate_test_data()