import arlo

database/ttaskqueue/curd.py

@@ -0,0 +1,21 @@
+from database.ttaskqueue.model import TTaskQueue
+
+
+def create_task(db, task: TTaskQueue):
+    db.add(task)
+    db.commit()
+    db.refresh(task)
+    return task
+
+def get_tasks_to_finish(db) -> list[TTaskQueue]:
+    return db.query(TTaskQueue).filter(TTaskQueue.finished == False).all()
+
+def finish_task(db, task_id: int):
+    task = db.query(TTaskQueue).filter(TTaskQueue.id == task_id).first()
+    if task:
+        task.finished = True
+        db.commit()
+        db.refresh(task)
+    return task
+
+