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', 'i', 'j', 'k',
'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v',
'w', 'x', 'y', 'z'
]
upperCase = [
'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K',
'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V',
'W', 'X', 'Y', 'Z',
]
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 = 25,
begin = (5,-2),outputType = 'GIF',mode = 'RGB' ,
backgroundColor = (255,255,255), foregroundColor = (0,0,255),
fonttype = "simsun.ttc",length = 4,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.jpg",self.outputType)
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)