avoid traverse_obj function

This commit is contained in:
Petr Vaněk 2022-02-12 12:28:30 +01:00 committed by dirkf
parent 8088ce036a
commit b1297308fb

View File

@ -7,7 +7,6 @@ from ..utils import (
float_or_none, float_or_none,
int_or_none, int_or_none,
parse_codecs, parse_codecs,
traverse_obj,
urljoin, urljoin,
) )
@ -38,7 +37,7 @@ class StreamCZIE(InfoExtractor):
def _extract_formats(self, spl_url, video): def _extract_formats(self, spl_url, video):
for ext, pref, streams in ( for ext, pref, streams in (
('ts', -1, traverse_obj(video, ('http_stream', 'qualities'))), ('ts', -1, video.get('http_stream', {}).get('qualities', {})),
('mp4', 1, video.get('mp4'))): ('mp4', 1, video.get('mp4'))):
for format_id, stream in streams.items(): for format_id, stream in streams.items():
if not stream.get('url'): if not stream.get('url'):
@ -50,8 +49,8 @@ class StreamCZIE(InfoExtractor):
'url': urljoin(spl_url, stream['url']), 'url': urljoin(spl_url, stream['url']),
'tbr': float_or_none(stream.get('bandwidth'), scale=1000), 'tbr': float_or_none(stream.get('bandwidth'), scale=1000),
'duration': float_or_none(stream.get('duration'), scale=1000), 'duration': float_or_none(stream.get('duration'), scale=1000),
'width': traverse_obj(stream, ('resolution', 0)), 'width': stream.get('resolution', 2 * [0])[0] or None,
'height': traverse_obj(stream, ('resolution', 1)) or int_or_none(format_id.replace('p', '')), 'height': stream.get('resolution', 2 * [0])[1] or int_or_none(format_id.replace('p', '')),
**parse_codecs(stream.get('codec')), **parse_codecs(stream.get('codec')),
} }