mirror of
https://github.com/Cccc-owo/CheckInApp.git
synced 2026-06-17 05:56:29 +00:00
chore: remove useless migration srcipts
This commit is contained in:
@@ -1,153 +0,0 @@
|
||||
"""
|
||||
为 users 表添加 Token 过期提醒字段的迁移脚本
|
||||
|
||||
新增的列:
|
||||
- token_expiring_notified (BOOLEAN) - Token 即将过期提醒是否已发送(过期前30分钟内)
|
||||
- token_expired_notified (BOOLEAN) - Token 已过期提醒是否已发送(过期后30分钟内)
|
||||
|
||||
这两个字段用于实现两次 Token 过期提醒:
|
||||
1. 过期前 30 分钟内:提醒用户 Token 即将过期
|
||||
2. 过期后 30 分钟内:提醒用户 Token 已过期,需要刷新
|
||||
|
||||
运行方式:
|
||||
python backend/scripts/migrate_add_token_expiry_notified.py
|
||||
或
|
||||
venv/Scripts/python.exe backend/scripts/migrate_add_token_expiry_notified.py
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
|
||||
# 添加项目根目录到 Python 路径
|
||||
project_root = Path(__file__).resolve().parent.parent.parent
|
||||
sys.path.insert(0, str(project_root))
|
||||
|
||||
from sqlalchemy import text, inspect
|
||||
from backend.models.database import engine
|
||||
|
||||
|
||||
def migrate():
|
||||
"""执行迁移:添加 token_expiring_notified 和 token_expired_notified 列"""
|
||||
print("开始迁移:为 users 表添加 Token 过期提醒字段...")
|
||||
print("=" * 60)
|
||||
|
||||
with engine.connect() as conn:
|
||||
# 检查表结构
|
||||
inspector = inspect(engine)
|
||||
columns = [col['name'] for col in inspector.get_columns('users')]
|
||||
|
||||
print(f"\n当前表列: {', '.join(columns)}")
|
||||
|
||||
# 检查是否已存在新字段
|
||||
has_expiring = 'token_expiring_notified' in columns
|
||||
has_expired = 'token_expired_notified' in columns
|
||||
|
||||
if has_expiring and has_expired:
|
||||
print("\n[OK] Token 过期提醒字段已存在,跳过迁移")
|
||||
return
|
||||
|
||||
print(f"\n需要添加的列:")
|
||||
print(f" - token_expiring_notified (BOOLEAN, DEFAULT False)")
|
||||
print(f" - token_expired_notified (BOOLEAN, DEFAULT False)")
|
||||
|
||||
# SQLite 不支持直接 ALTER TABLE ADD COLUMN with constraints,需要重建表
|
||||
# 步骤:
|
||||
# 1. 创建新表(包含两个新字段)
|
||||
# 2. 复制数据
|
||||
# 3. 删除旧表
|
||||
# 4. 重命名新表
|
||||
|
||||
print("\n正在重建表结构...")
|
||||
|
||||
# 1. 创建新表
|
||||
conn.execute(text("""
|
||||
CREATE TABLE users_new (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
jwt_sub VARCHAR(200) UNIQUE,
|
||||
alias VARCHAR(50) NOT NULL UNIQUE,
|
||||
email VARCHAR(100),
|
||||
password_hash VARCHAR(200),
|
||||
authorization TEXT,
|
||||
jwt_exp VARCHAR(20) DEFAULT '0',
|
||||
token_expiring_notified BOOLEAN NOT NULL DEFAULT 0,
|
||||
token_expired_notified BOOLEAN NOT NULL DEFAULT 0,
|
||||
role VARCHAR(20) DEFAULT 'user',
|
||||
is_approved BOOLEAN DEFAULT 0,
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
updated_at DATETIME
|
||||
)
|
||||
"""))
|
||||
print(" [OK] 创建新表结构")
|
||||
|
||||
# 2. 复制数据(为旧数据设置两个字段都为 False)
|
||||
conn.execute(text("""
|
||||
INSERT INTO users_new
|
||||
(id, jwt_sub, alias, email, password_hash, authorization, jwt_exp,
|
||||
token_expiring_notified, token_expired_notified,
|
||||
role, is_approved, created_at, updated_at)
|
||||
SELECT
|
||||
id, jwt_sub, alias, email, password_hash, authorization, jwt_exp,
|
||||
0, 0,
|
||||
role, is_approved, created_at, updated_at
|
||||
FROM users
|
||||
"""))
|
||||
print(" [OK] 复制数据到新表(所有用户的提醒字段默认为 False)")
|
||||
|
||||
# 3. 删除旧表
|
||||
conn.execute(text("DROP TABLE users"))
|
||||
print(" [OK] 删除旧表")
|
||||
|
||||
# 4. 重命名新表
|
||||
conn.execute(text("ALTER TABLE users_new RENAME TO users"))
|
||||
print(" [OK] 重命名新表")
|
||||
|
||||
# 5. 重建索引
|
||||
conn.execute(text("""
|
||||
CREATE INDEX ix_users_jwt_sub ON users(jwt_sub)
|
||||
"""))
|
||||
conn.execute(text("""
|
||||
CREATE INDEX ix_users_alias ON users(alias)
|
||||
"""))
|
||||
conn.execute(text("""
|
||||
CREATE INDEX ix_users_role ON users(role)
|
||||
"""))
|
||||
conn.execute(text("""
|
||||
CREATE INDEX ix_users_is_approved ON users(is_approved)
|
||||
"""))
|
||||
conn.execute(text("""
|
||||
CREATE INDEX ix_users_id ON users(id)
|
||||
"""))
|
||||
conn.execute(text("""
|
||||
CREATE INDEX ix_user_role_approved ON users(role, is_approved)
|
||||
"""))
|
||||
print(" [OK] 重建索引")
|
||||
|
||||
conn.commit()
|
||||
|
||||
print("\n[SUCCESS] 表结构迁移成功!")
|
||||
print("\n新的表结构:")
|
||||
inspector = inspect(engine)
|
||||
new_columns = [col['name'] for col in inspector.get_columns('users')]
|
||||
print(f" 列: {', '.join(new_columns)}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
migrate()
|
||||
print("\n" + "=" * 60)
|
||||
print("[完成] 迁移成功完成!")
|
||||
print("\n数据库已更新为新架构:")
|
||||
print(" - 添加了 token_expiring_notified 列(Token 即将过期提醒,过期前30分钟)")
|
||||
print(" - 添加了 token_expired_notified 列(Token 已过期提醒,过期后30分钟内)")
|
||||
print(" - 所有现有用户的提醒字段默认为 False")
|
||||
print("\n提醒逻辑:")
|
||||
print(" 1. 过期前 30 分钟内:发送\"即将过期\"邮件,设置 token_expiring_notified = True")
|
||||
print(" 2. 过期后 30 分钟内:发送\"已过期\"邮件,设置 token_expired_notified = True")
|
||||
print(" 3. 用户刷新 Token 后:重置两个字段为 False")
|
||||
print("=" * 60)
|
||||
except Exception as e:
|
||||
print(f"\n[ERROR] 迁移失败: {e}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
sys.exit(1)
|
||||
@@ -1,135 +0,0 @@
|
||||
"""
|
||||
删除 users 表中 registered_ip 列的迁移脚本
|
||||
|
||||
删除的列:
|
||||
- registered_ip (VARCHAR) - 注册IP地址,不再需要
|
||||
|
||||
新架构中移除该字段以保护用户隐私。
|
||||
|
||||
运行方式:
|
||||
python backend/scripts/migrate_remove_registered_ip.py
|
||||
或
|
||||
venv/Scripts/python.exe backend/scripts/migrate_remove_registered_ip.py
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
|
||||
# 添加项目根目录到 Python 路径
|
||||
project_root = Path(__file__).resolve().parent.parent.parent
|
||||
sys.path.insert(0, str(project_root))
|
||||
|
||||
from sqlalchemy import text, inspect
|
||||
from backend.models.database import engine
|
||||
|
||||
|
||||
def migrate():
|
||||
"""执行迁移:删除 registered_ip 列"""
|
||||
print("开始迁移:删除 users 表中的 registered_ip 列...")
|
||||
print("=" * 60)
|
||||
|
||||
with engine.connect() as conn:
|
||||
# 检查表结构
|
||||
inspector = inspect(engine)
|
||||
columns = [col['name'] for col in inspector.get_columns('users')]
|
||||
|
||||
print(f"\n当前表列: {', '.join(columns)}")
|
||||
|
||||
if 'registered_ip' not in columns:
|
||||
print("\n[OK] registered_ip 列已被删除,跳过迁移")
|
||||
return
|
||||
|
||||
print(f"\n需要删除的列: registered_ip")
|
||||
|
||||
# SQLite 不支持直接 DROP COLUMN,需要重建表
|
||||
# 步骤:
|
||||
# 1. 创建新表(不包含 registered_ip)
|
||||
# 2. 复制数据
|
||||
# 3. 删除旧表
|
||||
# 4. 重命名新表
|
||||
|
||||
print("\n正在重建表结构...")
|
||||
|
||||
# 1. 创建新表
|
||||
conn.execute(text("""
|
||||
CREATE TABLE users_new (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
jwt_sub VARCHAR(200) UNIQUE,
|
||||
alias VARCHAR(50) NOT NULL UNIQUE,
|
||||
email VARCHAR(100),
|
||||
password_hash VARCHAR(200),
|
||||
authorization TEXT,
|
||||
jwt_exp VARCHAR(20) DEFAULT '0',
|
||||
role VARCHAR(20) DEFAULT 'user',
|
||||
is_approved BOOLEAN DEFAULT 0,
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
updated_at DATETIME
|
||||
)
|
||||
"""))
|
||||
print(" [OK] 创建新表结构")
|
||||
|
||||
# 2. 复制数据(不包含 registered_ip)
|
||||
conn.execute(text("""
|
||||
INSERT INTO users_new
|
||||
(id, jwt_sub, alias, email, password_hash, authorization, jwt_exp,
|
||||
role, is_approved, created_at, updated_at)
|
||||
SELECT
|
||||
id, jwt_sub, alias, email, password_hash, authorization, jwt_exp,
|
||||
role, is_approved, created_at, updated_at
|
||||
FROM users
|
||||
"""))
|
||||
print(" [OK] 复制数据到新表")
|
||||
|
||||
# 3. 删除旧表
|
||||
conn.execute(text("DROP TABLE users"))
|
||||
print(" [OK] 删除旧表")
|
||||
|
||||
# 4. 重命名新表
|
||||
conn.execute(text("ALTER TABLE users_new RENAME TO users"))
|
||||
print(" [OK] 重命名新表")
|
||||
|
||||
# 5. 重建索引
|
||||
conn.execute(text("""
|
||||
CREATE INDEX ix_users_jwt_sub ON users(jwt_sub)
|
||||
"""))
|
||||
conn.execute(text("""
|
||||
CREATE INDEX ix_users_alias ON users(alias)
|
||||
"""))
|
||||
conn.execute(text("""
|
||||
CREATE INDEX ix_users_role ON users(role)
|
||||
"""))
|
||||
conn.execute(text("""
|
||||
CREATE INDEX ix_users_is_approved ON users(is_approved)
|
||||
"""))
|
||||
conn.execute(text("""
|
||||
CREATE INDEX ix_users_id ON users(id)
|
||||
"""))
|
||||
conn.execute(text("""
|
||||
CREATE INDEX ix_user_role_approved ON users(role, is_approved)
|
||||
"""))
|
||||
print(" [OK] 重建索引")
|
||||
|
||||
conn.commit()
|
||||
|
||||
print("\n[SUCCESS] 表结构迁移成功!")
|
||||
print("\n新的表结构:")
|
||||
inspector = inspect(engine)
|
||||
new_columns = [col['name'] for col in inspector.get_columns('users')]
|
||||
print(f" 列: {', '.join(new_columns)}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
migrate()
|
||||
print("\n" + "=" * 60)
|
||||
print("[完成] 迁移成功完成!")
|
||||
print("\n数据库已更新为新架构:")
|
||||
print(" - 删除了 registered_ip 列(保护用户隐私)")
|
||||
print(" - 保留了所有其他用户数据")
|
||||
print("=" * 60)
|
||||
except Exception as e:
|
||||
print(f"\n[ERROR] 迁移失败: {e}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
sys.exit(1)
|
||||
Reference in New Issue
Block a user