From 813bead6e55ed3531f108e67de1abd6bddf14a95 Mon Sep 17 00:00:00 2001 From: "Satur@it-depot.ru" Date: Tue, 4 Mar 2025 18:22:52 +0300 Subject: [PATCH] newcom --- app.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/app.py b/app.py index 8e5ad0c..8dc1ddd 100644 --- a/app.py +++ b/app.py @@ -32,6 +32,16 @@ CREATE TABLE IF NOT EXISTS installs ( """) conn.commit() +# Проверяем/создаем папку "Несортированные" +cursor.execute("SELECT id FROM folders WHERE name = 'Несортированные'") +unsorted_folder = cursor.fetchone() +if not unsorted_folder: + cursor.execute("INSERT INTO folders (name) VALUES ('Несортированные')") + conn.commit() + unsorted_folder_id = cursor.lastrowid +else: + unsorted_folder_id = unsorted_folder[0] + # Модели данных class Folder(BaseModel): name: str @@ -100,8 +110,10 @@ def get_installs(): @app.post("/api/install") def add_install(data: InstallData): install_time = data.install_time or datetime.datetime.now().isoformat() + # Если folder_id не указан (например, от скрипта), используем "Несортированные" + folder_id = data.folder_id if data.folder_id is not None else unsorted_folder_id cursor.execute("INSERT INTO installs (rust_id, computer_name, install_time, folder_id) VALUES (?, ?, ?, ?)", - (data.rust_id, data.computer_name, install_time, data.folder_id)) + (data.rust_id, data.computer_name, install_time, folder_id)) conn.commit() return {"status": "success"}