#!/usr/bin/env python3
"""
将现有期初库存和采购需求数据的工厂字段更新为"友辉"
"""

import sqlite3

DB_PATH = 'server/data.db'

def update_factory():
    """更新现有数据的工厂字段为友辉"""
    conn = sqlite3.connect(DB_PATH)
    c = conn.cursor()
    
    try:
        # 更新期初库存表
        c.execute("UPDATE initial_stock SET factory='友辉' WHERE factory IS NULL OR factory=''")
        initial_stock_rows = c.rowcount
        
        # 更新采购需求表
        c.execute("UPDATE purchase_demand SET factory='友辉' WHERE factory IS NULL OR factory=''")
        purchase_demand_rows = c.rowcount
        
        conn.commit()
        
        print(f"✅ 期初库存表：成功更新 {initial_stock_rows} 条记录的工厂字段为'友辉'")
        print(f"✅ 采购需求表：成功更新 {purchase_demand_rows} 条记录的工厂字段为'友辉'")
        
        # 查询验证
        c.execute("SELECT COUNT(*) FROM initial_stock WHERE factory='友辉'")
        initial_count = c.fetchone()[0]
        c.execute("SELECT COUNT(*) FROM purchase_demand WHERE factory='友辉'")
        demand_count = c.fetchone()[0]
        
        print(f"📊 当前友辉工厂的期初库存记录数: {initial_count}")
        print(f"📊 当前友辉工厂的采购需求记录数: {demand_count}")
        
    except Exception as e:
        print(f"❌ 更新失败: {e}")
        conn.rollback()
    finally:
        conn.close()

if __name__ == '__main__':
    print("开始更新数据的工厂字段...")
    print("=" * 50)
    update_factory()
    print("=" * 50)
    print("更新完成！")