__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]

    @staticmethod
    def get_all():
        return DAG_MAP