代码重构,删除无用文件
This commit is contained in:
@@ -1,28 +0,0 @@
|
||||
from flask_prompt_master import create_app, db
|
||||
from flask_prompt_master.models import PromptTemplate
|
||||
import pymysql
|
||||
|
||||
def migrate_to_mysql():
|
||||
"""迁移数据到MySQL"""
|
||||
app = create_app()
|
||||
|
||||
with app.app_context():
|
||||
# 创建所有表
|
||||
db.create_all()
|
||||
|
||||
try:
|
||||
# 检查是否已有数据
|
||||
if PromptTemplate.query.first() is None:
|
||||
# 执行init_db中的初始化
|
||||
from flask_prompt_master.init_db import init_db
|
||||
init_db()
|
||||
print("模板数据迁移完成!")
|
||||
else:
|
||||
print("数据库中已存在模板数据,跳过迁移。")
|
||||
|
||||
except Exception as e:
|
||||
print(f"迁移过程中出错: {str(e)}")
|
||||
db.session.rollback()
|
||||
|
||||
if __name__ == '__main__':
|
||||
migrate_to_mysql()
|
||||
@@ -1,66 +0,0 @@
|
||||
from flask_migrate import Migrate
|
||||
from flask_prompt_master import create_app, db
|
||||
import pymysql
|
||||
|
||||
def upgrade_database():
|
||||
"""添加微信相关字段并修改字段约束"""
|
||||
try:
|
||||
conn = pymysql.connect(
|
||||
host='localhost',
|
||||
user='root',
|
||||
password='123456',
|
||||
database='food_db',
|
||||
charset='utf8mb4'
|
||||
)
|
||||
cursor = conn.cursor()
|
||||
|
||||
# 修改现有字段约束
|
||||
modify_statements = [
|
||||
"ALTER TABLE user MODIFY mobile VARCHAR(20) NULL",
|
||||
"ALTER TABLE user MODIFY email VARCHAR(100) NULL",
|
||||
"ALTER TABLE user MODIFY avatar VARCHAR(64) NULL"
|
||||
]
|
||||
|
||||
print("\n=== 修改字段约束 ===")
|
||||
for statement in modify_statements:
|
||||
try:
|
||||
cursor.execute(statement)
|
||||
print(f"执行成功: {statement}")
|
||||
except Exception as e:
|
||||
print(f"执行失败: {statement}")
|
||||
print(f"错误: {str(e)}")
|
||||
|
||||
# 添加新字段
|
||||
add_statements = [
|
||||
"ALTER TABLE user ADD COLUMN IF NOT EXISTS openid VARCHAR(64) UNIQUE",
|
||||
"ALTER TABLE user ADD COLUMN IF NOT EXISTS session_key VARCHAR(64)",
|
||||
"ALTER TABLE user ADD COLUMN IF NOT EXISTS unionid VARCHAR(64) UNIQUE",
|
||||
"ALTER TABLE user ADD COLUMN IF NOT EXISTS wx_nickname VARCHAR(100)",
|
||||
"ALTER TABLE user ADD COLUMN IF NOT EXISTS wx_avatar VARCHAR(255)"
|
||||
]
|
||||
|
||||
print("\n=== 添加新字段 ===")
|
||||
for statement in add_statements:
|
||||
try:
|
||||
cursor.execute(statement)
|
||||
print(f"执行成功: {statement}")
|
||||
except Exception as e:
|
||||
print(f"执行失败: {statement}")
|
||||
print(f"错误: {str(e)}")
|
||||
|
||||
# 提交事务
|
||||
conn.commit()
|
||||
print("\n=== 数据库升级完成 ===")
|
||||
|
||||
except Exception as e:
|
||||
print(f"数据库升级失败: {str(e)}")
|
||||
if 'conn' in locals():
|
||||
conn.rollback()
|
||||
finally:
|
||||
if 'cursor' in locals():
|
||||
cursor.close()
|
||||
if 'conn' in locals():
|
||||
conn.close()
|
||||
|
||||
if __name__ == '__main__':
|
||||
upgrade_database()
|
||||
@@ -1,71 +0,0 @@
|
||||
import pymysql
|
||||
|
||||
def upgrade_database():
|
||||
"""创建微信用户相关表"""
|
||||
try:
|
||||
conn = pymysql.connect(
|
||||
host='localhost',
|
||||
user='root',
|
||||
password='123456',
|
||||
database='food_db',
|
||||
charset='utf8mb4'
|
||||
)
|
||||
cursor = conn.cursor()
|
||||
|
||||
# 创建微信用户表
|
||||
cursor.execute("""
|
||||
CREATE TABLE IF NOT EXISTS wx_user (
|
||||
id INT PRIMARY KEY AUTO_INCREMENT,
|
||||
openid VARCHAR(64) UNIQUE NOT NULL,
|
||||
session_key VARCHAR(64),
|
||||
unionid VARCHAR(64) UNIQUE,
|
||||
nickname VARCHAR(100),
|
||||
avatar_url VARCHAR(255),
|
||||
gender INT DEFAULT 0,
|
||||
country VARCHAR(50),
|
||||
province VARCHAR(50),
|
||||
city VARCHAR(50),
|
||||
language VARCHAR(20),
|
||||
phone VARCHAR(20),
|
||||
is_active BOOLEAN DEFAULT TRUE,
|
||||
last_login DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP
|
||||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
|
||||
""")
|
||||
print("创建wx_user表成功")
|
||||
|
||||
# 修改现有表添加外键
|
||||
alter_statements = [
|
||||
"ALTER TABLE prompt MODIFY user_id INT NULL",
|
||||
"ALTER TABLE prompt ADD COLUMN wx_user_id INT",
|
||||
"ALTER TABLE prompt ADD FOREIGN KEY (wx_user_id) REFERENCES wx_user(id)",
|
||||
|
||||
"ALTER TABLE feedback MODIFY user_id INT NULL",
|
||||
"ALTER TABLE feedback ADD COLUMN wx_user_id INT",
|
||||
"ALTER TABLE feedback ADD FOREIGN KEY (wx_user_id) REFERENCES wx_user(id)"
|
||||
]
|
||||
|
||||
for statement in alter_statements:
|
||||
try:
|
||||
cursor.execute(statement)
|
||||
print(f"执行成功: {statement}")
|
||||
except Exception as e:
|
||||
print(f"执行失败: {statement}")
|
||||
print(f"错误: {str(e)}")
|
||||
|
||||
conn.commit()
|
||||
print("\n=== 数据库升级完成 ===")
|
||||
|
||||
except Exception as e:
|
||||
print(f"数据库升级失败: {str(e)}")
|
||||
if 'conn' in locals():
|
||||
conn.rollback()
|
||||
finally:
|
||||
if 'cursor' in locals():
|
||||
cursor.close()
|
||||
if 'conn' in locals():
|
||||
conn.close()
|
||||
|
||||
if __name__ == '__main__':
|
||||
upgrade_database()
|
||||
Reference in New Issue
Block a user