#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
数据库迁移脚本：删除work_orders表中的station字段
"""

import sqlite3
import os
from datetime import datetime


def migrate_remove_station():
    """删除work_orders表中的station字段"""
    db_path = "wrench.db"
    
    if not os.path.exists(db_path):
        print(f"数据库文件不存在: {db_path}")
        return
    
    print("="*60)
    print("数据库迁移：删除station字段")
    print("="*60)
    
    conn = sqlite3.connect(db_path)
    cursor = conn.cursor()
    
    try:
        # 检查表是否存在station字段
        cursor.execute("PRAGMA table_info(work_orders)")
        columns = cursor.fetchall()
        column_names = [col[1] for col in columns]
        
        if 'station' not in column_names:
            print("[OK] station字段不存在，无需迁移")
            conn.close()
            return
        
        print(f"检测到station字段，开始迁移...")
        print(f"当前字段: {column_names}")
        
        # SQLite不支持直接删除列，需要重建表
        # 1. 创建新表（不包含station字段）
        print("\n1. 创建新表结构...")
        cursor.execute('''
            CREATE TABLE IF NOT EXISTS work_orders_new (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                trace_id TEXT NOT NULL,
                process_id TEXT NOT NULL,
                process_name TEXT,
                product_name TEXT,
                operator TEXT,
                status TEXT DEFAULT 'pending',
                bolts TEXT NOT NULL,
                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
                updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
                UNIQUE(trace_id, process_id)
            )
        ''')
        
        # 2. 复制数据（排除station字段）
        print("2. 复制数据...")
        cursor.execute('''
            INSERT INTO work_orders_new 
            (id, trace_id, process_id, process_name, product_name, operator, status, bolts, created_at, updated_at)
            SELECT 
                id, trace_id, process_id, process_name, product_name, operator, status, bolts, created_at, updated_at
            FROM work_orders
        ''')
        
        # 3. 删除旧表
        print("3. 删除旧表...")
        cursor.execute('DROP TABLE work_orders')
        
        # 4. 重命名新表
        print("4. 重命名新表...")
        cursor.execute('ALTER TABLE work_orders_new RENAME TO work_orders')
        
        conn.commit()
        
        # 验证
        cursor.execute("PRAGMA table_info(work_orders)")
        new_columns = cursor.fetchall()
        new_column_names = [col[1] for col in new_columns]
        
        print(f"\n[OK] 迁移完成！")
        print(f"新表字段: {new_column_names}")
        
        if 'station' in new_column_names:
            print("[WARN] 警告：station字段仍然存在")
        else:
            print("[OK] station字段已成功删除")
        
        # 统计数据
        cursor.execute("SELECT COUNT(*) FROM work_orders")
        count = cursor.fetchone()[0]
        print(f"[OK] 工单数量: {count}")
        
    except Exception as e:
        conn.rollback()
        print(f"[FAIL] 迁移失败: {e}")
        import traceback
        traceback.print_exc()
    finally:
        conn.close()
    
    print("="*60)


if __name__ == "__main__":
    migrate_remove_station()