main
Satur@it-depot.ru 2025-03-04 18:22:52 +03:00
parent 8fac444974
commit 813bead6e5
1 changed files with 13 additions and 1 deletions

14
app.py
View File

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