44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
from unittest import mock
|
|
|
|
from django.db import connection, migrations
|
|
|
|
try:
|
|
from django.contrib.postgres.operations import (
|
|
BloomExtension, BtreeGinExtension, BtreeGistExtension, CITextExtension,
|
|
CreateExtension, CryptoExtension, HStoreExtension, TrigramExtension,
|
|
UnaccentExtension,
|
|
)
|
|
except ImportError:
|
|
BloomExtension = mock.Mock()
|
|
BtreeGinExtension = mock.Mock()
|
|
BtreeGistExtension = mock.Mock()
|
|
CITextExtension = mock.Mock()
|
|
CreateExtension = mock.Mock()
|
|
HStoreExtension = mock.Mock()
|
|
TrigramExtension = mock.Mock()
|
|
UnaccentExtension = mock.Mock()
|
|
needs_crypto_extension = False
|
|
else:
|
|
needs_crypto_extension = (
|
|
connection.vendor == 'postgresql' and
|
|
not connection.features.is_postgresql_13
|
|
)
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
operations = [
|
|
BloomExtension(),
|
|
BtreeGinExtension(),
|
|
BtreeGistExtension(),
|
|
CITextExtension(),
|
|
# Ensure CreateExtension quotes extension names by creating one with a
|
|
# dash in its name.
|
|
CreateExtension('uuid-ossp'),
|
|
# CryptoExtension is required for RandomUUID() on PostgreSQL < 13.
|
|
CryptoExtension() if needs_crypto_extension else mock.Mock(),
|
|
HStoreExtension(),
|
|
TrigramExtension(),
|
|
UnaccentExtension(),
|
|
]
|