Przeglądaj źródła

[extractor/common] Add more subtitle mime types for guess when ext is missing

Sergey M․ 10 lat temu
rodzic
commit
5cdefc4625
1 zmienionych plików z 7 dodań i 2 usunięć
  1. 7 2
      youtube_dl/extractor/common.py

+ 7 - 2
youtube_dl/extractor/common.py

@@ -1144,8 +1144,13 @@ class InfoExtractor(object):
             ext = textstream.get('ext') or determine_ext(src)
             ext = textstream.get('ext') or determine_ext(src)
             if not ext:
             if not ext:
                 type_ = textstream.get('type')
                 type_ = textstream.get('type')
-                if type_ == 'text/srt':
-                    ext = 'srt'
+                SUBTITLES_TYPES = {
+                    'text/vtt': 'vtt',
+                    'text/srt': 'srt',
+                    'application/smptett+xml': 'tt',
+                }
+                if type_ in SUBTITLES_TYPES:
+                    ext = SUBTITLES_TYPES[type_]
             lang = textstream.get('systemLanguage') or textstream.get('systemLanguageName') or subtitles_lang
             lang = textstream.get('systemLanguage') or textstream.get('systemLanguageName') or subtitles_lang
             subtitles.setdefault(lang, []).append({
             subtitles.setdefault(lang, []).append({
                 'url': src,
                 'url': src,