#!/usr/bin/env python
import os
import django

os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'tech_portal.settings')
django.setup()

from django.contrib.auth.models import User
from portal.models import Service, ProductCategory, Product, PriceRecord, Customer, CompanyInfo
from django.utils import timezone

def create_sample_data():
    print("创建示例数据...")
    
    # 创建服务项目
    services = [
        {"name": "计算机维修", "description": "专业计算机硬件维修和软件维护服务", "icon": "fa-laptop"},
        {"name": "网络工程", "description": "企业网络规划、部署和维护服务", "icon": "fa-network-wired"},
        {"name": "弱电系统", "description": "安防监控、门禁系统等弱电工程", "icon": "fa-video"},
        {"name": "IT咨询", "description": "企业IT系统规划和咨询服务", "icon": "fa-chart-line"},
    ]
    
    for service_data in services:
        Service.objects.get_or_create(
            name=service_data['name'],
            defaults=service_data
        )
    
    # 创建产品分类
    categories = [
        "计算机配件",
        "网络设备", 
        "弱电材料",
        "办公设备"
    ]
    
    for cat_name in categories:
        ProductCategory.objects.get_or_create(name=cat_name)
    
    # 创建公司信息
    company, created = CompanyInfo.objects.get_or_create(
        name="科技技术有限公司",
        defaults={
            "description": "专业提供技术产品和服务解决方案",
            "address": "上海市浦东新区张江高科技园区",
            "phone": "400-123-4567",
            "email": "info@techcompany.com",
            "website": "https://www.techcompany.com"
        }
    )
    
    # 创建示例客户
    customer, created = Customer.objects.get_or_create(
        name="示例科技有限公司",
        defaults={
            "contact_person": "张经理",
            "phone": "13800138000",
            "email": "zhang@example.com",
            "address": "北京市海淀区中关村"
        }
    )
    
    print("示例数据创建完成！")
    print(f"服务项目: {Service.objects.count()}个")
    print(f"产品分类: {ProductCategory.objects.count()}个")
    print(f"公司信息: {CompanyInfo.objects.count()}个")
    print(f"客户: {Customer.objects.count()}个")

if __name__ == "__main__":
    create_sample_data()