验证图片代码

import random
from PIL import Image, ImageDraw, ImageFont

def random_color():
return random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)

def vcode():
    # 创建一个随机颜色的图片对象
    image_obj = Image.new(
        "RGB",
        (250, 35),
        random_color()
    )

with open("static/imgs/vcode.png", "wb") as f1:
image_obj.save(f1, format="PNG")

def vcode():
    # 创建一个随机颜色的图片对象
    image_obj = Image.new(
        "RGB",
        (250, 35),
        random_color()
    )

    # 在该图片对象上生成一个画笔对象
    draw_obj = ImageDraw.Draw(image_obj)
    # 加载一个字体对象
    font_obj = ImageFont.truetype('static/font/kumo.ttf', 28)
    tmp = []
    for i in range(5):
        l = chr(random.randint(97, 122)) # 生成随机的小写字母
        u = chr(random.randint(65, 90)) # 生成随机的大写字母
        n = str(random.randint(0, 9)) # 生成一个随机的数字
        # 从上面三个随机选一个
        r = random.choice([l, u, n])
        draw_obj.text((i * 45 + 30, 0), r, fill=random_color(), font=font_obj, )

with open("static/imgs/vcode.png", "wb") as f1:
image_obj.save(f1, format="PNG")

# 专门返回验证码图片的视图函数
def vcode(request):
from PIL import Image, ImageDraw, ImageFont

    # 定义一个生成随机颜色代码的函数
    def random_color():
        return random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)

    # 创建一个随机颜色的图片对象
    image_obj = Image.new(
        "RGB",
        (250, 35),
        random_color()
    )
    # 在该图片对象上生成一个画笔对象
    draw_obj = ImageDraw.Draw(image_obj)
    # 加载一个字体对象
    font_obj = ImageFont.truetype('static/font/kumo.ttf', 28)
    tmp = []
    for i in range(5):
        l = chr(random.randint(97, 122)) # 生成随机的小写字母
        u = chr(random.randint(65, 90)) # 生成随机的大写字母
        n = str(random.randint(0, 9)) # 生成一个随机的数字
        # 从上面三个随机选一个
        r = random.choice([l, u, n])
        # 将选中过的那个字符写到图片上
        draw_obj.text((40 * i + 30, 0), r, fill=random_color(), font=font_obj)
        tmp.append(r)

    # # 加干扰线
    # width = 250 # 图片宽度（防止越界）
    # height = 35
    # for i in range(5):
    #     x1 = random.randint(0, width)
    #     x2 = random.randint(0, width)
    #     y1 = random.randint(0, height)
    #     y2 = random.randint(0, height)
    #     draw_obj.line((x1, y1, x2, y2), fill=random_color())
    #
    # # 加干扰点
    # for i in range(40):
    #     draw_obj.point([random.randint(0, width), random.randint(0, height)], fill=random_color())
    #     x = random.randint(0, width)
    #     y = random.randint(0, height)
    #     draw_obj.arc((x, y, x+4, y+4), 0, 90, fill=random_color())

    v_code = "".join(tmp).upper()
    # 将生成的验证码保存
    request.session["v_code"] = v_code

    # with open("static/images/vcode.png", "wb") as f1:
    #     image_obj.save(f1, format="PNG")
    #
    # with open("static/images/vcode.png", "rb") as f:
    #     img_data = f.read()
    # 直接在内存中保存图片替代io操作
    from io import BytesIO
    f1 = BytesIO()
    image_obj.save(f1, format="PNG")
    img_data = f1.getvalue()
    return HttpResponse(img_data, content_type="image/png")

vcode()

猜你喜欢