Added colors and filters

Author: PSNAppz

modules/getweblinks.py

@@ -22,7 +22,7 @@ def getLinks(soup):
                 pass
         """Pretty print output as below"""
         print ('') 
-        print ('Websites Found - '+str(len(websites)))
+        print (bcolors.OKGREEN+'Websites Found - '+bcolors.ENDC+str(len(websites)))
         print ('-------------------------------')
         for web in websites:
             if (urllib.request.urlopen(web).getcode() == 200):

modules/pagereader.py

@@ -1,12 +1,16 @@
 import urllib.request 
 from bs4 import BeautifulSoup
+from modules.bcolors import bcolors
 
 
-def readPage(site):
+def readPage(site,printIP=0):
 
  headers = {'User-Agent': 'TorBot - Onion crawler | www.github.com/DedSecInside/TorBot' }
  req = urllib.request.Request(site,None,headers)
  response = urllib.request.urlopen(req)
  page = BeautifulSoup(response.read(),'html.parser')
- print (page.find_all('strong'))
+ if printIP:
+  pg = page.find('strong')
+  IP = pg.renderContents()
+  print(bcolors.WARNING+bcolors.BOLD+IP.decode("utf-8")+bcolors.ENDC)
  return page

tests/test_getweblinks.py

@@ -16,7 +16,7 @@ def setUp(self):
 
 	def test_print_links(self):
 		#data = "\nWebsites Found - 7\n-------------------------------\nhttp://ads.wsrs.net/www/delivery/ck.php?n=MyIP856a6b4\nhttp://ads.wsrs.net/www/delivery/ck.php?n=MyIPbf5d683\nhttp://aff.ironsocket.com/SH7L\nhttp://aff.ironsocket.com/SH7L\nhttp://ads.wsrs.net/www/delivery/ck.php?n=MyIPdb5f512\nhttp://wsrs.net/\nhttp://cmsgear.com/\n"
-		data = "\nWebsites Found - 0\n-------------------------------\n"
+		data = "\n"+bcolors.OKGREEN+"Websites Found - "+bcolors.ENDC+"0\n-------------------------------\n"
 
 		getweblinks.getLinks(soup)
 		self.assertEqual(sys.stdout.getvalue(),data)

torBot.py

@@ -58,25 +58,25 @@ def header():
 	print( "MMMMMMMMMMMMMWNklccclldk0OxOdcc;. .......;oKWWMMMMMMMM")
 	print( "MMMMMMMMMMMMMMMMWXOdl:::;cc;'... ..',:lx0NMMMMMMMMMMMM")
 	print( "MMMMMMMMMMMMMMMMMMMMMNKOkxddolloodk0XWMMMMMMMMMMMMMMMM")
-	print(bcolors.FAIL)
+	print(bcolors.FAIL+bcolors.BOLD)
 	print( " 	       __  ____  ____  __        ______ ")
 	print( "  	      / /_/ __ \/ __ \/ /_  ____/_  __/ ")
 	print( " 	     / __/ / / / /_/ / __ \/ __ \/ / ")
 	print( "	    / /_/ /_/ / _, _/ /_/ / /_/ / /  ")
 	print( "	    \__/\____/_/ |_/_.___/\____/_/  V 0.0.3")
-	print(bcolors.On_Black)
+	print(bcolors.FAIL+bcolors.On_Black)
 	print("#######################################################")
 	print("#  TorBot - A python Tor Crawler                      #")
 	print("#  GitHub : https://github.com/DedsecInside/TorBot    #")
-	print("######################################################")
+	print("#######################################################")
 	print(bcolors.FAIL + "LICENSE: GNU Public License" + bcolors.ENDC)
 	print()
 
 
 def main():
  header()
  print ("Tor Ip Address :")
- a = readPage("https://check.torproject.org/")
+ a = readPage("https://check.torproject.org/",1)
  b = readPage("http://torlinkbgs6aabns.onion/")
  getMails(b)
  getLinks(b)
@@ -88,6 +88,4 @@ def main():
  try:	
   main()
  except KeyboardInterrupt:
-  print("Interrupt received! Exiting cleanly...")
- 
-  	
+  print("Interrupt received! Exiting cleanly...")