Commit f19bbfee authored by Wei Shoulin's avatar Wei Shoulin
Browse files

pkl

parent 18630b0f
Loading
Loading
Loading
Loading
+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

@@ -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)