使用python的Pillow模块生成验证码


参考资料
准备

1from PIL import Image, ImageDraw, ImageFont, ImageFilter
2
3_letter_cases = "abcdefghjkmnpqrstuvwxy" # 去除可能干扰的i,l,o,z
4_upper_cases = _letter_cases.upper() # 大写字母
5_numbers = ''.join(map(str, range(3, 10))) # 数字
6init_chars = ''.join((_letter_cases, _upper_cases, _numbers))
7fontType="/usr/share/fonts/TTF/DejaVuSans.ttf"

1.创建图像

 1def create_validate_code(self,size=(120, 30),
 2                        chars=init_chars,
 3                        img_type="PNG",
 4                        mode="RGB",
 5                        bg_color=(255, 255, 255),
 6                        fg_color=(0, 0, 255),
 7                        font_size=18,
 8                        font_type=fontType,
 9                        length=4,
10                        draw_lines=True,
11                        n_line=(1, 2),
12                        draw_points=True,
13                        point_chance = 2):
14
15    width, height = size # 图像的宽,高
16    img = Image.new(mode, size, bg_color) # 创建新图像
17    img.save("validate.png", "PNG") # 存储图片
18if __name__ == '__main__':
19    create_validate_code()

2.绘制干扰线
在创建新图像后添加

1if draw_lines:
2    self.create_lines(draw,n_line,width,height) # 绘制干扰线

干扰线函数

1def create_lines(self,draw,n_line,width,height):
2    line_num = random.randint(n_line[0],n_line[1]) # 干扰线条数
3    for i in range(line_num):
4        # 起始点
5        begin = (random.randint(0, width), random.randint(0, height))
6        #结束点
7        end = (random.randint(0, width), random.randint(0, height))
8        draw.line([begin, end], fill=(0, 0, 0))

3.绘制干扰点
在绘制干扰线后添加

1if draw_points:
2    self.create_points(draw,point_chance,width,height) #绘制干扰点

干扰点函数

1def create_points(self,draw,point_chance,width,height):
2    chance = min(100, max(0, int(point_chance))) # 大小限制在[0, 100]
3
4    for w in range(width):
5        for h in range(height):
6            tmp = random.randint(0, 100)
7            if tmp > 100 - chance:
8                draw.point((w, h), fill=(0, 0, 0))

4.生成验证码
在绘制干扰点之后添加

1strs = self.create_strs(draw,chars,length,font_type,
2                         font_size,width,height,fg_color)

验证码函数

 1def create_strs(self,draw,chars,length,font_type, 
 2                font_size,width,height,fg_color):
 3    c_chars = random.sample(chars, length)
 4    strs = ' %s ' % ' '.join(c_chars) # 每个字符前后以空格隔开
 5
 6    font = ImageFont.truetype(font_type, font_size)
 7    font_width, font_height = font.getsize(strs)
 8
 9    draw.text(((width - font_width) / 3, (height - font_height) / 3),strs,\
10              font=font, fill=fg_color)

ok,简单的验证码就生成好了

作者: honmaple
链接: https://honmaple.me/articles/2015/12/使用python的Pillow模块生成验证码.html
版权: CC BY-NC-SA 4.0 知识共享署名-非商业性使用-相同方式共享4.0国际许可协议
wechat
alipay

加载评论