update DB table
This commit is contained in:
@@ -138,6 +138,26 @@ async def list_queues(db: Session = Depends(database.get_db)):
|
||||
queues = db.query(models.Queue).order_by(models.Queue.created_at.desc()).all()
|
||||
return queues
|
||||
|
||||
@app.delete("/api/delete/{id}")
|
||||
async def delete_queue(id: str, db: Session = Depends(database.get_db)):
|
||||
# 1. Delete from database
|
||||
queue = db.query(models.Queue).filter(models.Queue.id == id).first()
|
||||
if queue:
|
||||
# Delete associated tasks first
|
||||
db.query(models.Task).filter(models.Task.queue_id == id).delete()
|
||||
db.delete(queue)
|
||||
db.commit()
|
||||
|
||||
# 2. Delete folder
|
||||
queue_dir = os.path.join(BASE_DATA_DIR, id)
|
||||
if os.path.exists(queue_dir):
|
||||
import shutil
|
||||
shutil.rmtree(queue_dir)
|
||||
|
||||
return {"id": id, "status": "Deleted"}
|
||||
|
||||
raise HTTPException(status_code=404, detail="ID not found")
|
||||
|
||||
@app.get("/")
|
||||
async def root():
|
||||
return FileResponse(os.path.join(static_dir, "index.html"))
|
||||
|
||||
Reference in New Issue
Block a user