From 480a30e92276b43077032a4b45d8a7ae3ae14cd7 Mon Sep 17 00:00:00 2001 From: root Date: Sat, 5 Dec 2015 11:27:16 +0000 Subject: [PATCH] fix for python 3 compatibility --- moto/kinesis/models.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/moto/kinesis/models.py b/moto/kinesis/models.py index 40c8fec5d..01a83a73e 100644 --- a/moto/kinesis/models.py +++ b/moto/kinesis/models.py @@ -4,10 +4,11 @@ import datetime import time import boto.kinesis import re +import six +import itertools from operator import attrgetter from hashlib import md5 -from itertools import izip_longest from moto.compat import OrderedDict from moto.core import BaseBackend @@ -100,6 +101,11 @@ class Stream(object): self.shards = {} self.tags = {} + if six.PY3: + izip_longest = itertools.zip_longest + else: + izip_longest = itertools.izip_longest + for index, start, end in izip_longest(range(shard_count), range(0,2**128,2**128/shard_count), range(2**128/shard_count,2**128,2**128/shard_count),