使用 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())
验证码效果图