38 lines
928 B
Python
38 lines
928 B
Python
|
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
为 customer_orders 表添加 customer_name 列
|
||
|
|
"""
|
||
|
|
|
||
|
|
import sqlite3
|
||
|
|
import os
|
||
|
|
|
||
|
|
# 数据库路径
|
||
|
|
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
||
|
|
DB_PATH = os.path.join(BASE_DIR, 'server', 'data.db')
|
||
|
|
|
||
|
|
def add_column():
|
||
|
|
"""添加 customer_name 列"""
|
||
|
|
if not os.path.exists(DB_PATH):
|
||
|
|
print(f"错误: 数据库文件不存在: {DB_PATH}")
|
||
|
|
return
|
||
|
|
|
||
|
|
conn = sqlite3.connect(DB_PATH)
|
||
|
|
c = conn.cursor()
|
||
|
|
|
||
|
|
try:
|
||
|
|
# 尝试添加列
|
||
|
|
c.execute('ALTER TABLE customer_orders ADD COLUMN customer_name TEXT')
|
||
|
|
conn.commit()
|
||
|
|
print("成功添加 customer_name 列")
|
||
|
|
except Exception as e:
|
||
|
|
if 'duplicate column name' in str(e).lower():
|
||
|
|
print("customer_name 列已存在")
|
||
|
|
else:
|
||
|
|
print(f"添加列失败: {e}")
|
||
|
|
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
add_column()
|