生成一个验证码最主要的就是引用PIL(Python Imaging Library);PIL是Python平台事实上的图像处理标准库了。
生成验证码需要的一个数据的颜色,随机的数字或者字母
一.随机颜色
import random
def getRandomColor():
'''获取一个随机颜色(r,g,b)格式的'''
c1 = random.randint(0,255)
c2 = random.randint(0,255)
c3 = random.randint(0,255)
#print '=====',(c1,c2,c3)
return (c1,c2,c3)
二.随机数字和字母
import random
def getRandomStr():
'''获取一个随机字符串,每个字符的颜色也是随机的'''
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])
print '---->',random_char
return random_char
三.设置图片的规格数据等
def __init__(self,width=150,height=30,code_count=5,font_size=24,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: 图片格式
'''
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
四。得到验证码图片
def getValidCodeImg(self):
# 获取一个Image对象,参数分别是RGB模式。宽150,高30,随机颜色
#print 'ssssss',self.width
image = Image.new('RGB',(self.width,self.height),getRandomColor())
# 获取一个画笔对象,将图片对象传过去
draw = ImageDraw.Draw(image)
# 获取一个font字体对象参数是ttf的字体文件的目录,以及字体的大小
font=ImageFont.truetype("/home/dawn/ssvn/PIL/font1/simsun.ttf",size=self.font_size)
temp = []
for i in range(self.code_count):
# 循环5次,获取5个随机字符串
random_char = getRandomStr()
# 在图片上一次写入得到的随机字符串,参数是:定位,字符串,颜色,字体
draw.text((10+i*30,-2),random_char , getRandomColor(),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=getRandomColor())
# 画点
for i in range(self.point_count):
draw.point([random.randint(0, self.width), random.randint(0, self.height)], fill=getRandomColor())
x = random.randint(0, self.width)
y = random.randint(0, self.height)
draw.arc((x, y, x + 4, y + 4), 0, 90, fill=getRandomColor())
# 在内存生成图片
from io import BytesIO
f = BytesIO()
image.save(f, self.img_format)
data = f.getvalue()
f.close()
return data,valid_str
最终代码:
#!/usr/bin/python
# -*- coding: UTF-8 -*-
from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont
import random
def getRandomColor():
'''获取一个随机颜色(r,g,b)格式的'''
c1 = random.randint(0,255)
c2 = random.randint(0,255)
c3 = random.randint(0,255)
#print '=====',(c1,c2,c3)
return (c1,c2,c3)
def getRandomStr():
'''获取一个随机字符串,每个字符的颜色也是随机的'''
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])
print '---->',random_char
return random_char
class ValidCodeImg:
def __init__(self,width=150,height=30,code_count=5,font_size=24,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: 图片格式
'''
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
def getValidCodeImg(self):
# 获取一个Image对象,参数分别是RGB模式。宽150,高30,随机颜色
#print 'ssssss',self.width
image = Image.new('RGB',(self.width,self.height),getRandomColor())
# 获取一个画笔对象,将图片对象传过去
draw = ImageDraw.Draw(image)
# 获取一个font字体对象参数是ttf的字体文件的目录,以及字体的大小
font=ImageFont.truetype("/home/dawn/ssvn/PIL/font1/simsun.ttf",size=self.font_size)
temp = []
for i in range(self.code_count):
# 循环5次,获取5个随机字符串
random_char = getRandomStr()
# 在图片上一次写入得到的随机字符串,参数是:定位,字符串,颜色,字体
draw.text((10+i*30,-2),random_char , getRandomColor(),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=getRandomColor())
# 画点
for i in range(self.point_count):
draw.point([random.randint(0, self.width), random.randint(0, self.height)], fill=getRandomColor())
x = random.randint(0, self.width)
y = random.randint(0, self.height)
draw.arc((x, y, x + 4, y + 4), 0, 90, fill=getRandomColor())
# 在内存生成图片
from io import BytesIO
f = BytesIO()
image.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)
f = open('test.png', 'wb')
f.write(data)
注意:font=ImageFont.truetype("/home/dawn/ssvn/PIL/font1/simsun.ttf",size=self.font_size) 中文字需要自己文件字体