|
12 | 12 | from django.db import connection |
13 | 13 | connections = {'default': connection} |
14 | 14 |
|
15 | | -from django.db.backends import util |
| 15 | +try: |
| 16 | + from django.db.backends import utils # renamed in django 1.7 |
| 17 | +except ImportError: |
| 18 | + from django.db.backends import util as utils # removed in django 1.9 |
16 | 19 | from django.conf import settings as django_settings |
17 | 20 | #from django.template import Node |
18 | 21 |
|
@@ -52,9 +55,9 @@ def truncate_sql(sql, aggregates=True): |
52 | 55 | import django |
53 | 56 | version = float('.'.join([str(x) for x in django.VERSION[:2]])) |
54 | 57 | if version >= 1.6: |
55 | | - DatabaseStatTracker = util.CursorWrapper |
| 58 | + DatabaseStatTracker = utils.CursorWrapper |
56 | 59 | else: |
57 | | - DatabaseStatTracker = util.CursorDebugWrapper |
| 60 | + DatabaseStatTracker = utils.CursorDebugWrapper |
58 | 61 |
|
59 | 62 |
|
60 | 63 | class DatabaseStatTracker(DatabaseStatTracker): |
@@ -126,14 +129,14 @@ class SQLRealTimeModule(DevServerModule): |
126 | 129 | logger_name = 'sql' |
127 | 130 |
|
128 | 131 | def process_init(self, request): |
129 | | - if not issubclass(util.CursorDebugWrapper, DatabaseStatTracker): |
130 | | - self.old_cursor = util.CursorDebugWrapper |
131 | | - util.CursorDebugWrapper = DatabaseStatTracker |
| 132 | + if not issubclass(utils.CursorDebugWrapper, DatabaseStatTracker): |
| 133 | + self.old_cursor = utils.CursorDebugWrapper |
| 134 | + utils.CursorDebugWrapper = DatabaseStatTracker |
132 | 135 | DatabaseStatTracker.logger = self.logger |
133 | 136 |
|
134 | 137 | def process_complete(self, request): |
135 | | - if issubclass(util.CursorDebugWrapper, DatabaseStatTracker): |
136 | | - util.CursorDebugWrapper = self.old_cursor |
| 138 | + if issubclass(utils.CursorDebugWrapper, DatabaseStatTracker): |
| 139 | + utils.CursorDebugWrapper = self.old_cursor |
137 | 140 |
|
138 | 141 |
|
139 | 142 | class SQLSummaryModule(DevServerModule): |
|
0 commit comments