refactor: extract cookie

This commit is contained in:
debugtalk
2018-07-25 11:47:10 +08:00
parent 22632d7110
commit 0b92b38143
2 changed files with 31 additions and 4 deletions

View File

@@ -84,13 +84,16 @@ class ResponseObject(object):
# cookies
elif top_query == "cookies":
cookies = self.cookies
cookies = self.cookies.get_dict()
if not sub_query:
# extract cookies
return cookies
try:
return cookies[sub_query]
except KeyError:
err_msg = u"Failed to extract attribute from cookies!\n"
err_msg += u"cookies: {}\n".format(cookies)
err_msg += u"attribute: {}".format(sub_query)
err_msg = u"ParamsError: Failed to extract cookie! => {}\n".format(field)
err_msg += u"response cookies: {}\n".format(cookies)
logger.log_error(err_msg)
raise exceptions.ParamsError(err_msg)