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

时间：2024-12-20 21:35:44

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

相关文章