[python] 使用PIL生成图片验证码

我就是我 2022-02-28 08:56 165阅读 0赞

###  使用PIL库生成验证图片 ###

#-*- coding: UTF-8 -*-
    import random
    from io import BytesIO
    from PIL import Image, ImageDraw, ImageFont
    
    def check_code(width=120, height=30, char_length=5, font_file='kumo.ttf', font_size=28):
        f = BytesIO()
        img = Image.new(mode='RGB', size=(width, height),
                        color=(random.randint(50, 250), random.randint(50, 255), random.randint(50, 255)))
        draw = ImageDraw.Draw(img, mode='RGB')
    
        char_list = []
        # 画字
        for i in range(char_length):
            char = random.choice([chr(random.randint(65, 90)), str(random.randint(1, 9)), chr(random.randint(97, 122)), ])
            # 第一个参数：表示字体路径
            # 第二个参数：表示字体大小
            #font = ImageFont.truetype("blog/static/fonts/"+font_file, font_size)
            font = ImageFont.load_default().font #默认字体也可以用系统自带字体
            # draw.text()
            # 第一个参数：表示起始坐标
            # 第二个参数：表示写入内容
            # 第三个参数：表示颜色
            # 第四个参数：表示字体
            draw.text([i * 24, 0], char, (random.randint(200, 255), random.randint(0, 50), random.randint(0, 50)),
                      font=font)
            char_list.append(char)
    
        def rndColor():
            """
            生成随机颜色
            :return:
            """
            return (random.randint(0, 255), random.randint(10, 255), random.randint(64, 255))
    
        # 写干扰点
        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()
            # 第一个参数：表示起始坐标和结束坐标(圆要画在中间)
            # 第二个参数：表示开始角度
            # 第三个参数：表示结束角度
            # 第四个参数：表示颜色
            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())
    
        #保存到本地
        #with open("code.png", 'wb') as f:
        #    img.save(f, format="png")
        
        #保存到内存里
        img.save(f, "png")
        data = f.getvalue()
        s_code = ''.join(char_list)
        print(data)
        return data,s_code
    
    data,s_code = check_code()

###  调用方法 ###

def get_valid_img(request):
        '''
        登录和注册的验证码
        :param request:
        :return:
        '''
        data,s_code = check_code()
        request.session["valid_code"] = s_code
        return HttpResponse(data)