IntelligentRailwayCosting/SourceCode/IntelligentRailwayCosting/app/executor/__init__.py

from .task_runner import TaskRunner
from .task_sender import TaskSender
from core.dtos import ProjectTaskDto, ProjectQuotaDto


def init():
    TaskRunner.init()


def run_task(task: ProjectTaskDto):
    return TaskRunner.run_task(task)


def cancel_task(task: ProjectTaskDto):
    return TaskRunner.cancel(task)


def project_is_running(project_id: str):
    return TaskRunner.get_project_running_state(project_id)


def send_task(task: ProjectTaskDto):
    return TaskSender().send_task(task)


def send_quota(quota: ProjectQuotaDto):
    return TaskSender().send_quota(quota)


__all__ = [
    "init",
    "run_task",
    "cancel_task",
    "send_task",
    "send_quota",
    "project_is_running",
]