[netease:djradio] Use compat_itertools_count

This commit is contained in:
Yen Chi Hsuan 2015-07-10 18:59:12 +08:00
parent a0e060ac1e
commit 4eab60cbd2

View file

@ -4,7 +4,6 @@
from hashlib import md5 from hashlib import md5
from base64 import b64encode from base64 import b64encode
from datetime import datetime from datetime import datetime
import itertools
import re import re
from .common import InfoExtractor from .common import InfoExtractor
@ -12,6 +11,7 @@
compat_urllib_request, compat_urllib_request,
compat_urllib_parse, compat_urllib_parse,
compat_str, compat_str,
compat_itertools_count,
) )
@ -428,7 +428,7 @@ def _real_extract(self, url):
name = None name = None
desc = None desc = None
entries = [] entries = []
for offset in itertools.count(start=0, step=self._PAGE_SIZE): for offset in compat_itertools_count(start=0, step=self._PAGE_SIZE):
info = self.query_api( info = self.query_api(
'dj/program/byradio?asc=false&limit=%d&radioId=%s&offset=%d' 'dj/program/byradio?asc=false&limit=%d&radioId=%s&offset=%d'
% (self._PAGE_SIZE, dj_id, offset), % (self._PAGE_SIZE, dj_id, offset),