22 lines
520 B
Python
22 lines
520 B
Python
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
|
|
|
|
|