aboutsummaryrefslogtreecommitdiff
path: root/youtube_dl
diff options
context:
space:
mode:
authorremitamine <remitamine@gmail.com>2016-03-16 18:48:06 +0100
committerRemita Amine <remitamine@gmail.com>2016-06-26 14:03:58 +0100
commit4f3c5e062715bb8c2084bda139ddcd9a2036f267 (patch)
tree635303c308485ea0ce02113eb6ac0e8cc0c3dc26 /youtube_dl
parent1ac5705f62aa3f6fdb6f2a97fbd24594010b7598 (diff)
[utils] add helper function for parsing codecs
Diffstat (limited to 'youtube_dl')
-rw-r--r--youtube_dl/utils.py36
1 files changed, 36 insertions, 0 deletions
diff --git a/youtube_dl/utils.py b/youtube_dl/utils.py
index 562031fe1..fe175e82c 100644
--- a/youtube_dl/utils.py
+++ b/youtube_dl/utils.py
@@ -2060,6 +2060,42 @@ def mimetype2ext(mt):
}.get(res, res)
+def parse_codecs(codecs_str):
+ # http://tools.ietf.org/html/rfc6381
+ if not codecs_str:
+ return {}
+ splited_codecs = list(filter(None, map(
+ lambda str: str.strip(), codecs_str.strip().strip(',').split(','))))
+ vcodec, acodec = None, None
+ for full_codec in splited_codecs:
+ codec = full_codec.split('.')[0]
+ if codec in ('avc1', 'avc2', 'avc3', 'avc4', 'vp9', 'vp8', 'hev1', 'hev2', 'h263', 'h264', 'mp4v'):
+ if not vcodec:
+ vcodec = full_codec
+ elif codec in ('mp4a', 'opus', 'vorbis', 'mp3', 'aac'):
+ if not acodec:
+ acodec = full_codec
+ else:
+ write_string('WARNING: Unknown codec %s' % full_codec, sys.stderr)
+ if not vcodec and not acodec:
+ if len(splited_codecs) == 2:
+ return {
+ 'vcodec': vcodec,
+ 'acodec': acodec,
+ }
+ elif len(splited_codecs) == 1:
+ return {
+ 'vcodec': 'none',
+ 'acodec': vcodec,
+ }
+ else:
+ return {
+ 'vcodec': vcodec or 'none',
+ 'acodec': acodec or 'none',
+ }
+ return {}
+
+
def urlhandle_detect_ext(url_handle):
getheader = url_handle.headers.get