Display RiemannErrors nicely when using the command line client (closes #5)

Author: Sam Clements

riemann_client/__init__.py

@@ -1,4 +1,4 @@
 """A Python Riemann client and command line tool"""
 
-__version__ = '4.0.1'
+__version__ = '4.1.0-dev'
 __author__ = 'Sam Clements <sam.clements@datasift.com>'

riemann_client/command.py

@@ -4,6 +4,7 @@
 
 import argparse
 import json
+import sys
 import os
 
 import riemann_client.client
@@ -102,4 +103,9 @@ def main():
     transport = transport_class(args.host, args.port)
 
     with riemann_client.client.Client(transport=transport) as client:
-        args.function(args, client)
+        try:
+            args.function(args, client)
+        except riemann_client.transport.RiemannError as exception:
+            print("The Riemann server responded with an error: {}".format(
+                exception.message), file=sys.stderr)
+            exit(1)

setup.py

@@ -4,7 +4,7 @@
 
 setuptools.setup(
     name='riemann-client',
-    version='4.0.1',
+    version='4.1.0-dev',
 
     author="Sam Clements",
     author_email="sam.clements@datasift.com",