#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
初始化数据库，从JSON文件导入工单数据
"""

import json
import os
from database import Database

def import_from_json(db: Database, json_file: str = "work_orders.json"):
    """从JSON文件导入工单数据"""
    if not os.path.exists(json_file):
        print(f"文件不存在: {json_file}")
        return False
    
    try:
        with open(json_file, 'r', encoding='utf-8') as f:
            orders = json.load(f)
        
        count = 0
        for order in orders:
            if db.create_work_order(order):
                count += 1
                print(f"✓ 导入工单: {order.get('trace_id')} - {order.get('process_id')}")
            else:
                print(f"✗ 导入失败: {order.get('trace_id')} - {order.get('process_id')}")
        
        print(f"\n导入完成，共导入 {count} 个工单")
        return True
    except Exception as e:
        print(f"导入失败: {e}")
        return False


if __name__ == '__main__':
    print("="*60)
    print("初始化数据库")
    print("="*60)
    
    db = Database("wrench.db")
    
    # 从JSON文件导入数据
    if import_from_json(db):
        print("\n数据库初始化成功！")
    else:
        print("\n数据库初始化失败！")