python随机生成验证码

python随机生成验证码

需要导入pillow模块

from random import choice, randint

from PIL import Image, ImageDraw, ImageFont, ImageFilter


class VerifyCode(object):
    """生成验证码模块"""

    def __init__(self, length=4, width=160, height=50, font_size=40):
        self._random_code = ''  # 验证码字符串
        self._verify_code_image = None  # PIL图片Image对象
        self._length = length  # 验证码长度
        self._width = width  # 图片宽度
        self._height = height  # 图片高度
        self._font_size = font_size  # 字体大小
        self.random_code()  # 初始化验证码字符
        self.set_image()  # 绘制图片

    def random_code(self):
        chars = '1234567890qwertyuiopasdfghjklzxcvbnm'  # 验证码字符库
        code = ''  # 生成的验证码
        for _ in range(self._length):  # 循环随机取一个字符
            code += choice(chars)
        self._random_code = code  # 赋值给当前对象的random_code属性

    @staticmethod
    def random_color(s=0, e=255):
        return randint(0, 256), randint(0, 256), randint(0, 256)

    def set_image(self):
        # 创建画布
        image = Image.new('RGB', (self._width, self._height), (255, 255, 255))
        # 创建字体对象
        font = ImageFont.truetype('arial.ttf', self._font_size)
        # 创建画图对象
        draw = ImageDraw.Draw(image)
        # 随机生成噪点
        for x in range(self._width):
            for y in range(self._height):
                temp = x + y + randint(0, 10)
                if temp % 10 == 0:
                    draw.point((x, y), fill=self.random_color(0, 255))
        # 字符添加
        for t in range(self._length):
            dev_x = randint(0, 5)  # 随机左右浮动
            dev_y = randint(0, 5)  # 随机上下浮动
            x, y = self._font_size * t + dev_x, dev_y
            # 将字符通过随机颜色画到图片中
            draw.text((x, y), self._random_code[t],
                      font=font, fill=self.random_color(0, 200))
        # 高斯模糊
        image = image.filter(ImageFilter.GaussianBlur)
        # 将图片对象赋值给当前对象的verify_code_image属性
        self._verify_code_image = image

    @property
    def verify_code(self):
        return self._random_code

    @property
    def verify_image(self):
        return self._verify_code_image

    def get_verify(self):
        return self._random_code, self._verify_code_image


if __name__ == '__main__':
    v1 = VerifyCode()
    print(v1.get_verify())

# ('tspf', <PIL.Image.Image image mode=RGB size=160x50 at 0x1AA45B00D68>)