summaryrefslogtreecommitdiff
path: root/django/core/management/commands/sqlmigrate.py
diff options
context:
space:
mode:
authordjango-bot <ops@djangoproject.com>2022-02-03 20:24:19 +0100
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2022-02-07 20:37:05 +0100
commit9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch)
treef0506b668a013d0063e5fba3dbf4863b466713ba /django/core/management/commands/sqlmigrate.py
parentf68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff)
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'django/core/management/commands/sqlmigrate.py')
-rw-r--r--django/core/management/commands/sqlmigrate.py46
1 files changed, 29 insertions, 17 deletions
diff --git a/django/core/management/commands/sqlmigrate.py b/django/core/management/commands/sqlmigrate.py
index f687360fb4..880eb11d9b 100644
--- a/django/core/management/commands/sqlmigrate.py
+++ b/django/core/management/commands/sqlmigrate.py
@@ -10,34 +10,40 @@ class Command(BaseCommand):
output_transaction = True
def add_arguments(self, parser):
- parser.add_argument('app_label', help='App label of the application containing the migration.')
- parser.add_argument('migration_name', help='Migration name to print the SQL for.')
parser.add_argument(
- '--database', default=DEFAULT_DB_ALIAS,
+ "app_label", help="App label of the application containing the migration."
+ )
+ parser.add_argument(
+ "migration_name", help="Migration name to print the SQL for."
+ )
+ parser.add_argument(
+ "--database",
+ default=DEFAULT_DB_ALIAS,
help='Nominates a database to create SQL for. Defaults to the "default" database.',
)
parser.add_argument(
- '--backwards', action='store_true',
- help='Creates SQL to unapply the migration, rather than to apply it',
+ "--backwards",
+ action="store_true",
+ help="Creates SQL to unapply the migration, rather than to apply it",
)
def execute(self, *args, **options):
# sqlmigrate doesn't support coloring its output but we need to force
# no_color=True so that the BEGIN/COMMIT statements added by
# output_transaction don't get colored either.
- options['no_color'] = True
+ options["no_color"] = True
return super().execute(*args, **options)
def handle(self, *args, **options):
# Get the database we're operating from
- connection = connections[options['database']]
+ connection = connections[options["database"]]
# Load up a loader to get all the migration data, but don't replace
# migrations.
loader = MigrationLoader(connection, replace_migrations=False)
# Resolve command-line arguments into a migration
- app_label, migration_name = options['app_label'], options['migration_name']
+ app_label, migration_name = options["app_label"], options["migration_name"]
# Validate app_label
try:
apps.get_app_config(app_label)
@@ -48,21 +54,27 @@ class Command(BaseCommand):
try:
migration = loader.get_migration_by_prefix(app_label, migration_name)
except AmbiguityError:
- raise CommandError("More than one migration matches '%s' in app '%s'. Please be more specific." % (
- migration_name, app_label))
+ raise CommandError(
+ "More than one migration matches '%s' in app '%s'. Please be more specific."
+ % (migration_name, app_label)
+ )
except KeyError:
- raise CommandError("Cannot find a migration matching '%s' from app '%s'. Is it in INSTALLED_APPS?" % (
- migration_name, app_label))
+ raise CommandError(
+ "Cannot find a migration matching '%s' from app '%s'. Is it in INSTALLED_APPS?"
+ % (migration_name, app_label)
+ )
target = (app_label, migration.name)
# Show begin/end around output for atomic migrations, if the database
# supports transactional DDL.
- self.output_transaction = migration.atomic and connection.features.can_rollback_ddl
+ self.output_transaction = (
+ migration.atomic and connection.features.can_rollback_ddl
+ )
# Make a plan that represents just the requested migrations and show SQL
# for it
- plan = [(loader.graph.nodes[target], options['backwards'])]
+ plan = [(loader.graph.nodes[target], options["backwards"])]
sql_statements = loader.collect_sql(plan)
- if not sql_statements and options['verbosity'] >= 1:
- self.stderr.write('No operations found.')
- return '\n'.join(sql_statements)
+ if not sql_statements and options["verbosity"] >= 1:
+ self.stderr.write("No operations found.")
+ return "\n".join(sql_statements)