50 lines
1.6 KiB
Python
Executable File
50 lines
1.6 KiB
Python
Executable File
#!/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("更新完成！")
|