MISP
/
cti-python-stix2
mirror of https://github.com/MISP/cti-python-stix2
2
0
Fork 0
Code Issues Releases Wiki Activity

Simplify `related_to`

stix2.0
Chris Lenk 2017-11-21 16:19:15 -05:00
parent 6446be310c
commit 078474259a
1 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
stix2/sources/__init__.py
View File

@ -331,17 +331,14 @@ class DataSource(with_metaclass(ABCMeta)):
# Assume `obj` is an ID string # Assume `obj` is an ID string
obj_id = obj obj_id = obj
# Get all unique ids from the relationships except that of the object
ids = set()
for r in rels: for r in rels:
if not source_only: ids.update((r.source_ref, r.target_ref))
# relationships() found relationships where target_ref is obj_id ids.remove(obj_id)
source_id = r.source_ref
if source_id != obj_id: # needed if target_only is also false for i in ids:
results.append(self.get(source_id)) results.append(self.get(i))
if not target_only:
# relationships() found relationships where source_ref is obj_id
target_id = r.target_ref
if target_id != obj_id: # needed if source_only is also false
results.append(self.get(target_id))
return results return results