Explorar o código

[netease:djradio] Use compat_itertools_count

Yen Chi Hsuan %!s(int64=10) %!d(string=hai) anos
pai
achega
4eab60cbd2
Modificáronse 1 ficheiros con 2 adicións e 2 borrados
  1. 2 2
      youtube_dl/extractor/neteasemusic.py

+ 2 - 2
youtube_dl/extractor/neteasemusic.py

@@ -4,7 +4,6 @@ from __future__ import unicode_literals
 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 @@ from ..compat import (
     compat_urllib_request,
     compat_urllib_request,
     compat_urllib_parse,
     compat_urllib_parse,
     compat_str,
     compat_str,
+    compat_itertools_count,
 )
 )
 
 
 
 
@@ -428,7 +428,7 @@ class NetEaseMusicDjRadioIE(NetEaseMusicBaseIE):
         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),