__init__.py

from ._base_dag import BaseDAG
from ._dag_list import DAG_LIST
from .l1 import CsstL1


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


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