Commit 38d23cc5 authored by Wei Shoulin's avatar Wei Shoulin
Browse files

feat(level0): 增加运行 ID 参数并更新相关函数 

- 在 update_prc_status 函数中添加 run_id 参数
- 在 generate_prc_msg 函数中添加 level0_id 参数
- 更新相应测试用例以包含新增参数
parent d9ece63c

csst_dfs_client/level0.py

+6 −2
Original line number Diff line number Diff line
@@ -125,20 +125,21 @@ def update_qc0_status(level0_id: str, file_type: str, qc0_status: int, dataset: 
    """

    return request.put(f"/api/level0/qc0_status/{level0_id}", {'file_type': file_type, 'qc0_status': qc0_status, 'dataset': dataset})



def update_prc_status(level0_id: str, file_type: str, prc_status: int, dataset: str = constants.DEFAULT_DATASET) -> Result:

def update_prc_status(level0_id: str, file_type: str, run_id: str, prc_status: int, dataset: str = constants.DEFAULT_DATASET) -> Result:

    """

    更新0级数据的处理状态

    

    Args:

        level0_id (str): 0级数据的ID

        file_type (str): 文件类型

        run_id (str): 运行ID

        prc_status (int): 处理状态

        dataset (str): 数据集名称

    

    Returns:

        Result: 操作结果

    """

    return request.put(f"/api/level0/prc_status/{level0_id}", {'file_type': file_type, 'prc_status': prc_status, 'dataset': dataset})

    return request.put(f"/api/level0/prc_status/{level0_id}/{run_id}", {'file_type': file_type, 'prc_status': prc_status, 'dataset': dataset})



def write(local_file: str, 

        dataset: str = constants.DEFAULT_DATASET,
@@ -189,6 +190,7 @@ def write_cat(local_file: str, 
def generate_prc_msg(module_id: Literal['MSC', 'IFS', 'MCI', 'HSTDM', 'CPIC'], 

                    obs_id: str, 

                    detector_no: str,

                    level0_id: str,

                    dataset: str = constants.DEFAULT_DATASET,

                    batch_id: str = constants.DEFAULT_BATCH_ID) -> Result:

    """
@@ -198,6 +200,7 @@ def generate_prc_msg(module_id: Literal['MSC', 'IFS', 'MCI', 'HSTDM', 'CPIC'], 
        module_id (str): 模块ID

        obs_id (str): 观测ID

        detector_no (str): 探测器编号

        level0_id (str): 0级数据ID

        dataset (Optional[str], optional): 数据集名称..

        batch_id (Optional[str], optional): 批次ID. Defaults to 'auto'.
@@ -210,6 +213,7 @@ def generate_prc_msg(module_id: Literal['MSC', 'IFS', 'MCI', 'HSTDM', 'CPIC'], 
        'batch_id': batch_id,

        'obs_id': obs_id,

        'detector_no': detector_no,

        "level0_id": level0_id,

    }

    

    return request.put(f"/api/level0/prc/{module_id}", params)

tests/test_level0.py

+3 −1
Original line number Diff line number Diff line
@@ -36,7 +36,8 @@ class Level0TestCase(unittest.TestCase): 
        self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)



    def test_update_prc_status(self):

        result = level0.update_prc_status(level0_id = "1010910015799127", file_type='SCI', prc_status=3)

        result = level0.update_prc_status(level0_id = "1010910015799127", file_type='SCI',

                                        run_id="202411071002481234", prc_status=3)

        print(result)

        self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)
@@ -49,6 +50,7 @@ class Level0TestCase(unittest.TestCase): 
    def test_generate_prc_message(self):

        result = level0.generate_prc_msg(

            module_id='MSC',

            level0_id="1060940003452930",

            obs_id="10609400034529",

            detector_no="30")

        print(result)