From 02d8b81171e0506e0d30c4fe139f220b744c6dd5 Mon Sep 17 00:00:00 2001 From: Cian McElhinney Date: Thu, 15 Jul 2021 16:29:38 +0100 Subject: [PATCH] fix error when VT_APIKEY ENV var is set - --- sublist3r2.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/sublist3r2.py b/sublist3r2.py index e0ee1a2..f66517f 100755 --- a/sublist3r2.py +++ b/sublist3r2.py @@ -678,10 +678,12 @@ class Virustotal(enumratorBaseThreaded): base_url = 'https://www.virustotal.com/api/v3/domains/{domain}/subdomains' self.engine_name = "Virustotal" if os.getenv("VT_APIKEY") is None: - VT_APIKEY=input(B + "[+] Enter VirusTotal API key, press Enter for none: " + W) - VT_APIKEY=VT_APIKEY.strip() - if VT_APIKEY != "": - os.environ["VT_APIKEY"]=(VT_APIKEY) + VT_APIKEY=input(B + "[+] Enter VirusTotal API key, press Enter for none: " + W) + 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