Add -V / --version

asottile · asottile · commit a7506061bd5c · 2014-06-15T12:49:51.000-07:00
diff --git a/pre_commit/main.py b/pre_commit/main.py
@@ -1,6 +1,7 @@
 from __future__ import unicode_literals
 
 import argparse
+import pkg_resources
 
 from pre_commit import color
 from pre_commit.commands.autoupdate import autoupdate
@@ -16,6 +17,15 @@
 def main(argv):
     parser = argparse.ArgumentParser()
 
+    # http://stackoverflow.com/a/8521644/812183
+    parser.add_argument(
+        '-V', '--version',
+        action='version',
+        version='%(prog)s {0}'.format(
+            pkg_resources.get_distribution('pre-commit').version
+        )
+    )
+
     subparsers = parser.add_subparsers(dest='command')
 
     subparsers.add_parser('install', help='Intall the pre-commit script.')
