Add a simple command line support

2 files changed, 29 insertions, 2 deletions

diff --git a/prettytable/cli.py b/prettytable/cli.py
index e69de29..95b5a90 100644
--- a/prettytable/cli.py
+++ b/prettytable/cli.py
@@ -0,0 +1,16 @@
+from __future__ import print_function
+import argparse
+from .factory import from_csv
+
+
+def main():
+    parser = argparse.ArgumentParser(description='A simple Python library designed to make it quick and easy to '
+                                     'represent tabular data in visually appealing ASCII tables.')
+    parser.add_argument('--csv', help='CSV file name')
+    args = parser.parse_args()
+    with open(args.csv) as fp:
+        print(from_csv(fp))
+
+
+if __name__ == '__main__':
+    main()
diff --git a/setup.py b/setup.py
index 37061df..61e6b4d 100644
--- a/setup.py
+++ b/setup.py
@@ -2,9 +2,17 @@
 from setuptools import setup
 from prettytable import __version__ as version
 
+
 setup(
     name='prettytable',
     version=version,
+    include_package_data=True,
+    zip_safe=False,
+    entry_points={
+        'console_scripts': [
+            'ptable = prettytable.cli:main',
+        ]
+    },
     classifiers=[
         'Programming Language :: Python',
         'Programming Language :: Python :: 2.4',
@@ -19,7 +27,10 @@ setup(
     description='A simple Python library for easily displaying tabular data in a visually appealing ASCII table format',
     author='Luke Maurits',
     author_email='luke@maurits.id.au',
+    maintainer='Kane Blueriver',
+    maintainer_email='kxxoling@gmail.com',
     url='http://code.google.com/p/prettytable',
-    py_modules=['prettytable'],
-    test_suite="prettytable_test"
+    py_modules=['prettytable', 'prettytable.cli', 'prettytable.prettytable',
+                'prettytable.factory', 'prettytable._compact'],
+    test_suite="test_prettytable",
 )