Trafficsia

 #getCaptcha is used to get the current captcha as an image
def getCaptcha():
	#Find all image tags on the page
	imageTags = driver.find_elements_by_tag_name("img")
	#Checks whether tag is the one we're looking for
	#We do this by checking whether the src attribute is equal to truing.php
	for tag in imageTags:
		if (tag.get_attribute("src") == "http://www.trafficsia.com/traffics/turing.php"):
			capImg = tag
	#Now we take a screenshot of the image, as we can't just save since it's a PHP generated image.
	img = driver.get_screenshot_as_png()
	#We then proceed to crop the image so we only get the captcha
	capLoc = capImg.location
	imgSize = capImg.size
	im = Image.open(io.BytesIO(img))
	left = capLoc['x']
	top = capLoc['y']
	right = capLoc['x'] + imgSize['width']
	bottom = capLoc['y'] + imgSize['height']
	im = im.crop((left, top, right, bottom))
	#Return the cropped image
	return im

#In the splitCaptcha function we split the captcha in 5 different images containing the 5 different letters of the captcha
def splitCaptcha():
	captcha = getCaptcha()
	images = []
	for i in range(5):
		#Cropping math
		im = captcha.crop((7+(17*i), 7, 21+(17*i), 28))
		images.append(im)
	#Return the list of images
	return images

#This function will take the 5 images from splitCaptcha function and try to read them
#It has a pretty good success rate, but it's not always able to differentiate between 5s and 6s
def readCaptcha():
	images = splitCaptcha()
	captchaString = ""
	#Here we convert every image into a numpy arrays
	for image in images:
		image = np.array(image.convert("L"))
		ssimValues = []
		#Here we compare each numoy array to the numpy arrays of the saved image templates we have in the Letters folder
		for letter in letterImages:
			ssimValues.append(ssim(image, letter))
			#We then take the image with the highest similarity and add the corresponding letter to our guess at what captcha says
		captchaString += letters[ssimValues.index(max(ssimValues))]
	#Return out guess
	print(time.ctime() +" Returning captcha guess " + captchaString)
	return captchaString