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
|