Add a "version" command

[stgit] / stgit / main.py

diff --git a/stgit/main.py b/stgit/main.py
index fee0cc1..2350013 100644 (file)
--- a/stgit/main.py
+++ b/stgit/main.py
@@ -36,6 +36,8 @@ import stgit.commands.clean
 import stgit.commands.clone
 import stgit.commands.export
 import stgit.commands.files
+import stgit.commands.fold
+import stgit.commands.id
 import stgit.commands.imprt
 import stgit.commands.init
 import stgit.commands.mail
@@ -65,6 +67,8 @@ commands = {
     'clone':    stgit.commands.clone,
     'export':   stgit.commands.export,
     'files':    stgit.commands.files,
+    'fold':     stgit.commands.fold,
+    'id':       stgit.commands.id,
     'import':   stgit.commands.imprt,
     'init':     stgit.commands.init,
     'mail':     stgit.commands.mail,
@@ -87,6 +91,8 @@ def print_help():
     print
     print 'commands:'
     print '  help        print this message'
+    print '  version     display version information'
+    print
 
     cmds = commands.keys()
     cmds.sort()
@@ -112,8 +118,9 @@ def main():
     if cmd in ['-h', '--help', 'help']:
         print_help()
         sys.exit(0)
-    if cmd in ['-v', '--version']:
-        print '%s %s' % (prog, version)
+    if cmd in ['-v', '--version', 'version']:
+        print 'Stacked GIT %s' % version
+        print 'Python version %s' % sys.version
         sys.exit(0)
     if not cmd in commands:
         print >> sys.stderr, 'Unknown command: %s' % cmd