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)
|
conn = sqlite3.connect("/db/rustdesk.db", check_same_thread=False)
|
||||||
cursor = conn.cursor()
|
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("""
|
cursor.execute("""
|
||||||
CREATE TABLE IF NOT EXISTS folders (
|
CREATE TABLE IF NOT EXISTS folders (
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
|
@ -23,11 +30,11 @@ CREATE TABLE IF NOT EXISTS folders (
|
||||||
cursor.execute("""
|
cursor.execute("""
|
||||||
CREATE TABLE IF NOT EXISTS installs (
|
CREATE TABLE IF NOT EXISTS installs (
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
rust_id TEXT, -- Оставляем название поля для совместимости, но оно теперь универсальное (ID подключения)
|
rust_id TEXT,
|
||||||
computer_name TEXT,
|
computer_name TEXT,
|
||||||
install_time TEXT,
|
install_time TEXT,
|
||||||
folder_id INTEGER,
|
folder_id INTEGER,
|
||||||
protocol TEXT DEFAULT 'rustdesk', -- Новое поле с протоколом, по умолчанию RustDesk
|
protocol TEXT DEFAULT 'rustdesk',
|
||||||
FOREIGN KEY (folder_id) REFERENCES folders(id)
|
FOREIGN KEY (folder_id) REFERENCES folders(id)
|
||||||
)
|
)
|
||||||
""")
|
""")
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue