Lan
73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
from tortoise import connections
|
|
|
|
|
|
def _need_upgrade(columns: list[tuple]) -> bool:
|
|
for column in columns:
|
|
# PRAGMA table_info 返回 (cid, name, type, notnull, dflt_value, pk)
|
|
if column[1] == "size":
|
|
column_type = (column[2] or "").upper()
|
|
return "BIGINT" not in column_type
|
|
return False
|
|
|
|
|
|
async def migrate():
|
|
conn = connections.get("default")
|
|
result = await conn.execute_query("PRAGMA table_info(filecodes)")
|
|
columns = result[1] if result and len(result) > 1 else []
|
|
|
|
if not columns or not _need_upgrade(columns):
|
|
return
|
|
|
|
await conn.execute_script(
|
|
"""
|
|
BEGIN;
|
|
CREATE TABLE IF NOT EXISTS filecodes_new
|
|
(
|
|
id INTEGER not null
|
|
primary key autoincrement,
|
|
code VARCHAR(255) not null
|
|
unique,
|
|
prefix VARCHAR(255) default '' not null,
|
|
suffix VARCHAR(255) default '' not null,
|
|
uuid_file_name VARCHAR(255),
|
|
file_path VARCHAR(255),
|
|
size BIGINT default 0 not null,
|
|
text TEXT,
|
|
expired_at TIMESTAMP,
|
|
expired_count INT default 0 not null,
|
|
used_count INT default 0 not null,
|
|
created_at TIMESTAMP default CURRENT_TIMESTAMP not null,
|
|
file_hash VARCHAR(128),
|
|
is_chunked BOOL default False not null,
|
|
upload_id VARCHAR(128)
|
|
);
|
|
|
|
INSERT INTO filecodes_new (id, code, prefix, suffix, uuid_file_name, file_path, size, text,
|
|
expired_at, expired_count, used_count, created_at, file_hash,
|
|
is_chunked, upload_id)
|
|
SELECT id,
|
|
code,
|
|
prefix,
|
|
suffix,
|
|
uuid_file_name,
|
|
file_path,
|
|
size,
|
|
text,
|
|
expired_at,
|
|
expired_count,
|
|
used_count,
|
|
created_at,
|
|
file_hash,
|
|
is_chunked,
|
|
upload_id
|
|
FROM filecodes;
|
|
|
|
DROP TABLE filecodes;
|
|
ALTER TABLE filecodes_new
|
|
RENAME TO filecodes;
|
|
CREATE INDEX IF NOT EXISTS idx_filecodes_code_1c7ee7
|
|
on filecodes (code);
|
|
COMMIT;
|
|
"""
|
|
)
|