Satur@it-depot.ru 2025-03-05 09:48:00 +03:00
parent 51c58e22a0
commit 18a0d98190
1 changed files with 10 additions and 3 deletions

13
app.py
View File

@ -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)
)
""")