removed unecessary checks, (clearly I need to review python ref model)

stix2.0
= 2018-04-13 16:06:31 -04:00
parent 4734711137
commit 6df23e7268
2 changed files with 4 additions and 13 deletions

View File

@ -71,8 +71,7 @@ class Filter(collections.namedtuple("Filter", ['property', 'op', 'value'])):
if isinstance(value, datetime):
# if value is a datetime obj, convert to str
dt_str = format_datetime(value)
value = dt_str # use temp variable to avoid deepcopy operation
value = format_datetime(value)
_check_filter_components(prop, op, value)
@ -90,11 +89,10 @@ class Filter(collections.namedtuple("Filter", ['property', 'op', 'value'])):
False otherwise.
"""
if isinstance(stix_obj_property, datetime):
# if a datetime obj, use str format before comparison
# if a datetime obj, convert to str format before comparison
# NOTE: this check seems like it should be done upstream
# but will put here for now
tmp = format_datetime(stix_obj_property)
stix_obj_property = tmp # use tmp variable to avoid deepcopy op
stix_obj_property = format_datetime(stix_obj_property)
if self.op == "=":
return stix_obj_property == self.value
@ -174,9 +172,6 @@ def _check_filter(filter_, stix_obj):
return True
return False
elif isinstance(stix_obj[prop], dict):
return _check_filter(sub_filter, stix_obj[prop])
else:
return _check_filter(sub_filter, stix_obj[prop])
@ -242,8 +237,6 @@ class FilterSet(object):
if f not in self._filters:
self._filters.append(f)
return
def remove(self, filters=None):
"""remove a Filter, list of Filters, or FilterSet from the FilterSet
@ -263,5 +256,3 @@ class FilterSet(object):
for f in filters:
self._filters.remove(f)
return

View File

@ -510,7 +510,7 @@ def test_datetime_filter_behavior():
filter_with_dt_obj = Filter("created", "=", parse_into_datetime("2016-02-14T00:00:00.000Z", "millisecond"))
filter_with_str = Filter("created", "=", "2016-02-14T00:00:00.000Z")
# check taht filter value is converted from datetime to str
# check that filter value is converted from datetime to str
assert isinstance(filter_with_dt_obj.value, str)
# compare datetime string to filter w/ datetime obj