使用 Captcha 模块来生成验证码图片

使用 Captcha 模块来生成验证码图片

安装

pip3 install captcha

验证码生成代码

import base64
import random
from io import BytesIO

from captcha.image import ImageCaptcha

seed = '123456789abcdefghijkmnpqrstuvwxyzABCDEFGHIJKLMNPQRSTUVWXYZ'


def generate_captcha():
    """
    生成验证码

    验证码使用 4 位
    :return: 验证码图片的 base64 表示
    """
    
    captcha_str = ''.join(random.choice(seed) for _ in range(4))
    image = ImageCaptcha().generate_image(captcha_str)
    buffer = BytesIO()
    image.save(buffer, format='PNG')
    data = buffer.getvalue()
    return 'data:image/png;base64,' + base64.b64encode(data).decode()


if __name__ == '__main__':
    print(generate_captcha())

验证码效果图