使用python做验证码图片

来源：互联网发布：自我认知知乎编辑：程序博客网时间：2024/06/05 23:47
在网络应用中，验证码常常作为一个必备的手段，用来避免机器人恶意注册，保证坐在浏览器前的是一个人。

下面是一个实战项目的结果。
#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)
    
        
    