Class that supports the function of decorator, iterator and context manager for measuring the time and number of database queries

Project		Status
CI/CD
Quality
Package
Meta

Class that supports the function of decorator, iterator and context manager for measuring the time and number of database queries

Install

1. Install package

   pip install capture-db-queries

About decorator

CaptureQueries class as decorator can call the body of the decorated function or class as iterator can run code inside for loop the specified number of times for multiple measurements, it can validate the total number of queries. The functionality of the classic context manager is also available.

• Optional parameters:
  • assert_q_count: The expected number of database requests is otherwise "AssertionError: N not less than or equal to N queries"
  • number_runs: The number of runs of the test function / test for loop
  • verbose: Displaying the final results of the test measurements
  • advanced_verb: Displaying the result of each test measurement
  • auto_call_func: Autorun of the decorated function (without arguments)
  • queries: Displaying raw SQL queries to the database
  • explain: Displaying additional information about each request (has no effect on the orig. requests)
  • explain_opts: For more information about the parameters for explain, see the documentation for your DBMS
  • connection: Connecting to your database, by default: django.db.connection

Usage examples

from capture_db_queries.decorators import CaptureQueries

for ctx in CaptureQueries(number_runs=2, advanced_verb=True):
    response = self.client.get(url)

# OR

@CaptureQueries(number_runs=2, advanced_verb=True)
def test_request():
    response = self.client.get(url)

# OR

# NOTE: The with context manager does not support multi-launch number_runs > 1
with CaptureQueries(number_runs=1, advanced_verb=True) as ctx:
    response = self.client.get(url)

>>> Test №1 | Queries count: 10 | Execution time: 0.04s
>>> Test №2 | Queries count: 10 | Execution time: 0.04s
>>> Tests count: 2  |  Total queries count: 20  |  Total execution time: 0.08s  |  Median time one test is: 0.041s  |  Vendor: sqlite

# Example of output when using queries and explain:

for _ in CaptureQueries(advanced_verb=True, queries=True, explain=True):
    list(Reporter.objects.filter(pk=1))
    list(Article.objects.filter(pk=1))

>>> Test №1 | Queries count: 2 | Execution time: 0.22s
>>>
>>>
>>> №[1] time=[0.109] explain=['2 0 0 SEARCH TABLE tests_reporter USING INTEGER PRIMARY KEY (rowid=?)']
>>> SELECT "tests_reporter"."id",
>>>     "tests_reporter"."full_name"
>>> FROM "tests_reporter"
>>> WHERE "tests_reporter"."id" = %s
>>>
>>>
>>> №[2] time=[0.109] explain=['2 0 0 SEARCH TABLE tests_article USING INTEGER PRIMARY KEY (rowid=?)']
>>> SELECT "tests_article"."id",
>>>     "tests_article"."pub_date",
>>>     "tests_article"."headline",
>>>     "tests_article"."content",
>>>     "tests_article"."reporter_id"
>>> FROM "tests_article"
>>> WHERE "tests_article"."id" = %s
>>>
>>>
>>> Tests count: 1  |  Total queries count: 2  |  Total execution time: 0.22s  |  Median time one test is: 0.109s  |  Vendor: sqlite