Fixed the --version command

agronholm · agronholm · commit f8a25b8a6a6a · 2016-02-03T22:51:17.000+02:00
diff --git a/sqlacodegen/main.py b/sqlacodegen/main.py
@@ -1,14 +1,13 @@
-""" """
 from __future__ import unicode_literals, division, print_function, absolute_import
 import argparse
 import codecs
 import sys
 
 from sqlalchemy.engine import create_engine
 from sqlalchemy.schema import MetaData
+import pkg_resources
 
 from sqlacodegen.codegen import CodeGenerator
-import sqlacodegen
 
 
 def main():
@@ -27,7 +26,8 @@ def main():
     args = parser.parse_args()
 
     if args.version:
-        print(sqlacodegen.version)
+        version = pkg_resources.get_distribution('sqlacodegen').parsed_version
+        print(version.public)
         return
     if not args.url:
         print('You must supply a url\n', file=sys.stderr)
