如何用Python生成验证码

验证码（CAPTCHA）是一种用于验证用户是否为人类而不是机器的技术。在网站注册、登录、评论等场景中广泛应用。本文将从生成随机字符串、生成验证码图片和验证用户输入的验证码三个方面来讲解如何用Python生成验证码。

一、生成随机字符串

生成验证码首先需要生成一组包含字母和数字的随机字符串。Python中可以使用random模块来生成随机字符串。

import random
import string

def generate_random_string(length):
    chars = string.ascii_letters + string.digits
    return ''.join(random.choice(chars) for _ in range(length))

random_string = generate_random_string(6)
print(random_string)

以上代码使用random模块的choice函数随机选择字符串中的字符，通过循环生成指定长度的随机字符串。

二、生成验证码图片

生成验证码图片可以增加验证码的可读性和视觉效果。Python中可以使用Pillow库来创建和操作图像。

from PIL import Image, ImageDraw, ImageFont

def generate_captcha_image(text, font_path, img_size=(200, 100), bg_color=(255, 255, 255), text_color=(0, 0, 0)):
    image = Image.new('RGB', img_size, bg_color)
    draw = ImageDraw.Draw(image)
    font = ImageFont.truetype(font_path, 40)
    text_width, text_height = draw.textsize(text, font)
    x = (img_size[0] - text_width) // 2
    y = (img_size[1] - text_height) // 2
    draw.text((x, y), text, font=font, fill=text_color)
    image.show()

font_path = 'path/to/font.ttf'
generate_captcha_image(random_string, font_path)

以上代码使用Pillow库创建一个指定大小的空白图像，然后使用ImageDraw模块的text函数在图像上绘制验证码文本。

三、验证用户输入的验证码

在网站表单中，用户输入的验证码需要与生成的验证码进行比对以验证用户输入的是否正确。

def check_captcha(user_input, generated_captcha):
    return user_input.lower() == generated_captcha.lower()

user_input = input('请输入验证码：')
if check_captcha(user_input, random_string):
    print('验证码输入正确！')
else:
    print('验证码输入错误！')

以上代码通过定义一个函数来比对用户输入的验证码和生成的验证码是否相等，忽略大小写。根据比对结果输出相应的提示信息。

总结

本文介绍了如何用Python生成验证码，包括生成随机字符串、生成验证码图片和验证用户输入的验证码。通过这三个步骤可以实现一个简单的验证码系统。在实际应用中，还可以进一步增加验证码的复杂度，如添加干扰线、干扰点等。

通过本文的讲解，相信读者对如何用Python生成验证码有了更清晰的了解。