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

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

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

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

    task.status = "FAILURE"
    task.status_message = f"произошла ошибка {str(exception)}"

def run_state(task_id, task_record_id):
    task = Task.query.get(task_record_id)

    task.task_id = task_id
    task.status = "RUNNING"
    task.status_message = "задача запущена"