Ticket #825: south.diff

south/management/commands/schemamigration.py

@@ -39 +39 @@
             help='Attempt to automatically detect differences from the last migration.'),
         make_option('--empty', action='store_true', dest='empty', default=False,
             help='Make a blank migration.'),
+        make_option('--edit', action='store_true', dest='edit_migration', default=False,
+            help='Edit the file after creating it.'),
     )
     help = "Creates a new template schema migration for the given app"
     usage_str = "Usage: ./manage.py schemamigration appname migrationname [--empty] [--initial] [--auto] [--add-model ModelName] [--add-field ModelName.field_name] [--stdout]"
     
-    def handle(self, app=None, name="", added_model_list=None, added_field_list=None, freeze_list=None, initial=False, auto=False, stdout=False, added_index_list=None, verbosity=1, empty=False, **options):
+    def handle(self, app=None, name="", added_model_list=None, added_field_list=None, freeze_list=None, initial=False, auto=False, stdout=False, added_index_list=None, verbosity=1, empty=False, edit_migration=False, **options):
         
         # Any supposed lists that are None become empty lists
         added_model_list = added_model_list or []
@@ -170 +172 @@
                 print >>sys.stderr, "Created %s. You must now edit this migration and add the code for each direction." % new_filename
             else:
                 print >>sys.stderr, "Created %s. You can now apply this migration with: ./manage.py migrate %s" % (new_filename, app)
+            if edit_migration:
+                os.system('%s %s' % (os.environ['EDITOR'], os.path.join(migrations.migrations_dir(), new_filename)))
 
 
 MIGRATION_TEMPLATE = """# encoding: utf-8