milvus/tests/python_client/utils/api_request.py

48 lines
1.5 KiB
Python
Raw Normal View History

import traceback
from utils.util_log import test_log as log
class Error:
def __init__(self, error):
self.code = getattr(error, 'code', -1)
self.message = getattr(error, 'message', str(error))
log_row_length = 300
def api_request_catch():
def wrapper(func):
def inner_wrapper(*args, **kwargs):
try:
res = func(*args, **kwargs)
res_str = str(res)
log_res = res_str[0:log_row_length] + '......' if len(res_str) > log_row_length else res_str
log.debug("(api_response) : %s " % log_res)
return res, True
except Exception as e:
e_str = str(e)
log_e = e_str[0:log_row_length] + '......' if len(e_str) > log_row_length else e_str
log.error(traceback.format_exc())
log.error("(api_response) : %s" % log_e)
return Error(e), False
return inner_wrapper
return wrapper
@api_request_catch()
def api_request(_list, **kwargs):
if isinstance(_list, list):
func = _list[0]
if callable(func):
arg = []
if len(_list) > 1:
for a in _list[1:]:
arg.append(a)
arg_str = str(arg)
log_arg = arg_str[0:log_row_length] + '......' if len(arg_str) > log_row_length else arg_str
log.debug("(api_request) : [%s] args: %s, kwargs: %s" % (func.__qualname__, log_arg, str(kwargs)))
return func(*arg, **kwargs)
return False, False