Инструментирование базы данных

Для помощи в понимании и управлении запросами, выполняемыми вашим кодом, Django предоставляет возможность установки обёртки вокруг выполнения запросов к базе данных. Например, обёртки могут подсчитывать запросы, измерять время выполнения запроса, регистрировать запросы или даже предотвращать выполнение запроса (например, чтобы убедиться, что запросы не выполняются во время рендеринга шаблона с предварительно полученными данными).

Обёртки моделируются по аналогии с средством-посредником – они являются вызовами, принимающими другой вызов в качестве одного из своих аргументов. Они вызывают этот вызов для выполнения (возможно, обернутого) запроса к базе данных, и могут делать всё, что захотят, вокруг этого вызова. Однако они создаются и устанавливаются пользовательским кодом, поэтому им не нужен отдельный фабричный метод, как у средств-посредников.

Установка обёртки выполняется в контекстном менеджере – таким образом, обёртки временные и специфичные для некоторого потока в вашем коде.

Как упоминалось выше, примером обёртки является блокировщик выполнения запроса. Он может выглядеть так:

def blocker(*args):
    raise Exception("No database access allowed here.")

И он будет использоваться в представлении для блокировки запросов из шаблона следующим образом:

from django.db import connection
from django.shortcuts import render


def my_view(request):
    context = {...}  # Code to generate context with all data.
    template_name = ...
    with connection.execute_wrapper(blocker):
        return render(request, template_name, context)

Параметры, отправляемые в обёртки, следующие:

• execute – вызов, который должен быть вызван с остальными параметрами для выполнения запроса.
• sql – str, SQL запрос, который должен быть отправлен в базу данных.
• params – список/кортеж значений параметров для SQL команды или список/кортеж списков/кортежей, если обернутый вызов является executemany().
• many – bool, указывающий, является ли окончательно вызванный вызов execute() или executemany() (и ожидается ли params быть последовательностью значений или последовательностью последовательностей значений).
• context – словарь с дополнительными данными о контексте вызова. Он включает в себя соединение и курсор.

Используя параметры, немного более сложная версия блокировщика может включать имя соединения в сообщение об ошибке:

def blocker(execute, sql, params, many, context):
    alias = context["connection"].alias
    raise Exception("Access to database '{}' blocked here".format(alias))

Для более полного примера, регистрировщик запросов может выглядеть так:

import time


class QueryLogger:
    def __init__(self):
        self.queries = []

    def __call__(self, execute, sql, params, many, context):
        current_query = {"sql": sql, "params": params, "many": many}
        start = time.monotonic()
        try:
            result = execute(sql, params, many, context)
        except Exception as e:
            current_query["status"] = "error"
            current_query["exception"] = e
            raise
        else:
            current_query["status"] = "ok"
            return result
        finally:
            duration = time.monotonic() - start
            current_query["duration"] = duration
            self.queries.append(current_query)

Для использования этого, вы создаёте объект регистрировщика и устанавливаете его как обёртку:

from django.db import connection

ql = QueryLogger()
with connection.execute_wrapper(ql):
    do_queries()
# Now we can print the log.
print(ql.queries)