Loading csst_dfs_client/level2.py +7 −1 Original line number Diff line number Diff line import os import pickle from typing import Optional, Tuple, Literal from .common import request, Result Loading Loading @@ -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) Loading
csst_dfs_client/level2.py +7 −1 Original line number Diff line number Diff line import os import pickle from typing import Optional, Tuple, Literal from .common import request, Result Loading Loading @@ -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)