log extract field

This commit is contained in:
debugtalk
2018-03-30 18:47:07 +08:00
parent 3eb3a0eb57
commit 251a0a84e7
+19 -4
View File
@@ -98,10 +98,21 @@ class ResponseObject(object):
def extract_field(self, field): def extract_field(self, field):
""" extract value from requests.Response. """ extract value from requests.Response.
""" """
if text_extractor_regexp_compile.match(field): msg = "extract field: {}".format(field)
return self._extract_field_with_regex(field)
else: try:
return self._extract_field_with_delimiter(field) if text_extractor_regexp_compile.match(field):
value = self._extract_field_with_regex(field)
else:
value = self._extract_field_with_delimiter(field)
msg += "\t=> {}".format(value)
logger.log_debug(msg)
except exception.ParseResponseError:
logger.log_error("failed to extract field: {}".format(field))
raise
return value
def extract_response(self, extractors): def extract_response(self, extractors):
""" extract value from requests.Response and store in OrderedDict. """ extract value from requests.Response and store in OrderedDict.
@@ -114,6 +125,10 @@ class ResponseObject(object):
] ]
@return (OrderDict) variable binds ordered dict @return (OrderDict) variable binds ordered dict
""" """
if not extractors:
return {}
logger.log_info("start to extract from response object.")
extracted_variables_mapping = OrderedDict() extracted_variables_mapping = OrderedDict()
extract_binds_order_dict = utils.convert_to_order_dict(extractors) extract_binds_order_dict = utils.convert_to_order_dict(extractors)