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.asset_mgnt.models import FixedAssetsInventory, IntangibleAssetsInventory


def generate_asset_test_data():
    # 创建固定资产清单
    for i in range(5):
        asset = FixedAssetsInventory.objects.create(
            asset_id=f"FA{i}",
            asset_category=f"固定资产类别{i}",
            brand=f"品牌{i}",
            model=f"规格型号{i}",
            unit=f"单位{i}",
            quantity=random.randint(1, 10),
            affiliated_entity=f"归属主体{i}",
            location=f"存放地点{i}",
            department=f"所属部门{i}",
            user=f"使用人{i}",
            purchase_date=datetime.now().date() - timedelta(days=random.randint(1, 365)),
            recorded_date=datetime.now().date(),
            original_value=random.uniform(1000, 10000),
            residual_value=random.uniform(100, 1000),
            depreciation_method=f"折旧方法{i}",
            depreciation_years=random.randint(3, 10),
            depreciation_months=random.randint(1, 11),
            book_value=random.uniform(500, 5000),
            status=random.choice(['正常使用', '报废', '赠送', '闲置', '其他'])
        )
        print(f"Created FixedAssetsInventory: {asset}")

    # 创建无形资产清单
    for i in range(5):
        asset = IntangibleAssetsInventory.objects.create(
            asset_id=f"IA{i}",
            asset_category=f"无形资产类别{i}",
            brand=f"品牌{i}",
            model=f"规格型号{i}",
            unit=f"单位{i}",
            quantity=random.randint(1, 10),
            affiliated_entity=f"归属主体{i}",
            department=f"所属部门{i}",
            user=f"使用人{i}",
            purchase_date=datetime.now().date() - timedelta(days=random.randint(1, 365)),
            recorded_date=datetime.now().date(),
            original_value=random.uniform(1000, 10000),
            residual_value=random.uniform(100, 1000),
            amortization_method=f"摊销方法{i}",
            amortization_years=random.randint(3, 10),
            amortization_months=random.randint(1, 11),
            book_value=random.uniform(500, 5000),
            status=random.choice(['正常使用', '闲置', '出售', '其他'])
        )
        print(f"Created IntangibleAssetsInventory: {asset}")


if __name__ == "__main__":
    generate_asset_test_data()