python工具类之验证码

# !/usr/bin/env python
# -*-coding:utf-8 -*-
import random

from PIL import Image, ImageDraw, ImageFont


def generate_code():
"""生成4位字母数字混合验证码"""
code = ''
for i in range(5):
if i % 2 == 0: # 偶数位生成数字
num = str(random.randint(0, 20))
else: # 奇数位生成字母
num = chr(random.randint(65, 90)) # ASCII码65-90对应字母A-Z
code += num
return code


def draw_lines(draw, num):
"""绘制干扰线"""
for i in range(num):
x1 = random.randint(0, width)
y1 = random.randint(0, height)
x2 = random.randint(0, width)
y2 = random.randint(0, height)
draw.line((x1, y1, x2, y2), fill='black', width=1)

return draw_lines


if __name__ == '__main__':
# 图片大小120x50
width, height = 120, 50
# 创建图片对象
im = Image.new('RGB', (width, height), color='white')
# 创建画笔对象
draw = ImageDraw.Draw(im)
# 生成验证码
code = generate_code()
# 画干扰线
draw_lines(draw, 5)
# 验证码字体大小40
font = ImageFont.truetype('simsun.ttc', 40)
# 在图片上写入验证码字符串
for i in range(4):
draw.text((30 * i + 10, 10), text=code[i], fill='black', font=font)
# 在终端打印验证码
print(f'您的验证码是:{code}')
# 保存验证码图片
im.save('code.jpg')