49 lines
1.4 KiB
Python
49 lines
1.4 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(),
|
|
]
|