Python验证码生成

最新推荐文章于 2024-07-08 14:04:57 发布

原创最新推荐文章于 2024-07-08 14:04:57 发布 · 1.4k 阅读

2 ·

CC 4.0 BY-SA版权

Python 专栏收录该内容

24 篇文章

订阅专栏

from PIL import Image,ImageDraw,ImageFont,ImageFilter
import random
class checkcode:
    def __init__(self,**kwargs):
        self._width=kwargs['width']
        self._height=kwargs['height']
        self._checkcode=None
    @staticmethod
    def codegnertor(x=1):#默认为0-9 a-z
        source=['1','2','3','4','5','6','7','8','9','0','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']
        if x:
            t=random.sample(source,4)
        else:
            t=random.sample(source[:9],4)
        t="".join(t)
        return t
    def picgenertor(self):
        image = Image.new('RGB', (self._width, self._height), (255, 255, 255))
        font = ImageFont.truetype('C:\Windows\Fonts\Arial.ttf', 36)
        draw = ImageDraw.Draw(image)
        def rndColor():
            return (random.randint(64, 255), random.randint(64, 255), random.randint(64, 255))
        def rndColor2():
            return (random.randint(32, 127), random.randint(32, 127), random.randint(32, 127))

        for x in range(self._width):
            for y in range(self._height):
                draw.point((x, y), fill=rndColor())
        for t in range(4):
            draw.text((self._width/4 * t + self._width/12, 10), self._checkcode[t], font=font, fill=rndColor2())
        begin = (0, random.randint(0, self._height))
        end = (self._width, random.randint(0, self._height))
        draw.line([begin, end], fill=(0,0,0), width=3)
        image = image.filter(ImageFilter.EDGE_ENHANCE_MORE)
        #image = image.transform((self._width + 30, self._height + 10), Image.AFFINE, (1, -0.3, 0, -0.1, 1, 0),Image.BILINEAR)  # 创建扭曲
        return image

    def __getitem__(self, key):
        if(key=='picture'):
            self._checkcode=self.codegnertor()
            return self.picgenertor()
        elif(key=="size"):
            return self._width,self._height
        elif(key=='checkcode'):
            return self._checkcode
        else:
            raise Exception('the key %s is valiad!'%key)
    def __str__(self):
        return  'code genertor'

调用方法:

x=checkcode(width=240,height=60)#width,height为验证码图片
s=checkcode.codegenertor()#随机生成验证码 x--->1 0-9 A--Z   x--->0 0-9

pic=x['picture']#image对象

pic.save(path)#储存图片

code=x['checkcode']#获取图片所含有的验证码