From 18a0d981902bae4058e108ba502b7e0b17648866 Mon Sep 17 00:00:00 2001 From: "Satur@it-depot.ru" Date: Wed, 5 Mar 2025 09:48:00 +0300 Subject: [PATCH] fix --- app.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) 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) ) """)