45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
|
from flask import Flask, render_template, request, make_response, redirect
|
|||
|
|
from numpy import real
|
|||
|
|
from freecaptcha import captcha
|
|||
|
|
import uuid
|
|||
|
|
|
|||
|
|
app = Flask(__name__, static_url_path='', static_folder='static',)
|
|||
|
|
|
|||
|
|
captcha_solutions = {}
|
|||
|
|
captcha_solved = []
|
|||
|
|
|
|||
|
|
@app.route("/captcha", methods=['GET', 'POST'])
|
|||
|
|
def login():
|
|||
|
|
# This means they just submitted a CAPTCHA
|
|||
|
|
# We need to see if they got it right
|
|||
|
|
incorrect_captcha = False
|
|||
|
|
if request.method == 'POST':
|
|||
|
|
captcha_quess = request.form.get('captcha', None)
|
|||
|
|
captcha_cookie = request.cookies.get('freecaptcha_cookie')
|
|||
|
|
real_answer = captcha_solutions.get(captcha_cookie, None)
|
|||
|
|
if real_answer is not None:
|
|||
|
|
if int(captcha_quess) == int(real_answer):
|
|||
|
|
captcha_solved.append(captcha_cookie)
|
|||
|
|
return redirect("/", code=302)
|
|||
|
|
else:
|
|||
|
|
incorrect_captcha = True
|
|||
|
|
|
|||
|
|
# Select an image
|
|||
|
|
image_path = captcha.random_image()
|
|||
|
|
|
|||
|
|
# Generate list of rotated versions of image
|
|||
|
|
# and save which one is correct
|
|||
|
|
answer, options = captcha.captchafy(image_path)
|
|||
|
|
|
|||
|
|
# Provide the CAPTCHA options to the web page using the CAPTCHA
|
|||
|
|
resp = make_response(render_template("index.html", captcha_options=options, incorrect_captcha=incorrect_captcha))
|
|||
|
|
|
|||
|
|
# Track this user with a cookie and store the correct answer
|
|||
|
|
# by linking the cookie with the answer, we can check their answer
|
|||
|
|
# later
|
|||
|
|
freecaptcha_cookie = str(uuid.uuid4())
|
|||
|
|
resp.set_cookie('freecaptcha_cookie', freecaptcha_cookie)
|
|||
|
|
captcha_solutions[freecaptcha_cookie] = answer
|
|||
|
|
|
|||
|
|
return resp
|