Use VirusTotal with python3

TheHive-Project · Mar 26, 2019 · 97a1ff4 · 97a1ff4
1 parent 9cb7dcf
commit 97a1ff4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/analyzers/VirusTotal/virustotal.py b/analyzers/VirusTotal/virustotal.py
@@ -46,7 +46,7 @@ def check_response(self, response):
             self.error('Bad status : ' + str(status))
         results = response.get('results', {})
         if 'verbose_msg' in results:
-            print >> sys.stderr, str(results.get('verbose_msg'))
+            print (str(results.get('verbose_msg')), sys.stderr)
         if 'Missing IP address' in results.get('verbose_msg', ''):
             results['verbose_msg'] = 'IP address not available in VirusTotal'
         return results

diff --git a/analyzers/VirusTotal/virustotal_api.py b/analyzers/VirusTotal/virustotal_api.py
@@ -22,10 +22,11 @@
 """
 
 import os
-import StringIO
 try:
     import requests
+    import StringIO
 except ImportError:
+    from io import StringIO
     pass