__init__.py

from ._base_dag import BaseDAG
from ._dag_list import DAG_LIST
from .dags import GeneralDAGViaObsid, GeneralDAGViaObsgroup
from .dispatcher import Dispatcher


class CsstDAGs(dict):
    """
    A class to manage all DAGs.
    """

    dag_list = {
        "csst-msc-l1-qc0": GeneralDAGViaObsid(
            dag_group="msc-l1", dag="csst-msc-l1-qc0", use_detector=True
        ),
        "csst-msc-l1-mbi": GeneralDAGViaObsid(
            dag_group="msc-l1", dag="csst-msc-l1-mbi", use_detector=True
        ),
        "csst-msc-l1-ast": GeneralDAGViaObsid(
            dag_group="msc-l1", dag="csst-msc-l1-ast", use_detector=True
        ),
        "csst-msc-l1-sls": GeneralDAGViaObsid(
            dag_group="msc-l1", dag="csst-msc-l1-sls", use_detector=True
        ),
        "csst-msc-l1-ooc": GeneralDAGViaObsgroup(
            dag_group="msc-l1-ooc", dag="csst-msc-l1-ooc"
        ),
        "csst-cpic-l1": GeneralDAGViaObsid(
            dag_group="cpic-l1", dag="csst-cpic-l1", use_detector=True
        ),
        "csst-cpic-l1-qc0": GeneralDAGViaObsid(
            dag_group="cpic-l1", dag="csst-cpic-l1-qc0", use_detector=True
        ),
    }

    def __init__(self):
        super().__init__()  # 初始化空字典
        self.update(self.dag_list)  # 先添加默认键值对
        # self.update(*args, **kwargs)  # 用户传入值覆盖默认值

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