[compat] Add compat_itertools_count

'step' parameter is added in Python 2.7
This commit is contained in:
Yen Chi Hsuan 2015-07-10 18:58:12 +08:00
parent 397a8ea96e
commit a0e060ac1e
1 changed files with 11 additions and 0 deletions

View File

@ -9,6 +9,7 @@ import shutil
import socket import socket
import subprocess import subprocess
import sys import sys
import itertools
try: try:
@ -388,6 +389,15 @@ else:
pass pass
return _terminal_size(columns, lines) return _terminal_size(columns, lines)
try:
itertools.count(start=0, step=1)
compat_itertools_count = itertools.count
except TypeError: # Python 2.6
def compat_itertools_count(start=0, step=1):
n = start
while True:
yield n
n += step
__all__ = [ __all__ = [
'compat_HTTPError', 'compat_HTTPError',
@ -401,6 +411,7 @@ __all__ = [
'compat_html_entities', 'compat_html_entities',
'compat_http_client', 'compat_http_client',
'compat_http_server', 'compat_http_server',
'compat_itertools_count',
'compat_kwargs', 'compat_kwargs',
'compat_ord', 'compat_ord',
'compat_parse_qs', 'compat_parse_qs',