#!/usr/bin/env python3
"""导入仓库基础数据到数据库"""
import csv
import pymysql

DB_CONFIG = {
    "host": "127.0.0.1",
    "port": 3306,
    "user": "dy_app",
    "password": "dy_data_2026",
    "database": "dy_data",
    "charset": "utf8mb4"
}

conn = pymysql.connect(
    host=DB_CONFIG["host"],
    port=DB_CONFIG["port"],
    user=DB_CONFIG["user"],
    password=DB_CONFIG["password"],
    database=DB_CONFIG["database"],
    charset="utf8mb4",
    cursorclass=pymysql.cursors.DictCursor
)

with open('tmp_csv/warehouses_utf8.csv', 'r', encoding='utf-8') as f:
    reader = csv.DictReader(f)
    rows = list(reader)

print(f"读取到 {len(rows)} 条仓库记录")

with conn.cursor() as cur:
    for row in rows:
        cur.execute(
            "INSERT INTO warehouses (warehouse_code, warehouse_name, category) VALUES (%s, %s, %s) "
            "ON DUPLICATE KEY UPDATE warehouse_name=VALUES(warehouse_name), category=VALUES(category)",
            (row['仓库编号'].strip(), row['仓库名称'].strip(), row['类别'].strip())
        )
conn.commit()
conn.close()
print("导入完成 ✅")