__init__.py

from ._base_dag import BaseDAG
from ._dag_list import DAG_LIST
from .msc import CsstMscL1


DAG_MAP = {
    "csst-msc-l1-qc0": CsstMscL1(dag_id="csst-msc-l1-qc0"),
    "csst-msc-l1-mbi": CsstMscL1(dag_id="csst-msc-l1-mbi"),
    "csst-msc-l1-sls": CsstMscL1(dag_id="csst-msc-l1-sls"),
}


class CsstDAG:
    def __init__(self):
        pass

    @staticmethod
    def ls():
        print(DAG_MAP.keys())

    @staticmethod
    def get_dag(dag_id: str = ""):
        assert dag_id in DAG_LIST, f"{dag_id} not in DAG_LIST"
        return DAG_MAP[dag_id]

    def trigger(
        self,
        dag_id: str = "",
        **kwargs,
    ):
        pass

    def schedule(
        self,
        dag_id: str = "",
        **kwargs,
    ):
        pass

    def push(
        self,
        dag_id: str = "",
        **kwargs,
    ):
        pass


dags = CsstDAG()