diff --git a/misp_modules/modules/expansion/yeti.py b/misp_modules/modules/expansion/yeti.py index b53f653..f044f76 100644 --- a/misp_modules/modules/expansion/yeti.py +++ b/misp_modules/modules/expansion/yeti.py @@ -73,9 +73,9 @@ class Yeti(): object_misp_domain_ip = self.__get_object_domain_ip(obs_to_add) if object_misp_domain_ip: self.misp_event.add_object(object_misp_domain_ip) - # object_misp_url = self.__get_object_url(obs_to_add) - # if object_misp_url: - # self.misp_event.add_object(object_misp_url) + object_misp_url = self.__get_object_url(obs_to_add) + if object_misp_url: + self.misp_event.add_object(object_misp_url) def get_result(self): event = json.loads(self.misp_event.to_json()) @@ -104,7 +104,7 @@ class Yeti(): if obj_relation: print(obj_relation) print(obj_to_add['value']) - url_object.add_attribute('url', obj_to_add['value']) + url_object.add_attribute(obj_relation, obj_to_add['value']) obj_relation = self.__get_relation(self.attribute) if obj_relation: print(obj_relation)