Python简易验证码生成程序

Python简易验证码生成程序

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)

    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('c:\\windows\\fonts\\TIMESBD.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))