diff options
| author | Yen Chi Hsuan <[email protected]> | 2015-07-10 18:58:12 +0800 |
|---|---|---|
| committer | Yen Chi Hsuan <[email protected]> | 2015-07-10 18:58:12 +0800 |
| commit | a0e060ac1e9871974c02fe7cb3e812a3cbe79e73 (patch) | |
| tree | 9bca8595f1fc04578f2238c4fd202abba5b9bb8d /youtube_dl/compat.py | |
| parent | 397a8ea96e228f918f127313f837f7de7dd78aed (diff) | |
| download | youtube-dl-a0e060ac1e9871974c02fe7cb3e812a3cbe79e73.tar.gz youtube-dl-a0e060ac1e9871974c02fe7cb3e812a3cbe79e73.zip | |
[compat] Add compat_itertools_count
'step' parameter is added in Python 2.7
Diffstat (limited to 'youtube_dl/compat.py')
| -rw-r--r-- | youtube_dl/compat.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/youtube_dl/compat.py b/youtube_dl/compat.py index f9529210d..c3783337a 100644 --- a/youtube_dl/compat.py +++ b/youtube_dl/compat.py @@ -9,6 +9,7 @@ import shutil import socket import subprocess import sys +import itertools try: @@ -388,6 +389,15 @@ else: pass 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__ = [ 'compat_HTTPError', @@ -401,6 +411,7 @@ __all__ = [ 'compat_html_entities', 'compat_http_client', 'compat_http_server', + 'compat_itertools_count', 'compat_kwargs', 'compat_ord', 'compat_parse_qs', |