Possibility to use a Yeti apikey.

Optionally, we can provide a yeti apy_key. If so, we wolud need to modify the Yeti.json accordingly
TheHive-Project · Apr 26, 2019 · d0f4f53 · d0f4f53
1 parent 8dae6a8
commit d0f4f53
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/analyzers/Yeti/yeti.py b/analyzers/Yeti/yeti.py
@@ -9,6 +9,7 @@ class YetiAnalyzer(Analyzer):
     def __init__(self):
         Analyzer.__init__(self)
         self.url = self.get_param('config.url', None, 'Missing URL for Yeti API')
+        self.api_key = self.get_param('config.api_key')
 
     def summary(self, raw):
         count = len(raw.get('findings', []))
@@ -25,7 +26,7 @@ def summary(self, raw):
         return result
 
     def run(self):
-        api = pyeti.YetiApi("{}/api/".format(self.url))
+        api = pyeti.YetiApi("{}/api/".format(self.url), api_key=self.api_key)
         data = self.get_data()
 
         try: