Correct argparse --version flag for py3.

1 files changed, 7 insertions, 2 deletions

diff --git a/sqlparse/__main__.py b/sqlparse/__main__.py
index 28abb6c..e6c6ed7 100644
--- a/sqlparse/__main__.py
+++ b/sqlparse/__main__.py
@@ -18,11 +18,11 @@ _CASE_CHOICES = ['upper', 'lower', 'capitalize']
 # TODO: Add CLI Tests
 # TODO: Simplify formatter by using argparse `type` arguments
 parser = argparse.ArgumentParser(
-    prog='sqlparse',
+    prog='sqlformat',
     description='Format FILE according to OPTIONS. Use "-" as FILE '
                 'to read from stdin.',
     usage='%(prog)s  [OPTIONS] FILE, ...',
-    version=sqlparse.__version__,)
+)
 
 parser.add_argument('filename')
 
@@ -32,6 +32,11 @@ parser.add_argument(
     metavar='FILE',
     help='write output to FILE (defaults to stdout)')
 
+parser.add_argument(
+    '--version',
+    action='version',
+    version=sqlparse.__version__)
+
 group = parser.add_argument_group('Formatting Options')
 
 group.add_argument(