Merge pull request #4 from cianmce/fix-virus-total-ref-error

fix error when VT_APIKEY ENV var is set
2021-07-22 22:20:20 +00:00 · 2021-07-22 22:20:20 +00:00 · 9e74b23d27
parent 7e1cea6cbb 02d8b81171
commit 9e74b23d27
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