From e0506ee31e3e6fbe013320ff6e35d165fed17cec Mon Sep 17 00:00:00 2001 From: Sebdraven Date: Tue, 20 Apr 2021 12:40:01 +0200 Subject: [PATCH] Update yeti.py filter by id --- misp_modules/modules/expansion/yeti.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/misp_modules/modules/expansion/yeti.py b/misp_modules/modules/expansion/yeti.py index 672f76a..87fe1f8 100644 --- a/misp_modules/modules/expansion/yeti.py +++ b/misp_modules/modules/expansion/yeti.py @@ -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']]