使用pillow, 生成一张图片验证码

Love The Way You Lie 2023-02-21 06:58 11阅读 0赞

pillow 博客文档

> [模块pillow使用][pillow]

安装

pip install pillow

**生成图片验证码**

> 图片验证码需要使用到字体，一般字体库为ttf后缀结尾的文件，可以网上自省下载，这里我提供一个分享的下载地址。  
> 链接：https://pan.baidu.com/s/1MnzaZuxmEPGkNpMBMKjuEA  
> 提取码：nunu

import random
    from PIL import Image,ImageDraw,ImageFont,ImageFilter
    
    def check_code(width=120, height=30, char_length=5, font_file='xueyi.ttf', font_size=28):
        code = []
        img = Image.new(mode='RGB', size=(width, height), color=(255, 255, 255))
        draw = ImageDraw.Draw(img, mode='RGB')
     
        def rndChar():
            """ 生成随机字母 :return: """
            return chr(random.randint(65, 90))
     
        def rndColor():
            """ 生成随机颜色 :return: """
            return (random.randint(0, 255), random.randint(10, 255), random.randint(64, 255))
     
        # 写文字
        font = ImageFont.truetype(font_file, font_size)
        for i in range(char_length):
            char = rndChar()
            code.append(char)
            h = random.randint(0, 4)
            draw.text([i * width / char_length, h], char, font=font, fill=rndColor())
     
        # 写干扰点
        for i in range(40):
            draw.point([random.randint(0, width), random.randint(0, height)], fill=rndColor())
     
        # 写干扰圆圈
        for i in range(40):
            draw.point([random.randint(0, width), random.randint(0, height)], fill=rndColor())
            x = random.randint(0, width)
            y = random.randint(0, height)
            draw.arc((x, y, x + 4, y + 4), 0, 90, fill=rndColor())
     
        # 画干扰线
        for i in range(5):
            x1 = random.randint(0, width)
            y1 = random.randint(0, height)
            x2 = random.randint(0, width)
            y2 = random.randint(0, height)
     
            draw.line((x1, y1, x2, y2), fill=rndColor())
     
        img = img.filter(ImageFilter.EDGE_ENHANCE_MORE)
        return img,''.join(code)
     
     
    if __name__ == '__main__':
        # 1. 直接打开
        img,code = check_code()
        img.show()
         
        # 2. 写入文件
        # img,code = check_code()
        # with open('code.png','wb') as f:
        # img.save(f,format='png')
     
        # 3. 写入内存(Python3)
        # from io import BytesIO
        # stream = BytesIO()
        # img.save(stream, 'png')
        # stream.getvalue()
     
        # 4. 写入内存（Python2）
        # import StringIO
        # stream = StringIO.StringIO()
        # img.save(stream, 'png')
        # stream.getvalue()

[pillow]: https://www.cnblogs.com/wupeiqi/acticles/5812291.html