[extractor/common] Auto calculate tbr when missing

pull/8338/head
Sergey M․ 2016-01-27 21:11:17 +06:00
parent fab6f0e65b
commit b0d21deda9
1 changed files with 6 additions and 0 deletions

View File

@ -825,6 +825,12 @@ class InfoExtractor(object):
if not formats: if not formats:
raise ExtractorError('No video formats found') raise ExtractorError('No video formats found')
for f in formats:
# Automatically determine tbr when missing based on abr and vbr (improves
# formats sorting in some cases)
if 'tbr' not in f and 'abr' in f and 'vbr' in f:
f['tbr'] = f['abr'] + f['vbr']
def _formats_key(f): def _formats_key(f):
# TODO remove the following workaround # TODO remove the following workaround
from ..utils import determine_ext from ..utils import determine_ext