diff options
Diffstat (limited to 'django')
| -rw-r--r-- | django/db/backends/base/features.py | 3 | ||||
| -rw-r--r-- | django/db/backends/base/schema.py | 45 | ||||
| -rw-r--r-- | django/db/backends/mysql/schema.py | 5 | ||||
| -rw-r--r-- | django/db/backends/oracle/schema.py | 1 | ||||
| -rw-r--r-- | django/db/backends/postgresql/schema.py | 1 | ||||
| -rw-r--r-- | django/db/backends/sqlite3/features.py | 1 |
6 files changed, 41 insertions, 15 deletions
diff --git a/django/db/backends/base/features.py b/django/db/backends/base/features.py index 8afc7eb516..e0b7372eb1 100644 --- a/django/db/backends/base/features.py +++ b/django/db/backends/base/features.py @@ -179,6 +179,9 @@ class BaseDatabaseFeatures: # Does it support foreign keys? supports_foreign_keys = True + # Can it create foreign key constraints inline when adding columns? + can_create_inline_fk = True + # Does it support CHECK constraints? supports_column_check_constraints = True supports_table_check_constraints = True diff --git a/django/db/backends/base/schema.py b/django/db/backends/base/schema.py index 113d1b7f67..0522abcaad 100644 --- a/django/db/backends/base/schema.py +++ b/django/db/backends/base/schema.py @@ -77,6 +77,7 @@ class BaseDatabaseSchemaEditor: "REFERENCES %(to_table)s (%(to_column)s)%(deferrable)s" ) sql_create_inline_fk = None + sql_create_column_inline_fk = None sql_delete_fk = sql_delete_constraint sql_create_index = "CREATE INDEX %(name)s ON %(table)s (%(columns)s)%(extra)s%(condition)s" @@ -433,6 +434,22 @@ class BaseDatabaseSchemaEditor: db_params = field.db_parameters(connection=self.connection) if db_params['check']: definition += " " + self.sql_check_constraint % db_params + if field.remote_field and self.connection.features.supports_foreign_keys and field.db_constraint: + constraint_suffix = '_fk_%(to_table)s_%(to_column)s' + # Add FK constraint inline, if supported. + if self.sql_create_column_inline_fk: + to_table = field.remote_field.model._meta.db_table + to_column = field.remote_field.model._meta.get_field(field.remote_field.field_name).column + definition += " " + self.sql_create_column_inline_fk % { + 'name': self._fk_constraint_name(model, field, constraint_suffix), + 'column': self.quote_name(field.column), + 'to_table': self.quote_name(to_table), + 'to_column': self.quote_name(to_column), + 'deferrable': self.connection.ops.deferrable_sql() + } + # Otherwise, add FK constraints later. + else: + self.deferred_sql.append(self._create_fk_sql(model, field, constraint_suffix)) # Build the SQL and run it sql = self.sql_create_column % { "table": self.quote_name(model._meta.db_table), @@ -451,9 +468,6 @@ class BaseDatabaseSchemaEditor: self.execute(sql, params) # Add an index, if required self.deferred_sql.extend(self._field_indexes_sql(model, field)) - # Add any FK constraints later - if field.remote_field and self.connection.features.supports_foreign_keys and field.db_constraint: - self.deferred_sql.append(self._create_fk_sql(model, field, "_fk_%(to_table)s_%(to_column)s")) # Reset connection if required if self.connection.features.connection_persists_old_columns: self.connection.close() @@ -984,18 +998,8 @@ class BaseDatabaseSchemaEditor: } def _create_fk_sql(self, model, field, suffix): - def create_fk_name(*args, **kwargs): - return self.quote_name(self._create_index_name(*args, **kwargs)) - table = Table(model._meta.db_table, self.quote_name) - name = ForeignKeyName( - model._meta.db_table, - [field.column], - split_identifier(field.target_field.model._meta.db_table)[1], - [field.target_field.column], - suffix, - create_fk_name, - ) + name = self._fk_constraint_name(model, field, suffix) column = Columns(model._meta.db_table, [field.column], self.quote_name) to_table = Table(field.target_field.model._meta.db_table, self.quote_name) to_column = Columns(field.target_field.model._meta.db_table, [field.target_field.column], self.quote_name) @@ -1010,6 +1014,19 @@ class BaseDatabaseSchemaEditor: deferrable=deferrable, ) + def _fk_constraint_name(self, model, field, suffix): + def create_fk_name(*args, **kwargs): + return self.quote_name(self._create_index_name(*args, **kwargs)) + + return ForeignKeyName( + model._meta.db_table, + [field.column], + split_identifier(field.target_field.model._meta.db_table)[1], + [field.target_field.column], + suffix, + create_fk_name, + ) + def _delete_fk_sql(self, model, name): return self._delete_constraint_sql(self.sql_delete_fk, model, name) diff --git a/django/db/backends/mysql/schema.py b/django/db/backends/mysql/schema.py index 65d1f24429..172e9b9ac8 100644 --- a/django/db/backends/mysql/schema.py +++ b/django/db/backends/mysql/schema.py @@ -16,7 +16,10 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor): sql_rename_column = "ALTER TABLE %(table)s CHANGE %(old_column)s %(new_column)s %(type)s" sql_delete_unique = "ALTER TABLE %(table)s DROP INDEX %(name)s" - + sql_create_column_inline_fk = ( + ', ADD CONSTRAINT %(name)s FOREIGN KEY (%(column)s) ' + 'REFERENCES %(to_table)s(%(to_column)s)' + ) sql_delete_fk = "ALTER TABLE %(table)s DROP FOREIGN KEY %(name)s" sql_delete_index = "DROP INDEX %(name)s ON %(table)s" diff --git a/django/db/backends/oracle/schema.py b/django/db/backends/oracle/schema.py index 4355402c52..9ac71b927e 100644 --- a/django/db/backends/oracle/schema.py +++ b/django/db/backends/oracle/schema.py @@ -15,6 +15,7 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor): sql_alter_column_default = "MODIFY %(column)s DEFAULT %(default)s" sql_alter_column_no_default = "MODIFY %(column)s DEFAULT NULL" sql_delete_column = "ALTER TABLE %(table)s DROP COLUMN %(column)s" + sql_create_column_inline_fk = 'CONSTRAINT %(name)s REFERENCES %(to_table)s(%(to_column)s)%(deferrable)s' sql_delete_table = "DROP TABLE %(table)s CASCADE CONSTRAINTS" sql_create_index = "CREATE INDEX %(name)s ON %(table)s (%(columns)s)%(extra)s" diff --git a/django/db/backends/postgresql/schema.py b/django/db/backends/postgresql/schema.py index 9eecaac19a..2dbd9e28a1 100644 --- a/django/db/backends/postgresql/schema.py +++ b/django/db/backends/postgresql/schema.py @@ -15,6 +15,7 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor): sql_create_index = "CREATE INDEX %(name)s ON %(table)s%(using)s (%(columns)s)%(extra)s%(condition)s" sql_delete_index = "DROP INDEX IF EXISTS %(name)s" + sql_create_column_inline_fk = 'REFERENCES %(to_table)s(%(to_column)s)%(deferrable)s' # Setting the constraint to IMMEDIATE runs any deferred checks to allow # dropping it in the same transaction. sql_delete_fk = "SET CONSTRAINTS %(name)s IMMEDIATE; ALTER TABLE %(table)s DROP CONSTRAINT %(name)s" diff --git a/django/db/backends/sqlite3/features.py b/django/db/backends/sqlite3/features.py index 19e949899f..40b99a07b9 100644 --- a/django/db/backends/sqlite3/features.py +++ b/django/db/backends/sqlite3/features.py @@ -26,6 +26,7 @@ class DatabaseFeatures(BaseDatabaseFeatures): atomic_transactions = False can_rollback_ddl = True supports_atomic_references_rename = Database.sqlite_version_info >= (3, 26, 0) + can_create_inline_fk = False supports_paramstyle_pyformat = False supports_sequence_reset = False can_clone_databases = True |
