jester1177/cloud-native-debug-env

1"""Task registry for the environment."""2 3 4from typing import Dict, Type5 6from server.tasks.base import BaseTask7from server.tasks.task_1_build_errors import DockerfileSyntaxTask8from server.tasks.task_2_docker_runtime import DockerfileRuntimeTask9from server.tasks.task_3_workflow_syntax import WorkflowSyntaxStructureTask10from server.tasks.task_4_workflow_secrets_permissions import WorkflowSecretsPermissionsTask11from server.tasks.task_5_ci_docker_integration import CIDockerIntegrationTask12from server.tasks.task_6_multi_stage_matrix import MultiStageMatrixTask13from server.tasks.k8s_pod import K8sPodTask14from server.tasks.k8s_networking import K8sNetworkingTask15from server.tasks.pipeline_build_deploy import PipelineBuildDeployTask16from server.tasks.pipeline_full import PipelineFullTask17 18TASK_REGISTRY: Dict[str, Type[BaseTask]] = {19    "dockerfile_syntax": DockerfileSyntaxTask,20    "dockerfile_runtime": DockerfileRuntimeTask,21    "workflow_syntax_structure": WorkflowSyntaxStructureTask,22    "workflow_secrets_permissions": WorkflowSecretsPermissionsTask,23    "ci_docker_integration": CIDockerIntegrationTask,24    "multi_stage_pipeline_matrix": MultiStageMatrixTask,25    "k8s_pod_failures": K8sPodTask,26    "k8s_networking": K8sNetworkingTask,27    "pipeline_build_deploy": PipelineBuildDeployTask,28    "pipeline_full_stack": PipelineFullTask,29}30 31 32def get_task(task_id: str) -> BaseTask:33    if task_id not in TASK_REGISTRY:34        raise ValueError(f"Unknown task: {task_id}")35    return TASK_REGISTRY[task_id]()36