File: unique_index.py

package info (click to toggle)
python-django-postgres-extra 2.0.9-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 1,096 kB
  • sloc: python: 9,057; makefile: 17; sh: 7; sql: 1
file content (18 lines) | stat: -rw-r--r-- 539 bytes parent folder | download | duplicates (3) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
 
import django

from django.db.models.indexes import Index


class UniqueIndex(Index):
    def create_sql(self, *args, **kwargs):
        if django.VERSION >= (2, 0):
            statement = super().create_sql(*args, **kwargs)
            statement.template = self._rewrite_sql(statement.template)
            return statement

        sql = super().create_sql(*args, **kwargs)
        return self._rewrite_sql(sql)

    @staticmethod
    def _rewrite_sql(sql: str) -> str:
        return sql.replace("CREATE INDEX", "CREATE UNIQUE INDEX")