mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-09-06 16:07:01 +08:00
Remove downgrade methods for PostgreSQL and SQLite userid migration
Co-authored-by: jxxghp <jxxghp@live.cn>
This commit is contained in:
@@ -36,14 +36,7 @@ def downgrade() -> None:
|
|||||||
"""
|
"""
|
||||||
降级:将SiteUserData表的userid字段从String改回Integer
|
降级:将SiteUserData表的userid字段从String改回Integer
|
||||||
"""
|
"""
|
||||||
connection = op.get_bind()
|
pass
|
||||||
|
|
||||||
if settings.DB_TYPE.lower() == "postgresql":
|
|
||||||
# PostgreSQL数据库降级
|
|
||||||
downgrade_postgresql_userid(connection)
|
|
||||||
else:
|
|
||||||
# SQLite数据库降级
|
|
||||||
downgrade_sqlite_userid(connection)
|
|
||||||
|
|
||||||
|
|
||||||
def migrate_postgresql_userid(connection):
|
def migrate_postgresql_userid(connection):
|
||||||
@@ -150,109 +143,3 @@ def migrate_sqlite_userid(connection):
|
|||||||
raise
|
raise
|
||||||
|
|
||||||
|
|
||||||
def downgrade_postgresql_userid(connection):
|
|
||||||
"""
|
|
||||||
PostgreSQL数据库userid字段降级
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
logger.info("开始PostgreSQL数据库userid字段降级...")
|
|
||||||
|
|
||||||
# 1. 创建临时列
|
|
||||||
connection.execute(sa.text("""
|
|
||||||
ALTER TABLE siteuserdata
|
|
||||||
ADD COLUMN userid_old INTEGER
|
|
||||||
"""))
|
|
||||||
|
|
||||||
# 2. 将字符串转换为整数并复制到新列
|
|
||||||
connection.execute(sa.text("""
|
|
||||||
UPDATE siteuserdata
|
|
||||||
SET userid_old = CAST(userid AS INTEGER)
|
|
||||||
WHERE userid IS NOT NULL AND userid ~ '^[0-9]+$'
|
|
||||||
"""))
|
|
||||||
|
|
||||||
# 3. 删除旧列
|
|
||||||
connection.execute(sa.text("""
|
|
||||||
ALTER TABLE siteuserdata
|
|
||||||
DROP COLUMN userid
|
|
||||||
"""))
|
|
||||||
|
|
||||||
# 4. 重命名新列
|
|
||||||
connection.execute(sa.text("""
|
|
||||||
ALTER TABLE siteuserdata
|
|
||||||
RENAME COLUMN userid_old TO userid
|
|
||||||
"""))
|
|
||||||
|
|
||||||
logger.info("PostgreSQL数据库userid字段降级完成")
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
logger.error(f"PostgreSQL数据库userid字段降级失败: {e}")
|
|
||||||
raise
|
|
||||||
|
|
||||||
|
|
||||||
def downgrade_sqlite_userid(connection):
|
|
||||||
"""
|
|
||||||
SQLite数据库userid字段降级
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
logger.info("开始SQLite数据库userid字段降级...")
|
|
||||||
|
|
||||||
# SQLite不支持直接修改列类型,需要重建表
|
|
||||||
# 1. 创建新表结构
|
|
||||||
connection.execute(sa.text("""
|
|
||||||
CREATE TABLE siteuserdata_old (
|
|
||||||
id INTEGER PRIMARY KEY,
|
|
||||||
domain VARCHAR,
|
|
||||||
name VARCHAR,
|
|
||||||
username VARCHAR,
|
|
||||||
userid INTEGER,
|
|
||||||
user_level VARCHAR,
|
|
||||||
join_at VARCHAR,
|
|
||||||
bonus FLOAT DEFAULT 0,
|
|
||||||
upload FLOAT DEFAULT 0,
|
|
||||||
download FLOAT DEFAULT 0,
|
|
||||||
ratio FLOAT DEFAULT 0,
|
|
||||||
seeding FLOAT DEFAULT 0,
|
|
||||||
leeching FLOAT DEFAULT 0,
|
|
||||||
seeding_size FLOAT DEFAULT 0,
|
|
||||||
leeching_size FLOAT DEFAULT 0,
|
|
||||||
seeding_info JSON DEFAULT '{}',
|
|
||||||
message_unread INTEGER DEFAULT 0,
|
|
||||||
message_unread_contents JSON DEFAULT '[]',
|
|
||||||
err_msg VARCHAR,
|
|
||||||
updated_day VARCHAR,
|
|
||||||
updated_time VARCHAR
|
|
||||||
)
|
|
||||||
"""))
|
|
||||||
|
|
||||||
# 2. 复制数据,将字符串转换为整数(只转换数字字符串)
|
|
||||||
connection.execute(sa.text("""
|
|
||||||
INSERT INTO siteuserdata_old
|
|
||||||
SELECT
|
|
||||||
id, domain, name, username,
|
|
||||||
CASE
|
|
||||||
WHEN userid IS NULL THEN NULL
|
|
||||||
WHEN userid REGEXP '^[0-9]+$' THEN CAST(userid AS INTEGER)
|
|
||||||
ELSE NULL
|
|
||||||
END as userid,
|
|
||||||
user_level, join_at, bonus, upload, download, ratio,
|
|
||||||
seeding, leeching, seeding_size, leeching_size,
|
|
||||||
seeding_info, message_unread, message_unread_contents,
|
|
||||||
err_msg, updated_day, updated_time
|
|
||||||
FROM siteuserdata
|
|
||||||
"""))
|
|
||||||
|
|
||||||
# 3. 删除旧表
|
|
||||||
connection.execute(sa.text("DROP TABLE siteuserdata"))
|
|
||||||
|
|
||||||
# 4. 重命名新表
|
|
||||||
connection.execute(sa.text("ALTER TABLE siteuserdata_old RENAME TO siteuserdata"))
|
|
||||||
|
|
||||||
# 5. 重新创建索引
|
|
||||||
connection.execute(sa.text("CREATE INDEX ix_siteuserdata_domain ON siteuserdata (domain)"))
|
|
||||||
connection.execute(sa.text("CREATE INDEX ix_siteuserdata_updated_day ON siteuserdata (updated_day)"))
|
|
||||||
|
|
||||||
logger.info("SQLite数据库userid字段降级完成")
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
logger.error(f"SQLite数据库userid字段降级失败: {e}")
|
|
||||||
raise
|
|
||||||
Reference in New Issue
Block a user