Update yeti.py

fix keyerror
pull/488/head
Sebdraven 2021-04-20 12:30:22 +02:00
parent 9cb1a83e54
commit a2741e8eb7
1 changed files with 1 additions and 1 deletions

View File

@ -39,7 +39,7 @@ 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['id']: link['description'] for link in neighboors['links']}
links_by_id = {link['dst']['id']: link['description'] for link in neighboors['links']}
for n in neighboors['objs']:
yield n, links_by_id[n['id']]