from PIL import Image, ImageDraw, ImageFont
import random
import string
#所有可能的字符,主要是英文字母和数字
characters = string.ascii_letters+string.digits
#获取指定长度的字符串
def selectedCharacters(length):
'''length:the number of characters to show'''
result = ""
for i in range(length):
result += random.choice(characters)
print(result)
return result
def getColor():
'''get a random color'''
r = random.randint(0,255)
g = random.randint(0,255)
b = random.randint(0,255)
return (r,g,b)
def main(size=(200,100), characterNumber=6, bgcolor=(255,255,255)):
imageTemp = Image.new('RGB', size, bgcolor)
#设置字体和字号
font = ImageFont.truetype('./fonts/actionj.ttf', 48)
draw = ImageDraw.Draw(imageTemp)
text = selectedCharacters(characterNumber)
width, height = draw.textsize(text, font)
#绘制验证码字符串
offset = 2
for i in range(characterNumber):
offset += width//characterNumber
position = (offset, (size[1]-height)//2+random.randint(-10,10))
draw.text(xy=position, text=text[i], font=font, fill=getColor())
#对验证码图片进行简单变换,这里采用简单的点运算
imageFinal = Image.new('RGB', size, bgcolor)
pixelsFinal = imageFinal.load()
pixelsTemp = imageTemp.load()
for y in range(0, size[1]):
offset = random.randint(-1,1)
for x in range(0, size[0]):
newx = x+offset
if newx>=size[0]:
newx = size[0]-1
elif newx<0:
newx = 0
pixelsFinal[newx,y] = pixelsTemp[x,y]
draw = ImageDraw.Draw(imageFinal)
#绘制干扰噪点像素
for i in range(int(size[0]*size[1]*0.07)):
draw.point((random.randint(0,size[0]), random.randint(0,size[1])), fill=getColor())
#绘制干扰线条
for i in range(8):
start = (0, random.randint(0, size[1]-1))
end = (size[0], random.randint(0, size[1]-1))
draw.line([start, end], fill=getColor(), width=1)
#绘制干扰弧线
for i in range(8):
start = (-50, -50)
end = (size[0]+10, random.randint(0, size[1]+10))
draw.arc(start+end, 0, 360, fill=getColor())
#保存验证码图片
imageFinal.save("result.jpg")
imageFinal.show()
if __name__=="__main__":
main((200,100), 8, (255,255,255))