fix error when VT_APIKEY ENV var is set -

This commit is contained in:
Cian McElhinney 2021-07-15 16:29:38 +01:00
parent 7e1cea6cbb
commit 02d8b81171
1 changed files with 6 additions and 4 deletions

View File

@ -682,6 +682,8 @@ class Virustotal(enumratorBaseThreaded):
VT_APIKEY=VT_APIKEY.strip() VT_APIKEY=VT_APIKEY.strip()
if VT_APIKEY != "": if VT_APIKEY != "":
os.environ["VT_APIKEY"]=(VT_APIKEY) os.environ["VT_APIKEY"]=(VT_APIKEY)
else:
VT_APIKEY = os.getenv("VT_APIKEY")
os.environ["VT_APIKEY"]=(VT_APIKEY) os.environ["VT_APIKEY"]=(VT_APIKEY)
self.apikey = os.getenv('VT_APIKEY', None) self.apikey = os.getenv('VT_APIKEY', None)
self.q = q self.q = q