From a29779eff6f758d2f22ecd3bc5395f85bd9f6b8e Mon Sep 17 00:00:00 2001 From: Sebdraven Date: Mon, 19 Apr 2021 11:24:01 +0200 Subject: [PATCH] Update yeti.py add check --- misp_modules/modules/expansion/yeti.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/misp_modules/modules/expansion/yeti.py b/misp_modules/modules/expansion/yeti.py index 82a20d5..ed4d02d 100644 --- a/misp_modules/modules/expansion/yeti.py +++ b/misp_modules/modules/expansion/yeti.py @@ -100,9 +100,16 @@ class Yeti(): obj_to_add['type'] in ('Hostname', 'Domain', 'Ip') and self.attribute['type'] == 'url' ): url_object = MISPObject('Url') - url_object.add_attribute(self.__get_relation(obj_to_add), obj_to_add['value']) - url_object.add_attribute(self.__get_relation(self.attribute), + obj_relation = self.__get_relation(obj_to_add) + if obj_relation: + print(obj_relation) + url_object.add_attribute(self.__get_relation(obj_to_add), obj_to_add['value']) + obj_relation = self.__get_relation(self.attribute) + if obj_relation: + print(obj_relation) + url_object.add_attribute(self.__get_relation(self.attribute), self.attribute['value']) + url_object.add_reference(self.attribute['uuid'], 'related_to') print(url_object) return url_object