https域名验证码

from flask import Flask, render_template, request, session
import random
import string
from io import BytesIO
from PIL import Image, ImageDraw, ImageFont

app = Flask(__name__)
app.secret_key = 'your_secret_key'

def generate_captcha():
    captcha_text = ''.join(random.choices(string.ascii_uppercase + string.digits, k=6))
    session['captcha'] = captcha_text
    img = Image.new('RGB', (150, 50), color=(255, 255, 255))
    font = ImageFont.truetype('arial.ttf', 36)
    d = ImageDraw.Draw(img)
    d.text((10, 10), captcha_text, fill=(0, 0, 0), font=font)
    img_io = BytesIO()
    img.save(img_io, 'PNG')
    return img_io

@app.route('/')
def index():
    return render_template('index.html')

@app.route('/captcha')
def captcha():
    return generate_captcha().getvalue()

@app.route('/verify', methods=['POST'])
def verify():
    user_input = request.form['captcha']
    if user_input == session.get('captcha'):
        return '验证成功'
    else:
        return '验证失败'

if __name__ == '__main__':
    app.run(debug=True)