test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-12-03 04:19:18 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

[skip ci] Update codes of log output (#12329)

Browse Source 
Signed-off-by: wangting0128 <ting.wang@zilliz.com>
This commit is contained in:
wt 2021-11-29 19:50:06 +08:00 committed by GitHub
parent 3868649878
commit 3c04625381
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/python_client/utils/api_request.py
View File

@ -2,8 +2,10 @@ import traceback
import os
from utils.util_log import test_log as log
enable_traceback = os.getenv('ENABLE_TRACEBACK', "True")
log.info(f"enable_traceback:{enable_traceback}")
# enable_traceback = os.getenv('ENABLE_TRACEBACK', "True")
# log.info(f"enable_traceback:{enable_traceback}")
class Error:
def __init__(self, error):
self.code = getattr(error, 'code', -1)
@ -18,7 +20,8 @@ def api_request_catch():
def inner_wrapper(*args, **kwargs):
try:
res = func(*args, **kwargs)
if enable_traceback == "True":
# if enable_traceback == "True":
if kwargs.get("enable_traceback", True):
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)
@ -27,7 +30,8 @@ def api_request_catch():
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
if enable_traceback == "True":
# if enable_traceback == "True":
if kwargs.get("enable_traceback", True):
log.error(traceback.format_exc())
log.error("(api_response) : %s" % log_e)
return Error(e), False