49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
#!/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数据库初始化失败！")
|
||
|