Django ORM性能优化 和 图片验证码

一，ORM性能相关

1. 关联外键, 只拿一次数据
    all_users = models.User.objects.all().values('name', 'age', 'role__name')
2. 查外键会访问多次数据库, 用的时候注意, 只拿自己表中的字段, 别跨表
    all_users = models.User.objects.all()
3.  select_related(外键/一对一)       -->连表查询, 也只拿一次数据
    all_users = models.User.objects.all().select_related('role')
    print(user.name, user.age, user.role.name)
4. prefetch_related('role')  --> 子查询(把第一张表的数据拿出来, 再拿第一张表的数据去查第二张表)
    all_users = models.User.objects.all().prefetch_related('role')
5. only('只拿要用的字段')  -->根据指定字段查一次
    all_users = models.User.objects.all().only('name')
    all_users = models.User.objects.all().defer('name') # 跟上面相反, 除了name字段其他都拿

二， 图片验证码

1， 随机生成五位数的验证码

import random


def get_vcode():
    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])
        tmp.append(r)
    return "".join(tmp)


if __name__ == '__main__':
    ret = get_vcode()
    print(ret)

2， 随机生成位数的图片验证码

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()