diff options
Diffstat (limited to 'kafka/__init__.py')
| -rw-r--r-- | kafka/__init__.py | 22 |
1 files changed, 18 insertions, 4 deletions
diff --git a/kafka/__init__.py b/kafka/__init__.py index 2a99847..68ba597 100644 --- a/kafka/__init__.py +++ b/kafka/__init__.py @@ -4,14 +4,28 @@ __author__ = 'Dana Powers' __license__ = 'Apache License 2.0' __copyright__ = 'Copyright 2016 Dana Powers, David Arthur, and Contributors' -from kafka.client import KafkaClient as SimpleClient -from kafka.client_async import KafkaClient +from kafka.consumer import KafkaConsumer from kafka.conn import BrokerConnection from kafka.protocol import ( create_message, create_gzip_message, create_snappy_message) -from kafka.producer import SimpleProducer, KeyedProducer from kafka.partitioner import RoundRobinPartitioner, HashedPartitioner, Murmur2Partitioner -from kafka.consumer import KafkaConsumer, SimpleConsumer, MultiProcessConsumer + +# To be deprecated when KafkaProducer interface is released +from kafka.client import SimpleClient +from kafka.producer import SimpleProducer, KeyedProducer + +# deprecated in favor of KafkaConsumer +from kafka.consumer import SimpleConsumer, MultiProcessConsumer + + +import warnings +class KafkaClient(SimpleClient): + def __init__(self, *args, **kwargs): + warnings.warn('The legacy KafkaClient interface has been moved to' + ' kafka.SimpleClient - this import will break in a' + ' future release', DeprecationWarning) + super(KafkaClient, self).__init__(*args, **kwargs) + __all__ = [ 'KafkaConsumer', 'KafkaClient', 'BrokerConnection', |