From aad256372bafd637f0a7d66a4271da114f8f6cb0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bruno=20Reni=C3=A9?= Date: Fri, 29 Aug 2014 10:13:15 +0200 Subject: Cross-python iteritems --- kafka/consumer.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'kafka/consumer.py') diff --git a/kafka/consumer.py b/kafka/consumer.py index 7636a83..d811147 100644 --- a/kafka/consumer.py +++ b/kafka/consumer.py @@ -10,6 +10,8 @@ import numbers from threading import Lock from multiprocessing import Process, Queue as MPQueue, Event, Value +import six + try: from Queue import Empty, Queue except ImportError: # python 2 @@ -427,7 +429,7 @@ class SimpleConsumer(Consumer): for p in self.fetch_offsets.keys()) while partitions: requests = [] - for partition, buffer_size in partitions.iteritems(): + for partition, buffer_size in six.iteritems(partitions): requests.append(FetchRequest(self.topic, partition, self.fetch_offsets[partition], buffer_size)) -- cgit v1.2.1