소스 검색

[abc] Use int_or_none

Yen Chi Hsuan 10 년 전
부모
커밋
c0a656876c
1개의 변경된 파일5개의 추가작업 그리고 4개의 파일을 삭제
  1. 5 4
      youtube_dl/extractor/abc.py

+ 5 - 4
youtube_dl/extractor/abc.py

@@ -6,6 +6,7 @@ from .common import InfoExtractor
 from ..utils import (
     ExtractorError,
     js_to_json,
+    int_or_none,
 )
 
 
@@ -59,10 +60,10 @@ class ABCIE(InfoExtractor):
 
         formats = [{
             'url': url_info['url'],
-            'width': int(url_info['width']),
-            'height': int(url_info['height']),
-            'tbr': int(url_info['bitrate']),
-            'filesize': int(url_info['filesize']),
+            'width': int_or_none(url_info.get('width')),
+            'height': int_or_none(url_info.get('height')),
+            'tbr': int_or_none(url_info.get('bitrate')),
+            'filesize': int_or_none(url_info.get('filesize')),
         } for url_info in urls_info]
         self._sort_formats(formats)