newcom
parent
8fac444974
commit
813bead6e5
14
app.py
14
app.py
|
|
@ -32,6 +32,16 @@ CREATE TABLE IF NOT EXISTS installs (
|
||||||
""")
|
""")
|
||||||
conn.commit()
|
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):
|
class Folder(BaseModel):
|
||||||
name: str
|
name: str
|
||||||
|
|
@ -100,8 +110,10 @@ def get_installs():
|
||||||
@app.post("/api/install")
|
@app.post("/api/install")
|
||||||
def add_install(data: InstallData):
|
def add_install(data: InstallData):
|
||||||
install_time = data.install_time or datetime.datetime.now().isoformat()
|
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 (?, ?, ?, ?)",
|
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()
|
conn.commit()
|
||||||
return {"status": "success"}
|
return {"status": "success"}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue