diff --git a/csst_dfs_client/level1.py b/csst_dfs_client/level1.py index 58c8195f71b10473cd523347f112bb71b800f8ae..7197c7e5a14008dcc0a2fdffae511e4b2ce7bbd3 100644 --- a/csst_dfs_client/level1.py +++ b/csst_dfs_client/level1.py @@ -154,13 +154,13 @@ def update_prc_status(level1_id: str, file_type: str, prc_status: int, batch_id: def write(local_file: Union[IO, str], module_id: Literal['MSC', 'IFS', 'MCI', 'HSTDM', 'CPIC'], - level0_id: Optional[str], - level1_id: Optional[str], + level1_id: str, file_type: str, file_name: str, pipeline_id: str, pmapname: str, build: int, + level0_id: Optional[str] = None, dataset: str = constants.DEFAULT_DATASET, batch_id: str = constants.DEFAULT_BATCH_ID, **extra_kwargs) -> Result: @@ -171,7 +171,7 @@ def write(local_file: Union[IO, str], local_file (Union[IO, str]): 文件路径或文件对象 module_id ['MSC', 'IFS', 'MCI', 'HSTDM', 'CPIC']其中一个,代表: 模块ID level0_id (Optional[str]): 0级数据的ID默认为 None - level1_id (Optional[str]): 1级数据的ID默认为 None + level1_id (str): 1级数据的ID file_type (str): 文件类型 file_name (str): 1级数据文件名 pipeline_id (str): 管线ID diff --git a/csst_dfs_client/level2.py b/csst_dfs_client/level2.py index f5b3790a822685afc97ea5cd9c09fdcaf7497072..7d0b06411cf5a66a9ba8790920bfaf224246b422 100644 --- a/csst_dfs_client/level2.py +++ b/csst_dfs_client/level2.py @@ -147,14 +147,14 @@ def update_prc_status_by_file_name(file_name: str, prc_status: int, batch_id: st def write(local_file: Union[IO, str], module_id: Literal['MSC', 'IFS', 'MCI', 'HSTDM', 'CPIC'], - level0_id: Optional[str], - level1_id: Optional[str], - level2_id: Optional[str], - brick_id: Optional[int], + level2_id: str, data_type: str, file_name: str, pipeline_id: str, build: int, + level0_id: Optional[str] = None, + level1_id: Optional[str] = None, + brick_id: Optional[int] = 0, dataset: str = constants.DEFAULT_DATASET, batch_id: str = constants.DEFAULT_BATCH_ID, **extra_kwargs) -> Result: @@ -164,14 +164,14 @@ def write(local_file: Union[IO, str], Args: local_file (Union[IO, str]): 文件路径 或 文件对象 module_id ['MSC', 'IFS', 'MCI', 'HSTDM', 'CPIC']其中一个,代表: 模块ID - level0_id (Optional[str]): 0级数据的ID默认为 None - level1_id (Optional[str]): 1级数据的ID默认为 None - level2_id (Optional[str]): 2级数据的ID默认为 None - brick_id (Optional[int]): 天区的ID默认为 None + level2_id (str): 2级数据的ID data_type (str): 数据类型,如'csst-msc-l2-mbi-cat' file_name (str): 2级数据文件名 pipeline_id (str): 管线ID build (int): 构建号 + level0_id (Optional[str]): 0级数据的ID默认为 None + level1_id (Optional[str]): 1级数据的ID默认为 None + brick_id (Optional[int]): 天区的ID默认为 0 dataset (Optional[str], optional): 数据集名称. Defaults to None. batch_id (Optional[str], optional): 最后一次成功的批次ID. Defaults to None. **kwargs: 额外的关键字参数,这些参数将传递给DFS