HAHAHHA
This commit is contained in:
45
models/add_tg_web_app_data_sqlite.py
Normal file
45
models/add_tg_web_app_data_sqlite.py
Normal file
@@ -0,0 +1,45 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
为 SQLite haha.db 中 tg_phone_devices_copy1 表添加 tg_web_app_data 列(若不存在)。
|
||||
解决 踢设备.py 报错: no such column: t1.tg_web_app_data
|
||||
"""
|
||||
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
project_root = Path(__file__).resolve().parent.parent
|
||||
if str(project_root) not in sys.path:
|
||||
sys.path.insert(0, str(project_root))
|
||||
|
||||
from models.tg_phone_devices import db
|
||||
|
||||
TABLE = "tg_phone_devices_copy1"
|
||||
COLUMN = "tg_web_app_data"
|
||||
|
||||
|
||||
def column_exists_sqlite(cursor, table, column):
|
||||
cursor.execute("PRAGMA table_info({})".format(table))
|
||||
return any(row[1] == column for row in cursor.fetchall())
|
||||
|
||||
|
||||
def main():
|
||||
db.connect()
|
||||
try:
|
||||
with db.connection() as conn:
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("PRAGMA table_info({})".format(TABLE))
|
||||
if any(row[1] == COLUMN for row in cursor.fetchall()):
|
||||
print(f"表 {TABLE} 已存在列 {COLUMN},无需添加。")
|
||||
return
|
||||
cursor.execute("ALTER TABLE {} ADD COLUMN {} TEXT".format(TABLE, COLUMN))
|
||||
print(f"已在表 {TABLE}(SQLite)添加列 {COLUMN}。")
|
||||
finally:
|
||||
if not db.is_closed():
|
||||
try:
|
||||
db.close()
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user