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