Merge pull request #1 from shijuleon/master

Error handling

sherlock.py

@@ -1,6 +1,10 @@
 import requests
 import json
 import os
+import sys
+import argparse
+
+DEBUG = False
 
 # TODO: fix tumblr
 
@@ -8,8 +12,34 @@ def write_to_file(url, fname):
 	with open(fname, "a") as f:
 		f.write(url+"\n")
 
-
-def main():
+def make_request(url, headers, error_type, social_network):
+    try:
+        r = requests.get(url, headers=headers)
+        if r.status_code:
+            return r, error_type
+    except requests.exceptions.HTTPError as errh:
+        if DEBUG:
+            print ("\033[37;1m[\033[91;1m-\033[37;1m]\033[91;1m HTTP Error:\033[93;1m", errh)
+        else:
+            print ("\033[37;1m[\033[91;1m-\033[37;1m]\033[91;1m HTTP Error:\033[93;1m", social_network)
+    except requests.exceptions.ConnectionError as errc:
+        if DEBUG:
+            print ("\033[37;1m[\033[91;1m-\033[37;1m]\033[91;1m Error Connecting:\033[93;1m", errc)
+        else:
+            print ("\033[37;1m[\033[91;1m-\033[37;1m]\033[91;1m Error Connecting:\033[93;1m", social_network)
+    except requests.exceptions.Timeout as errt:
+        if DEBUG:
+            print ("\033[37;1m[\033[91;1m-\033[37;1m]\033[91;1m Timeout Error:\033[93;1m", errt)
+        else:
+            print ("\033[37;1m[\033[91;1m-\033[37;1m]\033[91;1m Timeout Error:\033[93;1m", social_network)
+    except requests.exceptions.RequestException as err:
+        if DEBUG:
+            print ("\033[37;1m[\033[91;1m-\033[37;1m]\033[91;1m Unknown error:\033[93;1m", err)
+        else:
+            print ("\033[37;1m[\033[91;1m-\033[37;1m]\033[91;1m Unknown error:\033[93;1m", social_network)
+    return None, ""
+    
+def sherlock(username):
     # Not sure why, but the banner messes up if i put into one print function
     print("\033[37;1m                                              .\"\"\"-.")
     print("\033[37;1m                                             /      \\")
@@ -21,7 +51,6 @@ def main():
     print("\033[37;1m                                           .'`-._ `.\    | J /")
     print("\033[37;1m                                          /      `--.|   \__/\033[0m")
 
-    username = input("\033[92;1m[\033[37;1m?\033[92;1m]\033[92;1m Input Username: \033[0m")
     print()
 
     fname = username+".txt"
@@ -30,7 +59,6 @@ def main():
     	os.remove(fname)
     	print("\033[1;92m[\033[0m\033[1;77m*\033[0m\033[1;92m] Removing previous file:\033[1;37m {}\033[0m".format(fname))
 
-
     print("\033[1;92m[\033[0m\033[1;77m*\033[0m\033[1;92m] Checking username\033[0m\033[1;37m {}\033[0m\033[1;92m on: \033[0m".format(username))
     raw = open("data.json", "r", encoding="utf-8")
     data = json.load(raw)
@@ -45,15 +73,14 @@ def main():
     for social_network in data:
         url = data.get(social_network).get("url").format(username)
         error_type = data.get(social_network).get("errorType")
-
         cant_have_period = data.get(social_network).get("noPeriod")
 
         if cant_have_period == "True":
             print("\033[37;1m[\033[91;1m-\033[37;1m]\033[92;1m {}:\033[93;1m Not Found!".format(social_network))
             continue
-
-        r = requests.get(url, headers=headers)
-    
+            
+        r, error_type = make_request(url=url, headers=headers, error_type=error_type, social_network=social_network)
+        
         if error_type == "message":
             error = data.get(social_network).get("errorMsg")
             # Checks if the error message is in the HTML
@@ -82,6 +109,17 @@ def main():
             else:
             	print("\033[37;1m[\033[91;1m-\033[37;1m]\033[92;1m {}:\033[93;1m Not Found!".format(social_network))
 
+        elif error_type == "":
+            print("\033[37;1m[\033[91;1m-\033[37;1m]\033[92;1m {}:\033[93;1m Error!".format(social_network))
+
     print("\033[1;92m[\033[0m\033[1;77m*\033[0m\033[1;92m] Saved: \033[37;1m{}\033[0m".format(username+".txt"))
-        	
-main()
+
+parser = argparse.ArgumentParser()
+parser.add_argument('username', help='check services with given username')
+parser.add_argument("-d", '--debug', help="enable debug mode", action="store_true")
+
+args = parser.parse_args()
+if args.debug:
+    DEBUG = True
+if args.username:
+    sherlock(args.username)