From 66fc121dbe0dbb7a69a62bfdaf98838a4f7a0bf3 Mon Sep 17 00:00:00 2001
From: Sebdraven <sebdraven@protonmail.com>
Date: Fri, 5 Feb 2021 11:17:40 +0100
Subject: [PATCH] Update yeti.py

add config and struct
---
 misp_modules/modules/expansion/yeti.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/misp_modules/modules/expansion/yeti.py b/misp_modules/modules/expansion/yeti.py
index 5742a08..0dd2275 100644
--- a/misp_modules/modules/expansion/yeti.py
+++ b/misp_modules/modules/expansion/yeti.py
@@ -1,6 +1,7 @@
 import json
 
 import json
+
 try:
     import pyeti
 except ImportError:
@@ -15,3 +16,23 @@ moduleinfo = {'version': '1', 'author': 'Sebastien Larinier @sebdraven',
               'description': 'Query on yeti',
               'module-type': ['expansion', 'hover']}
 
+moduleconfig = ['apikey', 'url']
+
+
+class Yeti:
+
+    def __init__(self, url, key):
+        self.api = pyeti.YetiApi(url, api_key=key)
+        self.dict = {'Ip': 'ip-src', 'Domain': 'domain', 'Hostname': 'hostname'}
+
+    def search(self, value):
+        obs = self.api.observable_search(value=value)
+        if obs:
+            return obs
+
+    def 
+def handler(q=False):
+    if q is False:
+        return False
+    request = json.loads(q)
+    attribute = request['attribute']