import random
from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont
from io import BytesIO


class GetValidImg(object):
    def __init__(self, width=300, height=60, code_count=5, font_size=32,
                 point_count=20, line_count=3, img_format='png'):
        """
        可以生成一个经过降噪后的随机验证码的图片
        :param width: 图片宽度 单位px
        :param height: 图片高度 单位px
        :param code_count: 验证码个数
        :param font_size: 字体大小
        :param point_count: 噪点个数
        :param line_count: 划线个数
        :param img_format: 图片格式
        :return 生成的图片的bytes类型的data

        FreeMono.ttf为字体文件,需要指定字体文件路径
        """
        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.img_format = img_format

    @staticmethod
    def get_random_color():
        """获取一个随机颜色(r,g,b)格式的"""
        c1 = random.randint(0, 255)
        c2 = random.randint(0, 255)
        c3 = random.randint(0, 255)
        return c1, c2, c3

    @staticmethod
    def get_random_str():
        """获取一个随机字符串，每个字符的颜色也是随机的"""
        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_valid_code_img(self):
        # 获取一个Image对象，参数分别是RGB模式。宽150，高30，随机颜色
        image = Image.new('RGB', (self.width, self.height), self.get_random_color())
        # 获取一个画笔对象，将图片对象传过去
        draw = ImageDraw.Draw(image)
        # 获取一个font字体对象参数是ttf的字体文件的目录，以及字体的大小
        # 这里不能使用相对路径,否则部署代码时会出错,所以要借助os库实现绝对路径
        import os
        base_dir = os.path.dirname(os.path.abspath(__file__))
        font_path = os.path.join(base_dir, 'FreeMono.ttf')
        font = ImageFont.truetype(font_path, size=self.font_size)
        # 保存随机字符，以供验证用户输入的验证码是否正确时使用
        get_valid_num = ""
        for i in range(self.code_count):
            # 循环5次，获取5个随机字符串
            random_char = self.get_random_str()
            # 在图片上一次写入得到的随机字符串,参数是：定位(宽度 高度定位)，字符串，颜色，字体
            draw.text((10 + i * self.height, 12), random_char, self.get_random_color(), font=font)
            # 保存随机字符，以供验证用户输入的验证码是否正确时使用
            get_valid_num += random_char

        # 噪点噪线
        # 划线
        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())

        # 保存本地或内存加载二选一
        # 保存到硬盘，名为test.png格式为png的图片
        image.save(open('test.png', 'wb'), 'png')

        # 不需要在硬盘上保存文件，直接在内存中加载就可以
        io_obj = BytesIO()
        # 将生成的图片数据保存在io对象中
        image.save(io_obj, "png")
        # 从io对象里面取上一步保存的数据
        img_data = io_obj.getvalue()
        return img_data, get_valid_num


if __name__ == '__main__':
    img = GetValidImg()
    img_show, get_valid_num = img.get_valid_code_img()
    print(get_valid_num)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103