fix for python 3 compatibility

This commit is contained in:
root 2015-12-05 11:27:16 +00:00
parent 80688ea2b2
commit 480a30e922

View File

@ -4,10 +4,11 @@ import datetime
import time import time
import boto.kinesis import boto.kinesis
import re import re
import six
import itertools
from operator import attrgetter from operator import attrgetter
from hashlib import md5 from hashlib import md5
from itertools import izip_longest
from moto.compat import OrderedDict from moto.compat import OrderedDict
from moto.core import BaseBackend from moto.core import BaseBackend
@ -100,6 +101,11 @@ class Stream(object):
self.shards = {} self.shards = {}
self.tags = {} 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), for index, start, end in izip_longest(range(shard_count),
range(0,2**128,2**128/shard_count), range(0,2**128,2**128/shard_count),
range(2**128/shard_count,2**128,2**128/shard_count), range(2**128/shard_count,2**128,2**128/shard_count),