python 之 pil生成验证码图片

#!/usr/bin/python
#coding=utf-8
import Image,ImageDraw,ImageFont,os,string,random,ImageFilter 
def initChars(): 
	""" 
	允许的字符集合，初始集合为数字、大小写字母 
	usage: initChars() 
	param: None 
	return: list 
	返回允许的字符集和 
	for: picChecker类初始字符集合 
	todo: Nothing 
	""" 
	nums = [str(i) for i in range(10)] 
	letterCase = [ 
	'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'j', 'k', 
	'm', 'n', 'p', 'q', 'r', 's', 't', 'u', 'v', 
	'w', 'x', 'y'
	] 
	upperCase = [ 
	'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'J', 'K', 
	'M', 'N', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 
	'W', 'X', 'Y'
	] 
	return(nums+letterCase+upperCase)
	
class picChecker(): 
	""" 
	图片验证代码： 
	1) 用户注册需填写图片验证码，以阻止机器人注册 
	2) 图片验证码字符数为 4 位(大小写字母与数字，不区分大小写)。 
	用户如果没有填写验证码或没有填写正确的验证码， 
	页面友好性提示用户填写(同时程序方面也做相应限制) 
	usage: pc = picChecker().createChecker() 
	param: 很多，如下 
		chars 允许的字符集合， 
			类型 list 
			默认值 initChars() 
			例子 ['1','2','3'] 
		length 字符串长度 
			类型 integer 
			默认值 4 
		size 图片大小 
			类型 tutle 
			默认值 (120,30) 
			例子 (120,30) 
		fontsize 字体大小 
			类型 integer 
			默认值 25 
		begin 字符其实位置，即左上角位置 
			类型 tutle 
			默认值 (5,-2) 
		outputType 输出类型 
			类型 string 
			默认值 GIF 
			可选值 GIF JPEG TIFF PNG 
		mode 图片模式 
			类型 string 
			可选值 RGB L (还有其他模式，但只推荐这2种) 
			默认值 RGB 
		backgroundColor 背景色 
		foregroundColor 前景色 
			当mode=RGB时，backgroundColor,foregroundColor为tutle类型 
			取值为(integer,integer,integer) 
			表示RGB颜色值 
			当mode=L时，backgroundColor,foregroundColor为数字，表示黑白模式 
			取值为0-255 
			表示灰度 
		fonttype 字体路径 
			类型 string 
			默认值 "simsum.ttc" 
		jamNum 干扰线条数 
			类型 (int1,int1) 
			int1 干扰线条数下限，包含 
			int2 干扰线条数上线，包含 
		pointBorder 散点噪音 
			构造方法：对每个像素点使用随机函数确定是否在该像素上画散点噪音 
			类型 (int1,int2) 
			int1越大 散点越多 
			int2越大 散点越少 
	return: [picCheckerStr,pic] 
	picCheckerStr: 表示返回图片中对应的字符串,可用于session验证以及其他用途 
	pic : 返回的图片，类型为Image 
	for : 
	todo : Nothing 
	""" 
	#默认字体路径 
	#DEFAULT_FONT_PATH = os.path.join(os.path.dirname(__file__),'simsun.ttc').replace('\\','/') 
	def __init__(self,chars = initChars(),size = (120,30),fontsize = 18, 
				begin = (5,-2),outputType = 'GIF',mode = 'RGB' , 
				backgroundColor = (255,255,255), foregroundColor = (0,0,255), 
				fonttype = "ae_AlArabiya.ttf",length = 5,jamNum = (1,2), 
				pointBorder = (40,39)): 
		""" 
		初始化配置 
		""" 
		#验证码配置 
		#允许的字符串 
		self.chars = chars 
		#图片大小 
		self.size = size 
		#字符起始插入点 
		self.begin = begin 
		#字符串长度 
		self.length = length 
		#输出类型 
		self.outputType = outputType 
		#字符大小 
		self.fontsize = fontsize 
		#图片模式 
		self.mode = mode 
		#背景色 
		self.backgroundColor = backgroundColor 
		#前景色 
		self.foregroundColor = foregroundColor 
		#干扰线条数 
		self.jamNum = jamNum 
		#散点噪音界限 
		self.pointBorder = pointBorder 
		#字体库路径 
		self.fonttype = fonttype 
		#设置字体,大小默认为18 
		self.font = ImageFont.truetype(self.fonttype, self.fontsize)
	
	def getPicString(self): 
		""" 
		usage: getPicString() 
		return: string 
		for : 生成给定长度的随机字符串 
		todo: Nothing 
		""" 
		#初始化字符串长度 
		length = self.length 
		#初始化字符集合 
		chars = self.chars 
		#获得字符集合 
		selectedChars = random.sample(chars,length) 
		charsToStr = string.join(selectedChars,'') 
		return charsToStr
	
	def createChecker(self): 
		""" 
		usage: createChecker() 
		return: [str,pic] 
		str:对应的字符串 
		pic:对应的图片 
		for: 
		todo: 
		""" 
		#获得验证码字符串 
		randStr = self.getPicString() 
		#将字符串加入空格 
		randStr1 = string.join([i+" " for i in randStr],"") 
		#创建图形 
		im = Image.new(self.mode,self.size,self.backgroundColor) 
		#创建画笔 
		draw = ImageDraw.Draw(im) 
		#输出随机文本 
		draw.text(self.begin, randStr1, font=self.font,fill=self.foregroundColor) 
		#im = self.drawText(draw,randStr,im) 
		#干扰线 
		self.createJam(draw) 
		#散点噪音 
		self.createPoints(draw) 
		#图形扭曲 
		para = [1-float(random.randint(1,2))/100, 
			0, 
			0, 
			0, 
			1-float(random.randint(1,10))/100, 
			float(random.randint(1,2))/500, 
			0.001, 
			float(random.randint(1,2))/500 
			] 
		#print randStr,para 
		im = im.transform(im.size, Image.PERSPECTIVE,para) 
		#图像滤镜 
		im=im.filter(ImageFilter.EDGE_ENHANCE_MORE) 
		im.save("checker.gif",self.outputType)
		print randStr
		return [randStr,im]
		
	def createJam(self,draw): 
		""" 
		usage: 创建干扰线 
		para: draw 表示画笔 
		return: None 
		for: 
		todo: 
		""" 
		#干扰线条数 
		lineNum = random.randint(self.jamNum[0],self.jamNum[1]) 
		for i in range(lineNum): 
			begin = (random.randint(0,self.size[0]),random.randint(0,self.size[1])) 
			end = (random.randint(0,self.size[0]),random.randint(0,self.size[1])) 
			draw.line([begin,end],fill = (0,0,0))
		
	def createPoints(self,draw): 
		""" 
		usage: 创建散点噪音 
		para: draw 表示画笔 
		return: None 
		for: 
		todo: 
		""" 
		#散点噪音 
		for x in range(self.size[0]): 
			for y in range(self.size[1]): 
				flag = random.randint(0,self.pointBorder[0]) 
		if flag > self.pointBorder[1]: 
			draw.point((x,y),fill = (0,0,0)) 
			del flag
		
if __name__ == '__main__': 
	c=picChecker() 
	t=c.createChecker() 
	print(t)