Merge pull request #751 from TheHive-Project/feature/emailrep_v2

added key to emailrep
TheHive-Project · Apr 27, 2020 · 48fb4d2 · 48fb4d2
2 parents 6263ff1 + 50628b0
commit 48fb4d2
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 38 deletions.
diff --git a/analyzers/EmailRep/EmailRep.json b/analyzers/EmailRep/EmailRep.json
@@ -7,5 +7,14 @@
   "description": "emailrep.io lookup.",
   "dataTypeList": ["mail"],
   "command": "EmailRep/emailrep_analyzer.py",
-  "baseConfig": "EmailRep"
+  "baseConfig": "EmailRep",
+  "configurationItems": [
+    {
+      "name": "key",
+      "description": "Define the API Key",
+      "type": "string",
+      "multi": false,
+      "required": false
+    }
+  ]  
 }
diff --git a/analyzers/EmailRep/emailrep.py b/analyzers/EmailRep/emailrep.py
diff --git a/analyzers/EmailRep/emailrep_analyzer.py b/analyzers/EmailRep/emailrep_analyzer.py
@@ -2,12 +2,14 @@
 # encoding: utf-8
 
 from cortexutils.analyzer import Analyzer
-from emailrep import EmailRepException, EmailRep
+from emailrep import EmailRep
 
 
 class EmailRepAnalyzer(Analyzer):
     def __init__(self):
         Analyzer.__init__(self)
+        self.key = self.get_param('config.key', None)
+
 
     def summary(self, raw):
         taxonomies = []
@@ -32,10 +34,10 @@ def run(self):
         data = self.get_data()
 
         try:
-            emailRep = EmailRep()
-            result = emailRep.get(data)
+            emailRep = EmailRep(self.key)
+            result = emailRep.query(data)
             self.report(result)
-        except EmailRepException as e:
+        except Exception as e:
             self.error(str(e))
 
 

diff --git a/analyzers/EmailRep/requirements.txt b/analyzers/EmailRep/requirements.txt
@@ -1,2 +1,3 @@
 cortexutils
 requests
+emailrep