Merge pull request #105 from deep-compute/feature-pretty

references #68, basescript is now a command-line tool which supports …

Author: prashanthellina

.travis.yml

@@ -19,8 +19,8 @@ deploy:
   - basescript/utils.py
   - examples/adder.py
   - examples/helloworld.py
-  name: basescript-0.3.6
-  tag_name: 0.3.6
+  name: basescript-0.3.7
+  tag_name: 0.3.7
   true:
     repo: deep-compute/basescript
 - provider: pypi

basescript/__init__.py

@@ -1,4 +1,4 @@
 from __future__ import absolute_import
 
-from .basescript import BaseScript
+from .basescript import BaseScript, main
 from .log import init_logger, get_logger

basescript/basescript.py

@@ -4,7 +4,7 @@
 import argparse
 import socket
 
-from .log import init_logger
+from .log import init_logger, pretty_print
 from deeputil import Dummy
 
 
@@ -107,7 +107,18 @@ def define_subcommands(self, subcommands):
         blah_command = subcommands.add_parser('blah')
         blah_command.set_defaults(func=fn_blah)
         """
-        pass
+        pretty_cmd = subcommands.add_parser("pretty")
+        pretty_cmd.add_argument(
+            "-c",
+            "--no-colors",
+            action="store_true",
+            default=False,
+            help="Do not emit colored output",
+        )
+
+        pretty_cmd.set_defaults(
+            func=lambda: pretty_print(colors=not self.args.no_colors)
+        )
 
     def define_baseargs(self, parser):
         """
@@ -182,3 +193,7 @@ def run(self):
         Override this method to define logic for `run` sub-command
         """
         pass
+
+
+def main():
+    BaseScript().start()

basescript/log.py

@@ -1,4 +1,5 @@
 import sys
+import json
 import time
 import uuid
 import atexit
@@ -466,5 +467,11 @@ def init_logger(
     return log
 
 
+def pretty_print(colors=True):
+    r = structlog.dev.ConsoleRenderer(colors=colors)
+    for line in sys.stdin:
+        print(r(None, None, json.loads(line)))
+
+
 def get_logger():
     return LOG

setup.py

@@ -25,7 +25,7 @@ def get_long_description():
 
 long_description = get_long_description()
 
-version = "0.3.6"
+version = "0.3.7"
 setup(
     name="basescript",
     version=version,
@@ -46,6 +46,7 @@ def get_long_description():
     ],
     package_dir={"basescript": "basescript"},
     packages=find_packages(".", exclude=["tests*"]),
+    entry_points={"console_scripts": ["basescript=basescript:main"]},
     include_package_data=True,
     classifiers=[
         "Programming Language :: Python :: 2.7",