python生成验证码图片代码分享

下面是“python生成验证码图片代码分享”的完整攻略。

1. 需求分析

我们需要实现一个Python程序，用于生成验证码图片。这个程序需要具备以下功能：

1. 生成一段随机的英文字母和数字字符组合的字符串。
2. 将生成的字符串渲染到一张图片上，并通过HTTP响应返回给用户。

2. 编写代码

2.1 安装依赖库

我们需要使用Pillow库来渲染图片，可以通过pip命令安装：

pip install Pillow

2.2 实现生成验证码字符串的函数

在Python中，我们可以使用random库和string库来生成随机字符。

下面是一个生成长度为4的随机字符串的示例代码：

import random
import string

def generate_code(length=4):
    """生成随机的验证码字符串"""
    chars = string.ascii_letters + string.digits  # 所有的英文大小写字母和数字
    code = ''.join(random.choice(chars) for _ in range(length))
    return code

2.3 实现生成图片的函数

我们使用Pillow库来生成图片，具体实现方法如下：

from PIL import Image, ImageDraw, ImageFont

def generate_image(code):
    """生成带有验证码的图片"""
    img_width = 200  # 图片宽度
    img_height = 100  # 图片高度
    font_size = int(img_height * 0.5)  # 字体大小
    image = Image.new('RGB', (img_width, img_height), color=(255, 255, 255))  # 创建一张白底图片
    draw = ImageDraw.Draw(image)  # 创建一个ImageDraw对象
    font = ImageFont.truetype('arial.ttf', font_size)  # 加载字体文件
    draw.text((10, 30), code, fill=(0, 0, 255), font=font)  # 使用ImageDraw的text方法渲染文字
    return image

2.4 实现HTTP服务

最后一步，我们需要将生成的图片通过HTTP响应返回给用户。

我们可以使用Python内置的http.server库来实现一个简单的HTTP服务。代码如下：

from http.server import BaseHTTPRequestHandler, HTTPServer
from io import BytesIO
import urllib

class MyHTTPHandler(BaseHTTPRequestHandler):

    def do_GET(self):
        self.send_response(200)
        self.send_header('Content-Type', 'image/jpeg')
        self.end_headers()

        code = generate_code()  # 生成随机的验证码字符串
        image = generate_image(code)  # 根据验证码字符串生成图片
        buffer = BytesIO()  # 创建一个BytesIO缓冲区
        image.save(buffer, 'jpeg')  # 将图片存储到缓冲区
        self.wfile.write(buffer.getvalue())  # 将缓冲区的数据作为HTTP响应返回给客户端

if __name__ == '__main__':
    server_address = ('', 8000)
    httpd = HTTPServer(server_address, MyHTTPHandler)
    print('Starting http server on localhost:%d ...' % server_address[1])
    httpd.serve_forever()

3. 测试结果

我们可以通过浏览器访问http://localhost:8000/，就能看到生成的验证码图片。

4. 示例说明

示例一

我们可以将上面的代码保存到文件中，比如命名为“captcha.py”，在命令行中运行：

python captcha.py

然后在浏览器中访问http://localhost:8000/，就能看到生成的验证码图片。

示例二

我们将上面的代码做一些修改，可以让生成的验证码图片加上日期时间的前缀，以避免出现重名的情况。

import time

class MyHTTPHandler(BaseHTTPRequestHandler):

    def do_GET(self):
        self.send_response(200)
        self.send_header('Content-Type', 'image/jpeg')
        self.end_headers()

        code = generate_code()  # 生成随机的验证码字符串
        timestamp = time.strftime('%Y%m%d%H%M%S', time.localtime())
        filename = 'captcha_{}_{}.jpeg'.format(timestamp, code)
        image = generate_image(code)  # 根据验证码字符串生成图片
        buffer = BytesIO()  # 创建一个BytesIO缓冲区
        image.save(buffer, 'jpeg')  # 将图片存储到缓冲区
        self.wfile.write(buffer.getvalue())  # 将缓冲区的数据作为HTTP响应返回给客户端

在命令行中运行：

python captcha.py

然后在浏览器中多次访问http://localhost:8000/，我们可以看到生成的验证码图片都带有日期时间前缀的文件名，避免了覆盖原文件的风险。