Merge pull request #76 from JelteF/master

Add edit command

docs/index.rst

@@ -101,7 +101,10 @@ The application will now have a ``db`` command line option with several sub-comm
 
 - ``manage.py db migrate [--message MESSAGE] [--sql] [--head HEAD] [--splice] [--branch-label BRANCH_LABEL] [--version-path VERSION_PATH] [--rev-id REV_ID]``
     Equivalent to ``revision --autogenerate``. The migration script is populated with changes detected automatically. The generated script should to be reviewed and edited as not all types of changes can be detected. This command does not make any changes to the database.
-
+
+- ``manage.py db edit <revision>``
+    Edit revision script(s) using $EDITOR.
+
 - ``manage.py db upgrade [--sql] [--tag TAG] [--x-arg ARG] <revision>``
     Upgrades the database. If ``revision`` isn't given then ``"head"`` is assumed.
 
@@ -150,6 +153,9 @@ The commands exposed by Flask-Migrate's interface to Flask-Script can also be ac
 - ``migrate(directory='migrations', message=None, sql=False, head='head', splice=False, branch_label=None, version_path=None, rev_id=None)``
     Creates an automatic revision script.
 
+- ``edit(directory='migrations', revision='head')``
+    Edit revision script(s) using $EDITOR.
+
 - ``merge(directory='migrations', revisions='', message=None, branch_label=None, rev_id=None)``
     Merge two revisions together.  Creates a new migration file.
 

flask_migrate/__init__.py

@@ -159,6 +159,20 @@ def migrate(directory=None, message=None, sql=False, head='head', splice=False,
         command.revision(config, message, autogenerate=True, sql=sql)
 
 
+@MigrateCommand.option('revision', nargs='?', default='head',
+                       help="revision identifier")
+@MigrateCommand.option('-d', '--directory', dest='directory', default=None,
+                       help=("migration script directory (default is "
+                             "'migrations')"))
+def edit(revision='current', directory=None):
+    """Upgrade to a later version"""
+    if alembic_version >= (0, 8, 0):
+        config = _get_config(directory)
+        command.edit(config, revision)
+    else:
+        raise RuntimeError('Alembic 0.8.0 or greater is required')
+
+
 @MigrateCommand.option('--rev-id', dest='rev_id', default=None,
                        help=('Specify a hardcoded revision id instead of '
                              'generating one'))