#42 Set the VT polling interval to 60 seconds by default

TheHive-Project · Apr 21, 2017 · 3329888 · 3329888
1 parent 814d561
commit 3329888
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/analyzers/VirusTotal/virustotal.py b/analyzers/VirusTotal/virustotal.py
@@ -15,18 +15,17 @@ class VirusTotalAnalyzer(Analyzer):
 
     def __init__(self):
         Analyzer.__init__(self)
-        self.service = self.getParam(
-            'config.service', None, 'Service parameter is missing')
-        self.virustotal_key = self.getParam(
-            'config.key', None, 'Missing VirusTotal API key')
+        self.service = self.getParam('config.service', None, 'Service parameter is missing')
+        self.virustotal_key = self.getParam('config.key', None, 'Missing VirusTotal API key')
+        self.polling_interval = self.getParam('config.polling_interval', 60)
 
     def wait_file_report(self, id):
         results = self.check_response(self.vt.get_file_report(id))
         code = results.get('response_code', None)
         if code == 1:
             self.report(results)
         else:
-            time.sleep(10)
+            time.sleep(self.polling_interval)
             self.wait_file_report(id)
 
     def wait_url_report(self, id):
@@ -35,7 +34,7 @@ def wait_url_report(self, id):
         if code == 1:
             self.report(results)
         else:
-            time.sleep(60)
+            time.sleep(self.polling_interval)
             self.wait_url_report(id)
 
     def check_response(self, response):