add option to save the output

Author: bing0o

webtech.py

@@ -17,13 +17,12 @@
 # 
 #  
 
-from concurrent.futures import ThreadPoolExecutor as executor
+from concurrent.futures import ThreadPoolExecutor as executor # pip install futures
 from Wappalyzer import Wappalyzer, WebPage # pip install python-Wappalyzer
 import urllib3, argparse
 
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning) # disable Python SSL warnings !
 
-yellow = "\033[93m"
 green = "\033[32m"
 blue = "\033[34m"
 red = "\033[31m"
@@ -45,30 +44,37 @@
 
 
 
-def check(ig, url):
+def check(out, ig, url):
 	if not url.startswith('http'):
 		url = 'http://' + url
 	try:
 		webpage = WebPage.new_from_url(url)
 		tech = wappalyzer.analyze(webpage)
-		print(yellow+"[+] " + str(url) + end + bold + " | " + green + ", ".join(tech) + end)
+		print("[+] " + str(url) + " | " + green + bold + " - ".join(tech) + end)
+		if out != 'None':
+			with open(out, 'a') as f:
+				f.write(url + " | " + " - ".join(tech) + "\n")
+				f.close()
+	
 	except Exception as e:
 		if ig == 'True':
 			pass
 		else:
-			print(red+"Error: " + end + "[ " + bold + str(url) + end + " ] | " + str(e))
+			print(red+"Error: " + end + "[ " + bold + str(url) + end + " ] > " + str(e))
 
 
 parser = argparse.ArgumentParser()
 parser.add_argument("-w", "--wordlist", help="Domains List File", type=str, required=True)
 parser.add_argument("-t", "--thread", help="Theads Number - (Default: 10)", type=int)
 parser.add_argument("-i", "--ignore", help="To Ignore The Errors", action='store_true')
+parser.add_argument("-o", "--output", help="Save The Results To a File", type=str)
 
 args = parser.parse_args()
 
 links = str(args.wordlist)
 threads = str(args.thread)
 ig = str(args.ignore)
+out = str(args.output)
 
 lines = len(open(links).readlines())
 
@@ -77,6 +83,7 @@ def check(ig, url):
 
 print(blue +"["+red+"+"+blue+"] File: " + end + links)
 print(blue +"["+red+"+"+blue+"] Length: " + end + str(lines))
+print(blue +"["+red+"+"+blue+"] Output: " + end + str(out))
 print(blue +"["+red+"+"+blue+"] Threads: " + end + str(threads))
 print(blue +"["+red+"+"+blue+"] Ignore: " + end + str(ig))
 
@@ -85,5 +92,5 @@ def check(ig, url):
 
 urls = open(links,'r')
 with executor(max_workers=int(threads)) as exe:
-	[exe.submit(check, ig, url.strip('\n')) for url in urls]
+	[exe.submit(check, out, ig, url.strip('\n')) for url in urls]