checking live websites using getcode()

Author: root

modules/bcolors.py

@@ -0,0 +1,13 @@
+class bcolors:
+    HEADER = '\033[95m'
+    OKBLUE = '\033[94m'
+    OKGREEN = '\033[92m'
+    WARNING = '\033[93m'
+    FAIL = '\033[91m'
+    ENDC = '\033[0m'
+    BOLD = '\033[1m'
+    UNDERLINE = '\033[4m'
+    WHITE = '\033[97m'
+    On_Black = '\033[40m'
+    On_Red = '\033[41m'
+	

modules/getemails.py

@@ -1,3 +1,4 @@
+from modules.bcolors import bcolors
 import bs4
 
 """Get all emails from the website"""
@@ -17,10 +18,10 @@ def getMails(soup):
                 pass
         """Pretty print output as below"""
         print ('') 
-        print ('Mails Found - '+str(len(emails)))
+        print (bcolors.OKGREEN+'Mails Found - '+bcolors.ENDC+str(len(emails)))
         print ('-------------------------------')
         for mail in emails:
             print (mail)
         return ''
     else:
-        raise('Method parameter is not of instance bs4.BeautifulSoup')
+        raise(bcolors.FAIL+'Method parameter is not of instance bs4.BeautifulSoup'+bcolors.ENDC)

modules/getweblinks.py

@@ -1,3 +1,5 @@
+import urllib.request 
+import modules.bcolors
 import bs4
 
 """Get all onion links from the website"""
@@ -20,7 +22,10 @@ def getLinks(soup):
         print ('Websites Found - '+str(len(websites)))
         print ('-------------------------------')
         for web in websites:
-            print (web)
+            if (urllib.request.urlopen(web).getcode() == 200):
+                print (web)
+            else :
+                print(bcolors.On_Red+web +bcolors.ENDC)    
         return websites
     else:
         raise('Method parameter is not of instance bs4.BeautifulSoup')

torBot.py

@@ -1,6 +1,7 @@
 from modules.getemails import getMails
 from modules.getweblinks import getLinks
 from modules.pagereader import readPage
+from modules.bcolors import bcolors
 import socket
 import socks
 import argparse
@@ -23,20 +24,6 @@ def getaddrinfo(*args):
 
 socket.getaddrinfo = getaddrinfo
 
-class bcolors:
-    HEADER = '\033[95m'
-    OKBLUE = '\033[94m'
-    OKGREEN = '\033[92m'
-    WARNING = '\033[93m'
-    FAIL = '\033[91m'
-    ENDC = '\033[0m'
-    BOLD = '\033[1m'
-    UNDERLINE = '\033[4m'
-    WHITE = '\033[97m'
-    On_Black = '\033[40m'
-    On_Red = '\033[41m'
-	
-
 def header():
 	""" Display the header of TorBot """
 	print("#######################################################")
@@ -97,4 +84,10 @@ def main():
  return 0
 
 if __name__ == '__main__':
+
+ try:	
   main()
+ except KeyboardInterrupt:
+  print("Interrupt received! Exiting cleanly...")
+ 
+