首先
pip install pillow
然后
from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont
import random
class ValidCodeImg:
def __init__(self, width=175, height=40, code_count=6, font_size=28, 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
'''
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_string():
'''獲取一個隨機字符串,每個字符的顏色也是隨機的'''
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的字體文件的目錄,以及字體的大小
font = ImageFont.truetype("kumo.ttf", size=self.font_size)
temp = []
for i in range(self.code_count):
# 循環5次,獲取5個隨機字符串
random_char = self.get_random_string()
# 在圖片上一次寫入得到的隨機字符串,參數是:定位,字符串,顏色,字體
draw.text((4 + i * 30, 10), random_char, self.get_random_color(), font=font)
# 保存隨機字符,以供驗證用戶輸入的驗證碼是否正確時使用
temp.append(random_char)
valid_str = "".join(temp)
# 噪點噪線
# 划線
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())
# 在內存生成圖片
from io import BytesIO
f = BytesIO()
image.save(f, self.img_format)
data = f.getvalue()
f.close()
return data, valid_str
def create_img(self):
data, valid_str = self.get_valid_code_img()
f = open('test.png', 'wb')
f.write(data)
f.close()
return valid_str
if __name__ == '__main__':
img = ValidCodeImg()
strs = img.create_img()
print(strs)
封裝好了,直接拿去用,另外附字體
kumo.ttf
百度下下來,然后放在文件目錄同級目錄下即可