from PIL import Image

from PIL import ImageDraw

from PIL import ImageFont

from io import BytesIO

import random

class ValidCodeImg:

"""

生成一个经过降噪后的随机验证码的图片

:return 生成图片的bytes类型的数据

"""

def __init__(self, width=175, height=35, code_count=5,

font_size=32, point_count=10, line_count=3,

img_format='png'):

# width: 图片宽度 单位px

self.width = width

# height: 图片高度 单位px

self.height = height

# code_count: 验证码个数

self.code_count = code_count

# font_size: 字体大小

self.font_size = font_size

# point_count: 噪点个数

self.point_count = point_count

# line_count: 划线个数

self.line_count = line_count

# img_format: 图片格式

self.img_format = img_format

@staticmethod

def getrandomcolor():

"""获取一个随机颜色-(r,g,b)格式"""

cor_r = random.randint(0, 255)

cor_g = random.randint(0, 255)

cor_b = random.randint(0, 255)

return cor_r, cor_g, cor_b

@staticmethod

def getrandomstr():

"""获取一个随机字符串"""

# 获取一个0-9的随机书

num_str = str(random.randint(0, 9))

# 获取一个A-Z的随机大写字母,65-90是大写字母的ascii码,通过chr转换成对应的字符

upper_str = chr(random.randint(65, 90))

# 获取一个a-z的随机小写字母,97-122是小写的ascii码,通过chr转换成对应的字符

lower_str = chr(random.randint(97, 122))

# 通过random模块的choice方法随机从以上的随机数字,大写字母和小写字母中获取一个字符,并返回

random_str = random.choice((num_str, upper_str, lower_str))

return random_str

def getvalidcodeimg(self):

"""获取一个随机验证码图片"""

# 通过Image.new(rgb格式, (图片的宽，图片的高), 颜色)获取一个Image对象

img_obj = Image.new('RGB', (self.width, self.height), self.getrandomcolor())

print(img_obj)

# 通过ImageDraw.Draw(Image对象)获取一个画笔对象

draw = ImageDraw.Draw(img_obj)

# 通过ImageFont.truetype(字体文件路径, 字体大小)获取一个字体对象

font = ImageFont.truetype(r'\static\font\STXINGKA.TTF', size=self.font_size)

tmp_list = [] # 用于存储随机字符串

for i in range(self.code_count):

random_char = self.getrandomstr()

# 在图片上写入得到的随机字符串，这样字符之间才会有距离，不会叠加

# 通过画笔对象的text方法，即 draw.text((x, y), 字符串, 颜色, 字体)

draw.text((10 + i * 32, -2), random_char, self.getrandomcolor(), font=font)

# 将随机字符串保存

tmp_list.append(random_char)

# 将所有的字符串拼接，稍后返回，用于校验用户录入的记录是否正确

valid_str = ''.join(tmp_list)

# 噪线的生成

for i in range(self.line_count):

# 噪线的起点横坐标和纵坐标

x1 = random.randint(0, self.width)

y1 = random.randint(0, self.height)

# 噪线的终点横坐标和纵坐标

x2 = random.randint(0, self.width)

y2 = random.randint(0, self.height)

# 通过画笔对象draw.line((起点的xy, 终点的xy), fill='颜色')来划线

draw.line((x1, y1, x2, y2), fill=self.getrandomcolor())

# 噪点的生成

for i in range(self.point_count):

draw.point([random.randint(0, self.width), random.randint(0, self.height)], fill=self.getrandomcolor())

x = random.randint(0, self.width)

y = random.randint(0, self.height)

draw.arc((x, y, x + 4, y + 4), 0, 40, fill=self.getrandomcolor())

# 将生成的图片保存到内存中

f = BytesIO()

img_obj.save(f, self.img_format)

data = f.getvalue()

f.close()

return data, valid_str

# # 测试

# if __name__ == '__main__':

# img = ValidCodeImg()

#

# data, valid_str = img.getvalidcodeimg()

# print(valid_str)

#

# with open(r'E:\Desktop\code_test.png', 'wb') as f:

# f.write(data)