from csst_dag.constants import MSC_MBI_CHIPID
from csst_dag import DFS
from csst_dfs_client import plan, level0
from ..dag._dag_list import DAG_LIST


class CsstMscL1Mbi:
    def __init__(self):
        pass

    @staticmethod
    def schedule(
        dag_id: str = "csst-msc-l1-mbi",
        dataset: str = "csst-msc-c9-25sqdeg-v3",
        obs_type: str = "WIDE",
        project_id: str = "none",
        batch_id: str | None = "default",
        **kwargs,
    ):
        pass


# Write plan to DFS
# result = plan.write_file("/nfs/simulation-collection/csst-msc-c9-25sqdeg-v3/plan.json")
# print(result)

# search plan
# # plan.find(instrument="MSC")
# res_plan = plan.find(
#     dataset="csst-msc-c9-25sqdeg-v3",
#     instrument="MSC",
#     obs_type="WIDE",
#     project_id="none",
# )
# print(res_plan.data[0])
# assert res_plan.success, res_plan.message

res_level0 = level0.find(
    dataset="csst-msc-c9-25sqdeg-v3",
    instrument="MSC",
    obs_type="WIDE",
    prc_status=-1024,
)

# process as there is
for rec in res_level0.data:
    print(rec)


print(res_level0.data[0])
print(res_level0.data[0].keys())


print(res_level0)
assert res_level0.success, res_level0.message


# 不对project的数据完整性进行检查，只对数据的存在性进行检查


# print(__file__)