from app.extensions import db
from app.models.task import Task

def success_state(task_id):
    task = Task.query.get(task_id)

    if not task:
        return

    task.status = "SUCCESS"
    task.status_message = "задача выполнена успешно"
    
    db.session.commit()

def failure_state(task_id, exception):
    task = Task.query.get(task_id)

    if not task:
        return

    task.status = "FAILURE"
    task.status_message = f"произошла ошибка {str(exception)}"
    
    db.session.commit()
    
def run_state(task_id, task_record_id):
    task = Task.query.get(task_record_id)

    if not task:
        return

    task.task_id = task_id
    task.status = "RUNNING"
    task.status_message = "задача запущена"
    
    db.session.commit()