WIP: KafkaProducer

kafka/__init__.py

@@ -11,13 +11,13 @@
 from kafka.protocol import (
     create_message, create_gzip_message, create_snappy_message
 )
-from kafka.producer import SimpleProducer, KeyedProducer
+from kafka.producer import SimpleProducer, KeyedProducer, KafkaProducer
 from kafka.partitioner import RoundRobinPartitioner, HashedPartitioner
 from kafka.consumer import SimpleConsumer, MultiProcessConsumer, KafkaConsumer
 
 __all__ = [
     'KafkaClient', 'KafkaConnection', 'SimpleProducer', 'KeyedProducer',
     'RoundRobinPartitioner', 'HashedPartitioner', 'SimpleConsumer',
     'MultiProcessConsumer', 'create_message', 'create_gzip_message',
-    'create_snappy_message', 'KafkaConsumer',
+    'create_snappy_message', 'KafkaConsumer', 'KafkaProducer'
 ]

kafka/common.py

@@ -102,9 +102,10 @@ class UnknownTopicOrPartitionError(BrokerResponseError):
     message = 'UNKNOWN_TOPIC_OR_PARTITON'
 
 
-class InvalidFetchRequestError(BrokerResponseError):
+# Formerly known (incorrectly) as InvalidFetchRequestError
+class InvalidMessageSizeError(BrokerResponseError):
     errno = 4
-    message = 'INVALID_FETCH_SIZE'
+    message = 'INVALID_MESSAGE_SIZE'
 
 
 class LeaderNotAvailableError(BrokerResponseError):

kafka/producer/__init__.py

@@ -1,6 +1,7 @@
 from .simple import SimpleProducer
 from .keyed import KeyedProducer
+from .kafka import KafkaProducer
 
 __all__ = [
-    'SimpleProducer', 'KeyedProducer'
+    'SimpleProducer', 'KeyedProducer', 'KafkaProducer'
 ]