Merge branch 'edit-latest' of https://bitbucket.org/exhuma/alembic into pr46

1 files changed, 15 insertions, 0 deletions

diff --git a/alembic/command.py b/alembic/command.py
index aab5cc2..13d8a57 100644
--- a/alembic/command.py
+++ b/alembic/command.py
@@ -5,6 +5,8 @@ from .runtime.environment import EnvironmentContext
 from . import util
 from . import autogenerate as autogen
 
+import editor
+
 
 def list_templates(config):
     """List available templates"""
@@ -353,3 +355,16 @@ def stamp(config, revision, sql=False, tag=None):
         tag=tag
     ):
         script.run_env()
+
+
+def edit(config):
+    """Edit the latest ervision"""
+
+    script = ScriptDirectory.from_config(config)
+    revisions = script.walk_revisions()
+    head = next(revisions)
+
+    try:
+        editor.edit(head.path)
+    except Exception as exc:
+        raise util.CommandError('Error executing editor (%s)' % (exc,))