From: Charles-Axel Dein <ca@d3in.org>
Date: Wed, 6 Mar 2013 18:31:09 +0000 (-0800)
Subject: Add --head-only option to current command
X-Git-Tag: rel_0_5_0~16^2~2
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=0815f5194bd8ac96988258f03f11554a629caf9f;p=thirdparty%2Fsqlalchemy%2Falembic.git

Add --head-only option to current command
---

diff --git a/alembic/command.py b/alembic/command.py
index 0ed98435..d326943d 100644
--- a/alembic/command.py
+++ b/alembic/command.py
@@ -169,15 +169,21 @@ def branches(config):
                     script.get_revision(rev)
                 )
 
-def current(config):
+def current(config, head_only=False):
     """Display the current revision for each database."""
 
     script = ScriptDirectory.from_config(config)
     def display_version(rev, context):
-        config.print_stdout("Current revision for %s: %s",
-                            util.obfuscate_url_pw(
-                                context.connection.engine.url),
-                            script.get_revision(rev))
+        rev = script.get_revision(rev)
+
+        if head_only:
+            config.print_stdout(str(rev.revision))
+
+        else:
+            config.print_stdout("Current revision for %s: %s",
+                                util.obfuscate_url_pw(
+                                    context.connection.engine.url),
+                                rev)
         return []
 
     with EnvironmentContext(
@@ -219,5 +225,3 @@ def splice(config, parent, child):
 
     """
     raise NotImplementedError()
-
-
diff --git a/alembic/config.py b/alembic/config.py
index 71f2731b..ca233e16 100644
--- a/alembic/config.py
+++ b/alembic/config.py
@@ -192,6 +192,11 @@ class CommandLine(object):
                                 help="Populate revision script with candidate "
                                     "migration operations, based on comparison "
                                     "of database to model.")
+            # "current" command
+            if 'head_only' in kwargs:
+                parser.add_argument("--head-only",
+                                    action="store_true",
+                                    help="Only show head version")
 
             positional_help = {
                 'directory': "location of scripts directory",
@@ -257,4 +262,3 @@ def main(argv=None, prog=None, **kwargs):
     """The console runner function for Alembic."""
 
     CommandLine(prog=prog).main(argv=argv)
-