add __all__ to all modules. fixes #141

pykafka/__init__.py

@@ -9,3 +9,6 @@
 from topic import Topic
 from client import KafkaClient
 from balancedconsumer import BalancedConsumer
+
+__all__ = ["Broker", "SimpleConsumer", "Cluster", "Partition", "Producer",
+           "Topic", "KafkaClient", "BalancedConsumer"]

pykafka/balancedconsumer.py

@@ -17,6 +17,7 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
+__all__ = ["BalancedConsumer"]
 import itertools
 import logging as log
 import socket

pykafka/base.py

@@ -16,6 +16,8 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
+__all__ = ["BaseCluster", "BaseBroker", "BasePartition", "BaseTopic",
+           "BaseSimpleConsumer", "BaseProducer", "BaseAsyncProducer"]
 from common import CompressionType
 from partitioners import random_partitioner
 

pykafka/broker.py

@@ -16,6 +16,7 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
+__all__ = ["Broker"]
 import logging
 import time
 

pykafka/client.py

@@ -17,6 +17,7 @@
 limitations under the License.
 """
 
+__all__ = ["KafkaClient"]
 import handlers
 import logging
 from cluster import Cluster

pykafka/cluster.py

@@ -17,7 +17,7 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
-
+__all__ = ["Cluster"]
 import logging
 import random
 import time

pykafka/common.py

@@ -17,7 +17,7 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
-
+__all__ = ["Message", "CompressionType", "OffsetType"]
 import logging
 
 

pykafka/connection.py

@@ -17,7 +17,7 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
-
+__all__ = ["BrokerConnection"]
 import logging
 import socket
 import struct

pykafka/handlers.py

@@ -16,7 +16,7 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
-
+__all__ = ["ResponseFuture", "Handler", "ThreadingHandler", "RequestHandler"]
 import atexit
 import threading
 import Queue

pykafka/partition.py

@@ -16,6 +16,7 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
+__all__ = ["Partition"]
 import logging
 
 import base

pykafka/partitioners.py

@@ -16,7 +16,8 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
-
+__all__ = ["random_partitioner", "BasePartitioner", "HashingPartitioner",
+           "hashing_partitioner"]
 import random
 
 

pykafka/producer.py

@@ -17,7 +17,7 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
-
+__all__ = ["Producer", "AsyncProducer"]
 import itertools
 import logging
 import time

pykafka/protocol.py

@@ -44,6 +44,16 @@
   CorrelationId => int32
   ResponseMessage => MetadataResponse | ProduceResponse | FetchResponse | OffsetResponse | OffsetCommitResponse | OffsetFetchResponse
 """
+__all__ = [
+    "MetadataRequest", "MetadataResponse", "ProduceRequest", "ProduceResponse",
+    "OffsetRequest", "OffsetResponse", "OffsetCommitRequest",
+    "FetchRequest", "FetchResponse", "PartitionFetchRequest",
+    "OffsetCommitResponse", "OffsetFetchRequest", "OffsetFetchResponse",
+    "PartitionOffsetRequest", "ConsumerMetadataRequest",
+    "ConsumerMetadataResponse", "PartitionOffsetCommitRequest",
+    "PartitionOffsetFetchRequest",
+    "Request", "Response", "Message", "MessageSet"
+]
 import itertools
 import logging
 import struct

pykafka/simpleconsumer.py

@@ -16,6 +16,7 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
+__all__ = ["SimpleConsumer"]
 import itertools
 import logging as log
 import time

pykafka/topic.py

@@ -16,6 +16,7 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
+__all__ = ["Topic"]
 import logging
 import weakref
 from collections import defaultdict

pykafka/utils/compression.py

@@ -16,7 +16,7 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
-
+__all__ = ["encode_gzip", "decode_gzip", "encode_snappy", "decode_snappy"]
 import gzip
 import logging
 

pykafka/utils/error_handlers.py

@@ -16,6 +16,7 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
+__all__ = ["handle_partition_responses", "raise_error"]
 from collections import defaultdict
 
 

pykafka/utils/socket.py

@@ -16,7 +16,7 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
-
+__all__ = ["recvall_into"]
 from pykafka.exceptions import SocketDisconnectedError
 
 

pykafka/utils/struct_helpers.py

@@ -16,7 +16,7 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
-
+__all__ = ["unpack_from"]
 import itertools
 import struct