Przeglądaj źródła

[abc] Use int_or_none

Yen Chi Hsuan 10 lat temu
rodzic
commit
c0a656876c
1 zmienionych plików z 5 dodań i 4 usunięć
  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 (
 from ..utils import (
     ExtractorError,
     ExtractorError,
     js_to_json,
     js_to_json,
+    int_or_none,
 )
 )
 
 
 
 
@@ -59,10 +60,10 @@ class ABCIE(InfoExtractor):
 
 
         formats = [{
         formats = [{
             'url': url_info['url'],
             '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]
         } for url_info in urls_info]
         self._sort_formats(formats)
         self._sort_formats(formats)