cli fixups: producer py2 compat; consumer no stacktrace on KafkaError

Author: dpkp

kafka/cli/consumer/__init__.py

@@ -4,6 +4,7 @@
 import logging
 
 from kafka import KafkaConsumer
+from kafka.errors import KafkaError
 
 
 def main_parser():
@@ -78,7 +79,10 @@ def run_cli(args=None):
     except KeyboardInterrupt:
         logger.info('Bye!')
         return 0
-    except Exception:
+    except KafkaError as e:
+        logger.error(e)
+        return 1
+    except Exception as e:
         logger.exception('Error!')
         return 1
     finally:

kafka/cli/producer/__init__.py

@@ -66,9 +66,14 @@ def log_result(res_or_err):
         else:
             logger.info("Message produced: %s", res_or_err)
 
+    try:
+        input_py23 = raw_input
+    except NameError:
+        input_py23 = input
+
     try:
         while True:
-            value = input()
+            value = input_py23()
             producer.send(config.topic, value=value.encode(config.encoding)).add_both(log_result)
     except KeyboardInterrupt:
         logger.info('Bye!')