updated to use kafka-python==1.3.1

python_kafka_logging/KafkaHandler.py

@@ -1,15 +1,14 @@
-from kafka.client import KafkaClient
+from kafka.client import SimpleClient
 from kafka.producer import SimpleProducer, KeyedProducer
-from kafka.conn import DEFAULT_SOCKET_TIMEOUT_SECONDS
 import logging
 
 
 class KafkaLoggingHandler(logging.Handler):
 
-    def __init__(self, hosts_list, topic, timeout_secs=DEFAULT_SOCKET_TIMEOUT_SECONDS, **kwargs):
+    def __init__(self, hosts_list, topic, **kwargs):
         logging.Handler.__init__(self)
 
-        self.kafka_client = KafkaClient(hosts_list, timeout=timeout_secs)
+        self.kafka_client = SimpleClient(hosts_list)
         self.key = kwargs.get("key", None)
         self.kafka_topic_name = topic
 

requirements.txt

@@ -1,2 +1,2 @@
-kafka-python==0.9.5
+kafka-python==1.3.1
 logstash-formatter==0.5.8