RomanNum3ral
/
Sublist3r
mirror of https://github.com/aboul3la/Sublist3r.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Clean VT_APIKEY code

This commit is contained in:
Thomas Bouve 2021-10-24 20:29:53 +02:00
parent 5157ac5454
commit 691766d827
1 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
sublist3r2/__init__.py
View File

@ -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,