MISP
/
misp-modules
mirror of https://github.com/MISP/misp-modules
2
0
Fork 0
Code Issues Releases Wiki Activity

Update yeti.py 

filter by id
pull/488/head
Sebdraven 2021-04-20 12:40:01 +02:00
parent f701256008
commit e0506ee31e
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
misp_modules/modules/expansion/yeti.py
View File

@ -39,8 +39,11 @@ class Yeti():
def get_neighboors(self, obs_id):
neighboors = self.yeti_client.neighbors_observables(obs_id)
if neighboors and 'objs' in neighboors:
links_by_id = {link['dst']['id']: link['description'] for link in neighboors['links']}
links_by_id.update({link['src']['id']: link['description'] for link in neighboors['links']})
links_by_id = {link['dst']['id']: link['description'] for link in neighboors['links']
if link['dst']['id'] != obs_id}
links_by_id.update({link['src']['id']: link['description'] for link in neighboors['links']
if link['src']['id'] != obs_id})
for n in neighboors['objs']:
yield n, links_by_id[n['id']]