41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
#!/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("导入完成 ✅")
|