Python code ported to python3

2 files changed, 7 insertions, 1 deletions

diff --git a/librabbitmq/__init__.py b/librabbitmq/__init__.py
index c3c831c..974f1f6 100644
--- a/librabbitmq/__init__.py
+++ b/librabbitmq/__init__.py
@@ -1,6 +1,8 @@
 from __future__ import absolute_import
 
+import sys
 import itertools
+from six.moves import xrange
 
 import _librabbitmq
 
@@ -43,7 +45,10 @@ class Channel(object):
     def __init__(self, connection, channel_id):
         self.connection = connection
         self.channel_id = channel_id
-        self.next_consumer_tag = itertools.count(1).next
+        if sys.version_info.major == 2:
+            self.next_consumer_tag = itertools.count(1).next
+        else:
+            self.next_consumer_tag = itertools.count(1).__next__
         self.no_ack_consumers = set()
 
     def __enter__(self):
diff --git a/setup.py b/setup.py
index c37c999..e926c32 100644
--- a/setup.py
+++ b/setup.py
@@ -210,6 +210,7 @@ setup(
     cmdclass=cmdclass,
     install_requires=[
         'amqp>=1.4.6',
+        'six>=1.0.0',
     ],
     ext_modules=ext_modules,
     classifiers=[