Browse Source

[extractor/common] Auto calculate tbr when missing

Sergey M․ 9 years ago
parent
commit
b0d21deda9
1 changed files with 6 additions and 0 deletions
  1. 6 0
      youtube_dl/extractor/common.py

+ 6 - 0
youtube_dl/extractor/common.py

@@ -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