newcom
parent
8fac444974
commit
813bead6e5
14
app.py
14
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"}
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue