From 7b35b829966595559ecabfd0f7561393f2e0f6b6 Mon Sep 17 00:00:00 2001 From: Emmanuelle Vargas-Gonzalez Date: Tue, 29 Sep 2020 11:21:02 -0400 Subject: [PATCH] TAXIICollectionSource custom object handling fix --- stix2/datastore/taxii.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/stix2/datastore/taxii.py b/stix2/datastore/taxii.py index 063fabd..41d1e54 100644 --- a/stix2/datastore/taxii.py +++ b/stix2/datastore/taxii.py @@ -210,7 +210,7 @@ class TAXIICollectionSource(DataSource): if len(stix_obj): 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 stix_obj = None 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] # 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