Merge pull request #460 from emmanvg/459-issue

TAXIICollectionSource custom object handling fix
pull/1/head
Chris Lenk 2020-09-30 09:55:52 -04:00 committed by GitHub
commit edfcb1641a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -210,7 +210,7 @@ class TAXIICollectionSource(DataSource):
if len(stix_obj): if len(stix_obj):
stix_obj = parse(stix_obj[0], allow_custom=self.allow_custom, version=version) stix_obj = parse(stix_obj[0], allow_custom=self.allow_custom, version=version)
if stix_obj.id != stix_id: if stix_obj['id'] != stix_id:
# check - was added to handle erroneous TAXII servers # check - was added to handle erroneous TAXII servers
stix_obj = None stix_obj = None
else: else:
@ -246,7 +246,7 @@ class TAXIICollectionSource(DataSource):
all_data = [parse(stix_obj, allow_custom=self.allow_custom, version=version) for stix_obj in all_data] all_data = [parse(stix_obj, allow_custom=self.allow_custom, version=version) for stix_obj in all_data]
# check - was added to handle erroneous TAXII servers # check - was added to handle erroneous TAXII servers
all_data_clean = [stix_obj for stix_obj in all_data if stix_obj.id == stix_id] all_data_clean = [stix_obj for stix_obj in all_data if stix_obj['id'] == stix_id]
return all_data_clean return all_data_clean