feat(level0): 更新QC状态函数及测试

csst_dfs_client/level0.py

@@ -5,7 +5,8 @@ from .common import request, Result, utils, constants
 DateTimeTuple = Tuple[str, str]
 
 def find(
-        instrument: Literal['MSC', 'IFS', 'MCI', 'HSTDM', 'CPIC'], 
+        instrument: Literal['MSC', 'IFS', 'MCI', 'HSTDM', 'CPIC'],
+        dataset: str, 
         obs_group: Optional[str] = None,
         obs_id: Optional[str] = None,
         detector: Optional[str] = None,
@@ -13,14 +14,13 @@ def find(
         filter: Optional[str] = None,
         obs_time: Optional[DateTimeTuple] = None,
         create_time: Optional[DateTimeTuple] = None,
-        qc0_status: Optional[int] = None,
+        qc_status: Optional[int] = None,
         prc_status: Optional[int] = None,
         file_name: Optional[str] = None,
         ra_obj: Optional[int] = None,
         dec_obj: Optional[int] = None,
         radius: Optional[float] = None,
         object_name: Optional[str] = None,
-        dataset: str = constants.DEFAULT_DATASET,
         page: int = 1,
         limit: int = 0) -> Result:
     """
@@ -35,7 +35,7 @@ def find(
         filter (Optional[str], optional): 滤光片. Defaults to None.
         obs_time (Optional[DateTimeTuple], optional): 观测时间范围. Defaults to None.
         create_time (Optional[DateTimeTuple], optional): 创建时间范围. Defaults to None.
-        qc0_status (Optional[int], optional): QC0状态. Defaults to None.
+        qc_status (Optional[int], optional): QC0状态. Defaults to None.
         prc_status (Optional[int], optional): 处理状态. Defaults to None.
         file_name (Optional[str], optional): 文件名. Defaults to None.
         ra_obj (Optional[int], optional): 目标赤经. Defaults to None.
@@ -58,7 +58,7 @@ def find(
         'detector': detector,
         'obs_type': obs_type,
         'filter': filter,
-        'qc0_status': qc0_status,
+        'qc_status': qc_status,
         'prc_status': prc_status,
         'file_name': file_name,
         'ra_obj': ra_obj,
@@ -111,32 +111,32 @@ def find_by_level0_id(level0_id: str) -> Result:
     """
     return request.get(f"/api/level0/{level0_id}")
 
-def update_qc0_status(level0_id: str, qc0_status: int, dataset: str = constants.DEFAULT_DATASET) -> Result:
+def update_qc_status(level0_id: str, qc_status: int, dataset: str) -> Result:
     """
-    更新0级数据的QC0状态
+    更新0级数据的QC状态
     
     Args:
         level0_id (str): 0级数据的ID
-        qc0_status (int): QC0状态
+        qc_status (int): QC0状态
         dataset (str): 数据集名称
     
     Returns:
         Result: 更新结果
     """
-    return request.put(f"/api/level0/qc0_status/{level0_id}", {'qc0_status': qc0_status, 'dataset': dataset})
+    return request.put(f"/api/level0/qc_status/{level0_id}", {'qc_status': qc_status, 'dataset': dataset})
 
-def update_qc0_status_by_ids(ids: List[str], qc0_status: int) -> Result:
+def update_qc_status_by_ids(ids: List[str], qc_status: int) -> Result:
     """
     根据内部_id，批量更新0级数据的QC0状态
     
     Args:
         ids (List[str]): 内部_id列表
-        qc0_status (int): QC0状态
+        qc_status (int): QC0状态
     
     Returns:
         Result: 更新结果
     """
-    return request.put("/api/level0/qc0_status/batch/update", {'qc0_status': qc0_status, 'ids': ids})
+    return request.put("/api/level0/qc_status/batch/update", {'qc_status': qc_status, 'ids': ids})
 
 def update_prc_status(level0_id: str, dag_run: str, prc_status: int, dataset: str = constants.DEFAULT_DATASET) -> Result:
     """

tests/test_level0.py

@@ -29,13 +29,13 @@ class Level0TestCase(unittest.TestCase):
         self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)
         self.assertIsNotNone(result.data, "error message: " + result.message)
 
-    def test_update_qc0_status(self):
-        result = level0.update_qc0_status(level0_id = "1010910015799127", qc0_status=1)
+    def test_update_qc_status(self):
+        result = level0.update_qc_status(level0_id = "1010910015799127", qc_status=1)
         print(result)
         self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)
 
-    def test_update_qc0_status_by_ids(self):
-        result = level0.update_qc0_status_by_ids(ids = ["676ac74a530b47ca41568858"], qc0_status=4)
+    def test_update_qc_status_by_ids(self):
+        result = level0.update_qc_status_by_ids(ids = ["676ac74a530b47ca41568858"], qc_status=4)
         print(result)
         self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)