Flask 验证码自动生成

Flask 验证码自动生成

想必验证码大家都有所了解，但是可以自己定义图片验证码，包含数字，英文以及数字计算，自动生成验证码。

生成图片以及结果

from captcha.image import ImageCaptcha
 
from PIL import Image
from random import choices
 
 
def gen_captcha(content='2345689abcdefghijklmnpqrstuvwxyzABCDEFGHIJKLMNPQRSTUVWXYZ'):
    """
    生成验证码，可自定义，这里是生成验证码结果，以及验证码文件
    """
    image = ImageCaptcha()
    # 获取字符串
    captcha_text = "".join(choices(content, k=4)).lower()
    # 生成图像
    captcha_image = Image.open(image.generate(captcha_text))
    return captcha_text, captcha_image

得到生成的验证码，进行处理，响应到页面

处理验证码

from io import BytesIO
from flask import  make_response，session
# 生成验证码
def get_captcha():
    code, image = gen_captcha() #得到验证码内容
    out = BytesIO() # 操作二进制数据，实例化
    image.save(out, 'png') # 保存验证码图片
    out.seek(0)
    resp = make_response(out.read()) #将验证码图片的bytes制作成页面响应结果， 具体可查看make_response这个方法
    resp.content_type = 'image/png' # 告诉页面这个响应的类型
    return resp, code

制作响应路由，使用蓝图

蓝图

from flask import Blueprint, session, redirect, url_for, render_template, request
passport_bp = Blueprint('passport', __name__, url_prefix='/passport') # 生成蓝图passport_bp
 
# 获取验证码
@passport_bp.get('/getCaptcha')
def get_captcha():
    resp, code = index_curd.get_captcha()  #获取图片
    session["code"] = code # 验证码结果保存到session或者数据库中，这里保存在session
    return resp # 返回图片结果

html页面展示

html展示

"""
<img src="{{ url_for('passport.get_captcha') }}" class="codeImage" id="captchaImage"/>
"""
<script> # 使用的layui框架
    layui.use(['form', 'jquery', 'layer', 'button', 'popup'], function () {
        let form = layui.form;
        let $ = layui.jquery;
        let layer = layui.layer;
        let button = layui.button;
        let popup = layui.popup;
        let captchaPath = "{{ url_for('passport.get_captcha') }}";
 
        $("#captchaImage").click(function () { # 点击切换
            document.getElementById("captchaImage").src = captchaPath + "?" + Math.random();
        });
        setInterval(function () { # 30秒自动切换
            document.getElementById("captchaImage").src = captchaPath + "?" + Math.random();
        }, 30 * 1000);
    })
</script>