captcha_生成图片验证码并返回给前端展示

使用pip install captcha 安装模块

import random
import string
import os
import io
 
from captcha.image import ImageCaptcha, random_color
 
def generate_captcha():
    # # 返回大写字母
    # uppercase = string.ascii_uppercase
    # # 返回小写字母
    # lowercase = string.ascii_lowercase
    # 返回大小写字母
    letters = string.ascii_letters
    # 返回数字
    digits = string.digits
 
    # 随机选取 4 个字符
    random_str = ''.join([random.choice(digits+letters) for j in range(4)])
 
    width, height = 170, 80
 
    # 创建图片生成器
    generator = ImageCaptcha(width=width, height=height)
 
    # 创建验证码图片
    img = generator.create_captcha_image(chars=random_str, color=random_color(1, 255), background=random_color(1, 255))
 
    # 为图片增加干扰点
    generator.create_noise_dots(img, random_color(1, 255))
    # 为图片增加干扰线
    generator.create_noise_curve(img, random_color(1, 255))
 
    # 展示图片（展示图片的操作与上面保不保存无关）
    img.show()
 
    # 保存生成的图片到本地
    fileName = "saveImg"
    if not os.path.exists(fileName):
        os.mkdir(fileName)
    img.save("%s/%s.png" % (fileName, random_str))
 
    # 将图片保存到缓存中
    buffer = io.BytesIO()
    img.save(buffer, 'PNG')
    # 读取图片的byte值
    image_data = buffer.getvalue()
    buffer.close()
 
    return random_str, image_data

生成的验证码样式如下

前端展示效果