From b42da0435b7927668bea2a40a6790d4e14d316e6 Mon Sep 17 00:00:00 2001
From: Sebdraven <sebdraven@protonmail.com>
Date: Fri, 19 Mar 2021 15:55:18 +0100
Subject: [PATCH] Update yeti.py

add key results
---
 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 a6318b5..1028f0c 100644
--- a/misp_modules/modules/expansion/yeti.py
+++ b/misp_modules/modules/expansion/yeti.py
@@ -27,7 +27,7 @@ class Yeti():
         self.yeti_client = pyeti.YetiApi(url=url, api_key=key)
         self.attribute = attribute
         self.misp_event = MISPEvent()
-        #self.misp_event.add_attribute(**attribute)
+        self.misp_event.add_attribute(**attribute)
 
     def search(self, value):
         obs = self.yeti_client.observable_search(value=value)
@@ -126,7 +126,7 @@ def handler(q=False):
 
     if yeti_client:
         yeti_client.parse_yeti_result()
-        return yeti_client.get_result()
+        return {'results': yeti_client.get_result()}
     else:
         misperrors['error'] = 'Yeti Config Error'
         return misperrors