From b333de0f5dc6e66bd77cce9ffe0eef06a9e31f02 Mon Sep 17 00:00:00 2001 From: Andrew Godwin Date: Wed, 12 Feb 2014 17:16:40 +0000 Subject: [PATCH] Add --empty option to makemigrations --- .../management/commands/makemigrations.py | 27 ++++++++++++++++++- 1 file changed, 26 insertions(+), 1 deletion(-) diff --git a/django/core/management/commands/makemigrations.py b/django/core/management/commands/makemigrations.py index 8fe2597adf..e012e247f5 100644 --- a/django/core/management/commands/makemigrations.py +++ b/django/core/management/commands/makemigrations.py @@ -6,6 +6,7 @@ from optparse import make_option from django.apps import apps from django.core.management.base import BaseCommand, CommandError from django.db import connections, DEFAULT_DB_ALIAS, migrations +from django.db.migrations.migration import Migration from django.db.migrations.loader import MigrationLoader from django.db.migrations.autodetector import MigrationAutodetector from django.db.migrations.questioner import MigrationQuestioner, InteractiveMigrationQuestioner @@ -20,6 +21,8 @@ class Command(BaseCommand): help="Just show what migrations would be made; don't actually write them."), make_option('--merge', action='store_true', dest='merge', default=False, help="Enable fixing of migration conflicts."), + make_option('--empty', action='store_true', dest='empty', default=False, + help="Create an empty migration."), ) help = "Creates new migration(s) for apps." @@ -31,6 +34,7 @@ class Command(BaseCommand): self.interactive = options.get('interactive') self.dry_run = options.get('dry_run', False) self.merge = options.get('merge', False) + self.empty = options.get('empty', False) # Make sure the app they asked for exists app_labels = set(app_labels) @@ -71,12 +75,27 @@ class Command(BaseCommand): if self.merge and conflicts: return self.handle_merge(loader, conflicts) - # Detect changes + # Set up autodetector autodetector = MigrationAutodetector( loader.graph.project_state(), ProjectState.from_apps(apps), InteractiveMigrationQuestioner(specified_apps=app_labels), ) + + # If they want to make an empty migration, make one for each app + if self.empty: + if not app_labels: + raise CommandError("You must supply at least one app label when using --empty.") + # Make a fake changes() result we can pass to arrange_for_graph + changes = dict( + (app, [Migration("custom", app)]) + for app in app_labels + ) + changes = autodetector.arrange_for_graph(changes, loader.graph) + self.write_migration_files(changes) + return + + # Detect changes changes = autodetector.changes(graph=loader.graph, trim_to_apps=app_labels or None) # No changes? Tell them. @@ -89,6 +108,12 @@ class Command(BaseCommand): self.stdout.write("No changes detected") return + self.write_migration_files(changes) + + def write_migration_files(self, changes): + """ + Takes a changes dict and writes them out as migration files. + """ directory_created = {} for app_label, app_migrations in changes.items(): if self.verbosity >= 1: