diff --git a/app.py b/app.py index 8badfb8..d4a834a 100644 --- a/app.py +++ b/app.py @@ -11,7 +11,14 @@ app = FastAPI() conn = sqlite3.connect("/db/rustdesk.db", check_same_thread=False) cursor = conn.cursor() -# Создаем таблицы (добавляем поле protocol) +# Проверяем и обновляем структуру таблицы installs +cursor.execute("PRAGMA table_info(installs)") +columns = [row[1] for row in cursor.fetchall()] +if 'protocol' not in columns: + cursor.execute("ALTER TABLE installs ADD COLUMN protocol TEXT DEFAULT 'rustdesk'") + conn.commit() + +# Создаем таблицы (добавляем поле protocol, если таблицы нет) cursor.execute(""" CREATE TABLE IF NOT EXISTS folders ( id INTEGER PRIMARY KEY AUTOINCREMENT, @@ -23,11 +30,11 @@ CREATE TABLE IF NOT EXISTS folders ( cursor.execute(""" CREATE TABLE IF NOT EXISTS installs ( id INTEGER PRIMARY KEY AUTOINCREMENT, - rust_id TEXT, -- Оставляем название поля для совместимости, но оно теперь универсальное (ID подключения) + rust_id TEXT, computer_name TEXT, install_time TEXT, folder_id INTEGER, - protocol TEXT DEFAULT 'rustdesk', -- Новое поле с протоколом, по умолчанию RustDesk + protocol TEXT DEFAULT 'rustdesk', FOREIGN KEY (folder_id) REFERENCES folders(id) ) """)