username-checker.py

# Tab = 2
import urllib.request as urllib2
import sys
# Define ANSI escape sequences for colored output
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'

def checkUrl(username, url):
	url=url.format(username)
	request=urllib2.Request(url)
	return parse(request,username)+"\n"

# Process the request and capture the response code
def middleware(request, username):
	try:
		response=urllib2.urlopen(request)
	except urllib2.HTTPError as e:
		if e.code==404:
			return (e.code, bcolors.OKGREEN, "Available")
		else:
			return (e.code, bcolors.WARNING, e.reason)
	else:
		if response.getcode() == 200:
			return (response.getcode(), bcolors.FAIL, "Already exists")

# Used for generating colored responses
def parse(request, username):
	response = middleware(request, username)
	return (bcolors.HEADER+"├─ "+bcolors.ENDC +response[1]+ request.get_full_url()+" ["+str(response[0])+"] "+response[2]+bcolors.ENDC)

def checkUsername(username):
	# Read file
	try:
		with open("URL_CHECK.txt") as f:
			l_url=f.readlines()
		f.close()
	except:
		sys.exit("❌ [URL_CHECK.txt] Not found")
		
	# Check the urls
	rtn =""
	for url in l_url:
		# print(url)
		rtn+=checkUrl(username, url)
		
	rtn=sorted(rtn.split("\n"))
	rtn="\n".join(rtn).strip()
	
	return rtn

def showHeader(username):
	rtn =bcolors.HEADER+"__________________________________________________________________\n"+bcolors.ENDC
	rtn+=(bcolors.HEADER+"█ "+bcolors.ENDC+bcolors.BOLD+"{}"+bcolors.ENDC+bcolors.HEADER+" █ Checking for availability, please wait...\n"+bcolors.ENDC).format(username)
	return rtn
	
if __name__ == "__main__":
	# Ask for a username, instead of an argument
	msg=""
	if len(sys.argv[1:])==0:
		username=input("Username to check: ")
		print("This may take a while...")
		
		msg+=showHeader(username)
		msg+=checkUsername(username)
	else:
		# Take username as a system argument
		msg="Checking: "
		for username in sys.argv[1:]:
			msg+=bcolors.BOLD+username+bcolors.ENDC+", "
		print(msg[:-2])
		print("This may take a while...")
		
		msg=""
		for username in sys.argv[1:]:
			msg+=showHeader(username)
			msg+=checkUsername(username)
	
	## Show
	print(msg)