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

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

fix error when VT_APIKEY ENV var is set
This commit is contained in:
RoninNakomoto 2021-07-22 22:20:20 +00:00 committed by GitHub
parent 7e1cea6cbb 02d8b81171
commit 9e74b23d27
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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' base_url = 'https://www.virustotal.com/api/v3/domains/{domain}/subdomains'
self.engine_name = "Virustotal" self.engine_name = "Virustotal"
if os.getenv("VT_APIKEY") is None: if os.getenv("VT_APIKEY") is None:
VT_APIKEY=input(B + "[+] Enter VirusTotal API key, press Enter for none: " + W) VT_APIKEY=input(B + "[+] Enter VirusTotal API key, press Enter for none: " + W)
VT_APIKEY=VT_APIKEY.strip() VT_APIKEY=VT_APIKEY.strip()
if VT_APIKEY != "": if VT_APIKEY != "":
os.environ["VT_APIKEY"]=(VT_APIKEY) os.environ["VT_APIKEY"]=(VT_APIKEY)
else:
VT_APIKEY = os.getenv("VT_APIKEY")
os.environ["VT_APIKEY"]=(VT_APIKEY) os.environ["VT_APIKEY"]=(VT_APIKEY)
self.apikey = os.getenv('VT_APIKEY', None) self.apikey = os.getenv('VT_APIKEY', None)
self.q = q self.q = q