Fixed #26542 -- Fixed quoting in CreateExtension operation.
This commit is contained in:
parent
8ccb8ff453
commit
c112198332
|
@ -14,10 +14,10 @@ class CreateExtension(Operation):
|
||||||
def database_forwards(self, app_label, schema_editor, from_state, to_state):
|
def database_forwards(self, app_label, schema_editor, from_state, to_state):
|
||||||
if schema_editor.connection.vendor != 'postgresql':
|
if schema_editor.connection.vendor != 'postgresql':
|
||||||
return
|
return
|
||||||
schema_editor.execute("CREATE EXTENSION IF NOT EXISTS %s" % self.name)
|
schema_editor.execute("CREATE EXTENSION IF NOT EXISTS %s" % schema_editor.quote_name(self.name))
|
||||||
|
|
||||||
def database_backwards(self, app_label, schema_editor, from_state, to_state):
|
def database_backwards(self, app_label, schema_editor, from_state, to_state):
|
||||||
schema_editor.execute("DROP EXTENSION %s" % self.name)
|
schema_editor.execute("DROP EXTENSION %s" % schema_editor.quote_name(self.name))
|
||||||
|
|
||||||
def describe(self):
|
def describe(self):
|
||||||
return "Creates extension %s" % self.name
|
return "Creates extension %s" % self.name
|
||||||
|
|
|
@ -5,20 +5,21 @@ from django.db import migrations
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from django.contrib.postgres.operations import (
|
from django.contrib.postgres.operations import (
|
||||||
HStoreExtension, UnaccentExtension,
|
CreateExtension, HStoreExtension, UnaccentExtension,
|
||||||
)
|
)
|
||||||
except ImportError:
|
except ImportError:
|
||||||
from django.test import mock
|
from django.test import mock
|
||||||
|
CreateExtension = mock.Mock()
|
||||||
HStoreExtension = mock.Mock()
|
HStoreExtension = mock.Mock()
|
||||||
UnaccentExtension = mock.Mock()
|
UnaccentExtension = mock.Mock()
|
||||||
|
|
||||||
|
|
||||||
class Migration(migrations.Migration):
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
dependencies = [
|
|
||||||
]
|
|
||||||
|
|
||||||
operations = [
|
operations = [
|
||||||
|
# Ensure CreateExtension quotes extension names by creating one with a
|
||||||
|
# dash in its name.
|
||||||
|
CreateExtension('uuid-ossp'),
|
||||||
HStoreExtension(),
|
HStoreExtension(),
|
||||||
UnaccentExtension(),
|
UnaccentExtension(),
|
||||||
]
|
]
|
||||||
|
|
Loading…
Reference in New Issue