feat(client): add optional dataset parameter to count_by_dataset

csst_dfs_client/level0.py

@@ -392,12 +392,15 @@ def delete(dataset: str, data_model: str = 'raw') -> Result:
     """
     return request.delete("/api/level0/delete", {"dataset": dataset, "data_model": data_model})
 
-def count_by_dataset() -> Result:
+def count_by_dataset(dataset: Optional[str] = None) -> Result:
     """
     查询0级数据按数据集分组的数量
 
+    Args:
+        dataset (str): 数据集
     Returns:
         Result: 成功后，Result.data为数据列表，失败message为失败原因
 
     """
-    return request.get("/api/level0/count_by_dataset")
\ No newline at end of file
+    params = {"dataset": dataset}
+    return request.post("/api/level0/count_by_dataset", params)

csst_dfs_client/level1.py

@@ -360,12 +360,16 @@ def delete(dataset: str, batch_id: str, data_model: str) -> Result:
     """
     return request.delete("/api/level1/delete", {"dataset": dataset, "batch_id": batch_id, "data_model": data_model})
 
-def count_by_dataset() -> Result:
+def count_by_dataset(dataset: Optional[str] = None) -> Result:
     """
     查询0级数据按数据集分组的数量
 
+    Args:
+        dataset (str): 数据集
+
     Returns:
         Result: 成功后，Result.data为数据列表，失败message为失败原因
 
     """
-    return request.get("/api/level1/count_by_dataset")
+    params = {"dataset": dataset}
+    return request.post("/api/level1/count_by_dataset", params)

tests/test_level0.py

@@ -74,7 +74,7 @@ class Level0TestCase(unittest.TestCase):
     #     print(result)
     #     self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)    
     def test_count_by_dataset(self):
-        result = level0.count_by_dataset()
+        result = level0.count_by_dataset(dataset="msc-v093")
         print(result)
         self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)    
         self.assertIsNotNone(result.data, "error message: " + result.message)

tests/test_level1.py

@@ -81,7 +81,7 @@ class Level1TestCase(unittest.TestCase):
     #     self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)
     
     def test_count_by_dataset(self):
-        result = level1.count_by_dataset()
+        result = level1.count_by_dataset(dataset="msc-v093")
         print(result)
         self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)    
         self.assertIsNotNone(result.data, "error message: " + result.message)
\ No newline at end of file