Ticket #12763: django_db_debug_query_caller.diff

django/db/backends/util.py

@@ -8 +8 @@
 except ImportError:
     from django.utils import _decimal as decimal    # for Python 2.3
 
+import traceback
+import re
+
+def _query_caller(dbpath_re = re.compile('.*/django/db/.*')):
+    """Returns a formatted traceback containing only the ORM invocation that initiated this query."""
+    return traceback.format_list([t for t in traceback.extract_stack() if not dbpath_re.match(t[0])][-1:])
+
 class CursorDebugWrapper(object):
     def __init__(self, cursor, db):
         self.cursor = cursor
@@ -23 +30 @@
             self.db.queries.append({
                 'sql': sql,
                 'time': "%.3f" % (stop - start),
+                'caller': _query_caller(),
             })
 
     def executemany(self, sql, param_list):
@@ -34 +42 @@
             self.db.queries.append({
                 'sql': '%s times: %s' % (len(param_list), sql),
                 'time': "%.3f" % (stop - start),
+                'caller': _query_caller(),
             })
 
     def __getattr__(self, attr):