Python实现登录验证码

本文将从多个方面详细阐述Python如何实现登录验证码，让读者能够更深入地了解验证码的相关原理以及具体的实现方法。

一、生成验证码图片

首先，我们需要生成验证码图片。通常的做法是使用Python的PIL库（Python Imaging Library）来生成图片。具体步骤如下：

from PIL import Image, ImageDraw, ImageFont
import random

# 生成随机字符
def random_char():
    return chr(random.randint(65, 90))

# 生成随机颜色
def random_color():
    return (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255))

# 创建图片
def create_image(size=(120, 30), bgcolor=(255, 255, 255), fontcolor=(0, 0, 255)):
    width, height = size
    # 创建画布
    image = Image.new("RGB", size, bgcolor)
    # 创建画笔
    draw = ImageDraw.Draw(image)
    # 创建字体
    font = ImageFont.truetype("arial.ttf", 20)
    # 生成随机字符，并绘制到画布上
    for i in range(4):
        char = random_char()
        draw.text((i * 30 + 15, 5), char, font=font, fill=random_color())
    # 添加噪点
    for x in range(width):
        for y in range(height):
            if random.randint(0, 100) > 90:
                draw.point((x, y), fill=random_color())
    # 模糊处理
    image = image.filter(ImageFilter.BLUR)
    return image

上述代码通过PIL库生成了一个大小为120x30的随机验证码图片，并将图片保存在本地。

二、将验证码图片输出到HTML页面

接下来，我们需要将验证码图片输出到HTML页面中，供用户输入。

from io import BytesIO
from flask import Flask, make_response

app = Flask(__name__)

# 输出验证码图片
@app.route("/captcha/")
def captcha():
    # 生成验证码图片
    image = create_image()
    # 将验证码图片输出到内存中
    buf = BytesIO()
    image.save(buf, 'png')
    buf_str = buf.getvalue()
    response = make_response(buf_str)
    response.headers["Content-Type"] = "image/png"
    return response

上述代码使用Flask框架将验证码图片输出到HTML页面中，路由为/captcha/。

三、验证码验证

最后，我们需要对用户输入的验证码进行验证，确保用户输入正确。

from flask import Flask, request

app = Flask(__name__)

# 验证用户输入的验证码是否正确
@app.route("/verify/")
def verify():
    code = request.args.get("code", "")
    return "验证码正确" if code == session["code"] else "验证码错误"

上述代码使用Flask框架获取用户输入的验证码，与session中保存的验证码进行比对，如果相同，则验证成功。

四、总结

本文从生成验证码图片、输出到HTML页面、验证码验证三个方面详细阐述了Python实现登录验证码的相关知识，希望读者能够对验证码的实现方法有更深入的了解。