From 28b554d975818881135734790ed44d5f3f746474 Mon Sep 17 00:00:00 2001 From: Sebdraven Date: Fri, 19 Mar 2021 12:24:15 +0100 Subject: [PATCH] Update yeti.py add test --- misp_modules/modules/expansion/yeti.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/misp_modules/modules/expansion/yeti.py b/misp_modules/modules/expansion/yeti.py index fe4f4db..c76cc9d 100644 --- a/misp_modules/modules/expansion/yeti.py +++ b/misp_modules/modules/expansion/yeti.py @@ -71,7 +71,8 @@ class Yeti(): types = [] for obs_to_add in self.get_neighboors(obs['id']): object_misp = self.get_object(obs_to_add) - self.misp_event.add_object(object_misp) + if object_misp: + self.misp_event.add_object(object_misp) print(self.misp_event) def get_result(self): @@ -86,7 +87,6 @@ class Yeti(): domain_ip_object.add_attribute(**self.__get_attribute(obj_to_add)) #domain_ip_object.add_reference(self.attribute['uuid'], 'related_to') domain_ip_object.add_attribute(**self.attribute) - print(domain_ip_object.to_json()) return domain_ip_object def __get_attribute(self, obj_yeti):