最近需要做一个验证码的功能,在网上找了许多资料,都是一些静态的验证码图片,即都是png或jpg的图片。于是,借鉴网上的一些资料,弄一个gif的验证码图片,现分享给大家。
所需依赖如下:
pip3 install pillow
pip3 install imageio
完整代码如下:
import os
import random
from io import BytesIO
import imageio
from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont
class GifCodeImage(object):
def __init__(self, width=200, height=75, code_count=4, font_size=32, point_count=20, line_count=3, frame_count=30):
"""
:param width: 图片宽度
:param height: 图片高度
:param code_count: 验证码位数
:param font_size: 字体大小
:param point_count: 噪点数量
:param line_count: 噪线数量
:param frame_count: gif的帧数
"""
self.width = width
self.height = height
self.code_count = code_count
self.font_size = font_size
self.point_count = point_count
self.line_count = line_count
self.frame_count = frame_count
@staticmethod
def get_random_color():
"""
获取一个随机颜色(r,g,b)格式的
:return:
"""
c1 = random.randint(0, 255)
c2 = random.randint(0, 255)
c3 = random.randint(0, 255)
return c1, c2, c3
@staticmethod
def get_random_str():
"""
获取一个随机字符串,每个字符的颜色也是随机的
:return:
"""
random_num = str(random.randint(0, 9))
random_low_alpha = chr(random.randint(97, 122))
random_upper_alpha = chr(random.randint(65, 90))
random_char = random.choice([random_num, random_low_alpha, random_upper_alpha])
return random_char
def get_code_and_image(self):
"""
生成验证码与动画帧数
:return:
"""
code_str_list = []
for _ in range(self.code_count):
s = self.get_random_str()
code_str_list.append(s)
bg_color = self.get_random_color()
frame_list = []
for item in range(self.frame_count):
image = Image.new('RGB', (self.width, self.height), bg_color)
draw = ImageDraw.Draw(image)
path = os.path.join(os.getcwd(), "cabourgot-bold.otf")
font = ImageFont.truetype(path, size=self.font_size)
for i, code in enumerate(code_str_list):
v = random.randint(0, 10)
x = random.randint(2, 22)
draw.text((x + i * 40, v), code, self.get_random_color(), font=font)
# 噪点噪线
# 划线
for i in range(self.line_count):
x1 = random.randint(0, self.width)
x2 = random.randint(0, self.width)
y1 = random.randint(0, self.height)
y2 = random.randint(0, self.height)
draw.line((x1, y1, x2, y2), fill=self.get_random_color())
# 画点
for i in range(self.point_count):
draw.point([random.randint(0, self.width), random.randint(0, self.height)],
fill=self.get_random_color())
x = random.randint(0, self.width)
y = random.randint(0, self.height)
draw.arc((x, y, x + 4, y + 4), 0, 90, fill=self.get_random_color())
f = BytesIO()
image.save(f, "png")
data = f.getvalue()
f.close()
data = imageio.imread(data, format="png")
frame_list.append(data)
return frame_list, "".join(code_str_list)
if __name__ == "__main__":
img = GifCodeImage()
frame_list, code_str = img.get_code_and_image()
imageio.mimsave("code.gif", frame_list, 'GIF', duration=0.35)
print(code_str)
不多说,效果如下