pkl

csst_dfs_client/level2.py

 import os
+import pickle
 from typing import Optional, Tuple, Literal
 from .common import request, Result
 
@@ -174,4 +175,9 @@ def write(local_file: str,
     return request.post_file("/api/level2/file", local_file, params)
 
 def catalog_query(sql: str, limit: int = 0) -> Result:
-    return request.post("/api/level2/catalog/query", {'sql': sql, 'limit': limit})
+    datas = request.post("/api/level2/catalog/query", {'sql': sql, 'limit': limit})
+    if datas and isinstance(datas, Result):
+        return datas
+    records = pickle.loads(datas._content)
+    df, total_count = records['records'], records['totalCount']
+    return Result.ok_data(data = df).append("totalCount", total_count)