diff options
Diffstat (limited to 'kafka/consumer.py')
| -rw-r--r-- | kafka/consumer.py | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/kafka/consumer.py b/kafka/consumer.py index 0935dd2..7636a83 100644 --- a/kafka/consumer.py +++ b/kafka/consumer.py @@ -1,12 +1,19 @@ from __future__ import absolute_import -from itertools import izip_longest, repeat +try: + from itertools import zip_longest as izip_longest, repeat # pylint: disable-msg=E0611 +except ImportError: # python 2 + from itertools import izip_longest as izip_longest, repeat import logging import time import numbers from threading import Lock from multiprocessing import Process, Queue as MPQueue, Event, Value -from Queue import Empty, Queue + +try: + from Queue import Empty, Queue +except ImportError: # python 2 + from queue import Empty, Queue import kafka from kafka.common import ( |