46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
|
#!/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()
|