Code

Ticket #12234: charfield_index_opclass_fix_v2.diff

File charfield_index_opclass_fix_v2.diff, 2.1 KB (added by jbronn, 5 years ago)

simplifed get_index_sql

Line 
1Index: django/db/backends/postgresql/creation.py
2===================================================================
3--- django/db/backends/postgresql/creation.py   (revision 11876)
4+++ django/db/backends/postgresql/creation.py   (working copy)
5@@ -34,3 +34,40 @@
6         if settings.TEST_DATABASE_CHARSET:
7             return "WITH ENCODING '%s'" % settings.TEST_DATABASE_CHARSET
8         return ''
9+
10+    def sql_indexes_for_field(self, model, f, style):
11+        if f.db_index and not f.unique:
12+            from django.db.models.fields import CharField, TextField
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+
25+            def get_index_sql(index_name, opclass=''):
26+                return (style.SQL_KEYWORD('CREATE INDEX') + ' ' +
27+                        style.SQL_TABLE(qn(index_name)) + ' ' +
28+                        style.SQL_KEYWORD('ON') + ' ' +
29+                        style.SQL_TABLE(qn(db_table)) + ' ' +
30+                        "(%s%s)" % (style.SQL_FIELD(qn(f.column)), opclass) +
31+                        "%s;" % tablespace_sql)
32+
33+            output = [get_index_sql('%s_%s' % (db_table, f.column))]
34+
35+            # CharFields and TextFields require a second index that specifies
36+            # their operator class for performing LIKE queries. See #12234.
37+            if isinstance(f, CharField):
38+                output.append(get_index_sql('%s_%s_like' % (db_table, f.column),
39+                                            ' varchar_pattern_ops'))
40+            elif isinstance(f, TextField):
41+                output.append(get_index_sql('%s_%s_like' % (db_table, f.column),
42+                                            ' text_pattern_ops'))
43+        else:
44+            output = []
45+        return output