fix error when VT_APIKEY ENV var is set -

2021-07-15 16:29:38 +01:00 · 2021-07-15 16:29:38 +01:00 · 02d8b81171
parent 7e1cea6cbb
commit 02d8b81171
1 changed files with 6 additions and 4 deletions
--- a/sublist3r2.py
+++ b/sublist3r2.py
@ -682,6 +682,8 @@ class Virustotal(enumratorBaseThreaded):
            VT_APIKEY=VT_APIKEY.strip()
            if VT_APIKEY != "":
                os.environ["VT_APIKEY"]=(VT_APIKEY)
+        else:
+            VT_APIKEY = os.getenv("VT_APIKEY")
        os.environ["VT_APIKEY"]=(VT_APIKEY)
        self.apikey = os.getenv('VT_APIKEY', None)
        self.q = q