GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-31 21:39:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: add ExtendJSONEncoder to safely dump json data with python object, such as MultipartEncoder

Browse Source
This commit is contained in:
debugtalk
2020-06-07 22:26:57 +08:00
parent 5c06b91b62
commit fbf5f245c7
4 changed files with 28 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
httprunner/compat.py
View File

@@ -264,7 +264,7 @@ import time
import pytest
from loguru import logger
from httprunner.utils import get_platform
from httprunner.utils import get_platform, ExtendJSONEncoder
@pytest.fixture(scope="session", autouse=True)
@@ -316,7 +316,7 @@ def session_fixture(request):
os.makedirs(summary_dir, exist_ok=True)
with open(summary_path, "w", encoding="utf-8") as f:
json.dump(summary, f, indent=4)
json.dump(summary, f, indent=4, ensure_ascii=False, cls=ExtendJSONEncoder)
logger.info(f"generated task summary: {summary_path}")

11
httprunner/utils.py
View File

@@ -212,3 +212,14 @@ def ensure_file_path_valid(file_path: Text) -> Text:
new_file_path = os.path.join(os.getcwd(), f"{os.sep.join(path_names)}{file_suffix}")
return new_file_path
class ExtendJSONEncoder(json.JSONEncoder):
""" especially used to safely dump json data with python object, such as MultipartEncoder
"""
def default(self, obj):
try:
return super(ExtendJSONEncoder, self).default(obj)
except (UnicodeDecodeError, TypeError):
return repr(obj)