GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-25 10:20:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: query json content

Browse Source
This commit is contained in:
debugtalk
2018-07-25 17:32:12 +08:00
parent fe9d5c497b
commit dee51b6682
7 changed files with 32 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
httprunner/utils.py
View File

@@ -17,7 +17,7 @@ from datetime import datetime
import yaml
from httprunner import exceptions, logger
from httprunner.compat import OrderedDict, is_py2, is_py3
from httprunner.compat import OrderedDict, is_py2, is_py3, str
from requests.structures import CaseInsensitiveDict
SECRET_KEY = "DebugTalk"
@@ -186,20 +186,28 @@ def query_json(json_content, query, delimiter='.'):
}
@param (str) query
"person.name.first_name" => "Leo"
"person.name.first_name.0" => "L"
"person.cities.0" => "Guangzhou"
@return queried result
"""
raise_flag = False
response_body = u"from: {}\n".format(json_content)
try:
for key in query.split(delimiter):
if isinstance(json_content, list):
if isinstance(json_content, (list, str)):
json_content = json_content[int(key)]
elif isinstance(json_content, dict):
json_content = json_content[key]
else:
raise exceptions.ParseResponseFailure(
"response content is in text format! failed to query key {}!".format(key))
raise_flag = True
except (KeyError, ValueError, IndexError):
raise exceptions.ParseResponseFailure("failed to query json when extracting response!")
raise_flag = True
if raise_flag:
err_msg = u"ExtractFailure: Failed to extract! => {}\n".format(query)
err_msg += response_body
logger.log_error(err_msg)
raise exceptions.ExtractFailure(err_msg)
return json_content