fix
parent
51c58e22a0
commit
18a0d98190
13
app.py
13
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)
|
||||
)
|
||||
""")
|
||||
|
|
|
|||
Loading…
Reference in New Issue