From 691766d8271e2ed6822c4094b790e8ee2e511b0c Mon Sep 17 00:00:00 2001 From: Thomas Bouve Date: Sun, 24 Oct 2021 20:29:53 +0200 Subject: [PATCH] Clean VT_APIKEY code --- sublist3r2/__init__.py | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/sublist3r2/__init__.py b/sublist3r2/__init__.py index f2e0d0f..1ebce6a 100755 --- a/sublist3r2/__init__.py +++ b/sublist3r2/__init__.py @@ -690,15 +690,13 @@ class Virustotal(enumratorBaseThreaded): subdomains = subdomains or [] 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) - else: - VT_APIKEY = os.getenv("VT_APIKEY") - os.environ["VT_APIKEY"] = (VT_APIKEY) - self.apikey = os.getenv('VT_APIKEY', None) + self.apikey = os.getenv("VT_APIKEY") + + if self.apikey is None: + vt_apikey = input(B + "[+] Enter VirusTotal API key, press Enter for none: " + W).strip() + if vt_apikey != "": + self.apikey = vt_apikey + self.q = q super(Virustotal, self).__init__( base_url, self.engine_name, domain, subdomains,