From 0a6af29625d8266858868112b21f9d00294046e7 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Sat, 23 Jan 2016 22:39:39 -0800 Subject: Add KafkaProducer to kafka and kafka.producer module imports --- kafka/__init__.py | 3 ++- kafka/producer/__init__.py | 4 +++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/kafka/__init__.py b/kafka/__init__.py index 68ba597..80eb025 100644 --- a/kafka/__init__.py +++ b/kafka/__init__.py @@ -5,6 +5,7 @@ __license__ = 'Apache License 2.0' __copyright__ = 'Copyright 2016 Dana Powers, David Arthur, and Contributors' from kafka.consumer import KafkaConsumer +from kafka.producer import KafkaProducer from kafka.conn import BrokerConnection from kafka.protocol import ( create_message, create_gzip_message, create_snappy_message) @@ -28,7 +29,7 @@ class KafkaClient(SimpleClient): __all__ = [ - 'KafkaConsumer', 'KafkaClient', 'BrokerConnection', + 'KafkaConsumer', 'KafkaProducer', 'KafkaClient', 'BrokerConnection', 'SimpleClient', 'SimpleProducer', 'KeyedProducer', 'RoundRobinPartitioner', 'HashedPartitioner', 'create_message', 'create_gzip_message', 'create_snappy_message', diff --git a/kafka/producer/__init__.py b/kafka/producer/__init__.py index bc0e7c6..3664eb2 100644 --- a/kafka/producer/__init__.py +++ b/kafka/producer/__init__.py @@ -1,6 +1,8 @@ +from .kafka import KafkaProducer from .simple import SimpleProducer from .keyed import KeyedProducer __all__ = [ - 'SimpleProducer', 'KeyedProducer' + 'KafkaProducer', + 'SimpleProducer', 'KeyedProducer' # deprecated ] -- cgit v1.2.1