Code

Ticket #2495: 2495-against-15801.diff

File 2495-against-15801.diff, 7.8 KB (added by salgado, 3 years ago)
Line 
1Index: django/db/backends/postgresql/creation.py
2===================================================================
3--- django/db/backends/postgresql/creation.py   (revision 15796)
4+++ django/db/backends/postgresql/creation.py   (working copy)
5@@ -36,41 +36,28 @@
6             return "WITH ENCODING '%s'" % self.connection.settings_dict['TEST_CHARSET']
7         return ''
8 
9-    def sql_indexes_for_field(self, model, f, style):
10-        if f.db_index and not f.unique:
11+    def compile_sql_index_for_field(self, db_table, f, style, tablespace_sql):
12+        def _get_index_sql(index_name, opclass=''):
13             qn = self.connection.ops.quote_name
14-            db_table = model._meta.db_table
15-            tablespace = f.db_tablespace or model._meta.db_tablespace
16-            if tablespace:
17-                sql = self.connection.ops.tablespace_sql(tablespace)
18-                if sql:
19-                    tablespace_sql = ' ' + sql
20-                else:
21-                    tablespace_sql = ''
22-            else:
23-                tablespace_sql = ''
24+            return (style.SQL_KEYWORD('CREATE INDEX') + ' ' +
25+                    style.SQL_TABLE(qn(truncate_name(
26+                        index_name, self.connection.ops.max_name_length()))) + ' ' +
27+                    style.SQL_KEYWORD('ON') + ' ' +
28+                    style.SQL_TABLE(qn(db_table)) + ' ' +
29+                    "(%s%s)" % (style.SQL_FIELD(qn(f.column)), opclass) +
30+                    "%s;" % tablespace_sql)
31 
32-            def get_index_sql(index_name, opclass=''):
33-                return (style.SQL_KEYWORD('CREATE INDEX') + ' ' +
34-                        style.SQL_TABLE(qn(truncate_name(index_name,self.connection.ops.max_name_length()))) + ' ' +
35-                        style.SQL_KEYWORD('ON') + ' ' +
36-                        style.SQL_TABLE(qn(db_table)) + ' ' +
37-                        "(%s%s)" % (style.SQL_FIELD(qn(f.column)), opclass) +
38-                        "%s;" % tablespace_sql)
39+        output = [_get_index_sql('%s_%s' % (db_table, f.column))]
40 
41-            output = [get_index_sql('%s_%s' % (db_table, f.column))]
42-
43-            # Fields with database column types of `varchar` and `text` need
44-            # a second index that specifies their operator class, which is
45-            # needed when performing correct LIKE queries outside the
46-            # C locale. See #12234.
47-            db_type = f.db_type(connection=self.connection)
48-            if db_type.startswith('varchar'):
49-                output.append(get_index_sql('%s_%s_like' % (db_table, f.column),
50-                                            ' varchar_pattern_ops'))
51-            elif db_type.startswith('text'):
52-                output.append(get_index_sql('%s_%s_like' % (db_table, f.column),
53-                                            ' text_pattern_ops'))
54-        else:
55-            output = []
56+        # Fields with database column types of `varchar` and `text` need
57+        # a second index that specifies their operator class, which is
58+        # needed when performing correct LIKE queries outside the
59+        # C locale. See #12234.
60+        db_type = f.db_type(connection=self.connection)
61+        if db_type.startswith('varchar'):
62+            output.append(_get_index_sql('%s_%s_like' % (db_table, f.column),
63+                                         ' varchar_pattern_ops'))
64+        elif db_type.startswith('text'):
65+            output.append(_get_index_sql('%s_%s_like' % (db_table, f.column),
66+                                         ' text_pattern_ops'))
67         return output
68Index: django/db/backends/mysql/creation.py
69===================================================================
70--- django/db/backends/mysql/creation.py        (revision 15796)
71+++ django/db/backends/mysql/creation.py        (working copy)
72@@ -63,3 +63,29 @@
73                 field.rel.to._meta.db_table, field.rel.to._meta.pk.column)
74             ]
75         return table_output, deferred
76+
77+    def compile_sql_index_for_field(self, db_table, f, style, tablespace_sql):
78+        """
79+        Adds a  prefix length to the TextField which is currently represented
80+        by MySQL's 'longtext' type -- which requires a prefix-length parameter
81+        in order to install an index.
82+
83+        Use largest possible value allowed - 255 characters.
84+        """
85+        from django.db.backends.util import truncate_name
86+
87+        qn = self.connection.ops.quote_name
88+        index_name = '%s_%s' % (db_table, self._digest(f.column))
89+
90+        if f.db_type() == self.data_types['TextField']:
91+            field_name = style.SQL_FIELD(qn(f.column)) + '(255)'
92+        else:
93+            field_name = style.SQL_FIELD(qn(f.column))
94+
95+        return [style.SQL_KEYWORD('CREATE INDEX') + ' ' +
96+                style.SQL_TABLE(qn(truncate_name(index_name, self.connection.ops.max_name_length()))) + ' ' +
97+                style.SQL_KEYWORD('ON') + ' ' +
98+                style.SQL_TABLE(qn(db_table)) + ' ' +
99+                "(%s)" % field_name +
100+                "%s;" % tablespace_sql]
101+
102Index: django/db/backends/creation.py
103===================================================================
104--- django/db/backends/creation.py      (revision 15796)
105+++ django/db/backends/creation.py      (working copy)
106@@ -250,12 +250,22 @@
107             output.extend(self.sql_indexes_for_field(model, f, style))
108         return output
109 
110+    def compile_sql_index_for_field(self, db_table, f, style, tablespace_sql):
111+        "Compile CREATE INDEX SQL statements for sql_indexes_for_field"
112+        from django.db.backends.util import truncate_name
113+
114+        qn = self.connection.ops.quote_name
115+        index_name = '%s_%s' % (db_table, self._digest(f.column))
116+        return [style.SQL_KEYWORD('CREATE INDEX') + ' ' +
117+                style.SQL_TABLE(qn(truncate_name(index_name, self.connection.ops.max_name_length()))) + ' ' +
118+                style.SQL_KEYWORD('ON') + ' ' +
119+                style.SQL_TABLE(qn(db_table)) + ' ' +
120+                "(%s)" % style.SQL_FIELD(qn(f.column)) +
121+                "%s;" % tablespace_sql]
122+
123     def sql_indexes_for_field(self, model, f, style):
124         "Return the CREATE INDEX SQL statements for a single model field"
125-        from django.db.backends.util import truncate_name
126-
127         if f.db_index and not f.unique:
128-            qn = self.connection.ops.quote_name
129             tablespace = f.db_tablespace or model._meta.db_tablespace
130             if tablespace:
131                 sql = self.connection.ops.tablespace_sql(tablespace)
132@@ -265,13 +275,7 @@
133                     tablespace_sql = ''
134             else:
135                 tablespace_sql = ''
136-            i_name = '%s_%s' % (model._meta.db_table, self._digest(f.column))
137-            output = [style.SQL_KEYWORD('CREATE INDEX') + ' ' +
138-                style.SQL_TABLE(qn(truncate_name(i_name, self.connection.ops.max_name_length()))) + ' ' +
139-                style.SQL_KEYWORD('ON') + ' ' +
140-                style.SQL_TABLE(qn(model._meta.db_table)) + ' ' +
141-                "(%s)" % style.SQL_FIELD(qn(f.column)) +
142-                "%s;" % tablespace_sql]
143+            output = self.compile_sql_index_for_field(model._meta.db_table, f, style, tablespace_sql)
144         else:
145             output = []
146         return output
147Index: tests/regressiontests/bug2495/models.py
148===================================================================
149--- tests/regressiontests/bug2495/models.py     (revision 0)
150+++ tests/regressiontests/bug2495/models.py     (revision 0)
151@@ -0,0 +1,15 @@
152+from unittest import TestCase
153+
154+from django.db import models, connection
155+
156+class IndexedTextModel(models.Model):
157+    indexed_text = models.TextField(db_index=True)
158+
159+class TestTextFieldIndices(TestCase):
160+    def test_indexed_text_has_an_index(self):
161+        try:
162+            indexes = connection.introspection.get_indexes(connection.cursor(), IndexedTextModel._meta.db_table)
163+        except NotImplementedError:
164+            indexes = {}
165+        assert 'indexed_text' in indexes
166+