def test_rename_field(self):
project_state = self.set_up_test_model()
operation = migrations.RenameField('Pony', 'pink', 'rosa')
new_state = project_state.clone()
operation.state_forwards('tests', new_state)
self.assertColumnExists('tests_pony', 'pink')
self.assertColumnNotExists('tests_pony', 'rosa')
with connection.schema_editor() as editor:
operation.database_forwards('tests', editor, project_state, new_state)
self.assertColumnNotExists('tests_pony', 'pink')
self.assertColumnExists('tests_pony', 'rosa')
with connection.schema_editor() as editor:
operation.database_backwards('tests', editor, new_state, project_state)
self.assertColumnExists('tests_pony', 'pink')
self.assertColumnNotExists('tests_pony', 'rosa')
python类RenameField()的实例源码
def rename_field(field, filters: List[str]):
"""Renames a field from one name to the other.
Arguments:
field:
Field to be renamed.
filters:
List of strings to filter
SQL statements on.
"""
model = define_fake_model({'title': field})
project = migrations.state.ProjectState.from_apps(apps)
with connection.schema_editor() as schema_editor:
execute_migration(schema_editor, [
migrations.CreateModel(
model.__name__,
fields=[
('title', field.clone())
]
)
], project)
with filtered_schema_editor(*filters) as (schema_editor, calls):
execute_migration(schema_editor, [
migrations.RenameField(
model.__name__,
'title',
'newtitle'
)
], project)
yield calls
def reduce_rename_field_self(self, operation, other, in_between):
if (operation.model_name_lower == other.model_name_lower and
operation.new_name_lower == other.old_name_lower):
return [
migrations.RenameField(
operation.model_name,
operation.old_name,
other.new_name,
),
]
# THROUGH CHECKS