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

fix error when VT_APIKEY ENV var is set -

This commit is contained in:
Cian McElhinney 2021-07-15 16:29:38 +01:00
parent 7e1cea6cbb
commit 02d8b81171
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
sublist3r2.py
View File

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