使用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
生成的验证码样式如下
前端展示效果