mirror of
https://github.com/ytdl-org/youtube-dl
synced 2024-08-28 20:15:10 +02:00
[utils] add helper function for parsing codecs
This commit is contained in:
parent
1ac5705f62
commit
4f3c5e0627
@ -78,6 +78,7 @@ from youtube_dl.utils import (
|
|||||||
cli_option,
|
cli_option,
|
||||||
cli_valueless_option,
|
cli_valueless_option,
|
||||||
cli_bool_option,
|
cli_bool_option,
|
||||||
|
parse_codecs,
|
||||||
)
|
)
|
||||||
from youtube_dl.compat import (
|
from youtube_dl.compat import (
|
||||||
compat_chr,
|
compat_chr,
|
||||||
@ -579,6 +580,29 @@ class TestUtil(unittest.TestCase):
|
|||||||
limit_length('foo bar baz asd', 12).startswith('foo bar'))
|
limit_length('foo bar baz asd', 12).startswith('foo bar'))
|
||||||
self.assertTrue('...' in limit_length('foo bar baz asd', 12))
|
self.assertTrue('...' in limit_length('foo bar baz asd', 12))
|
||||||
|
|
||||||
|
def test_parse_codecs(self):
|
||||||
|
self.assertEqual(parse_codecs(''), {})
|
||||||
|
self.assertEqual(parse_codecs('avc1.77.30, mp4a.40.2'), {
|
||||||
|
'vcodec': 'avc1.77.30',
|
||||||
|
'acodec': 'mp4a.40.2',
|
||||||
|
})
|
||||||
|
self.assertEqual(parse_codecs('mp4a.40.2'), {
|
||||||
|
'vcodec': 'none',
|
||||||
|
'acodec': 'mp4a.40.2',
|
||||||
|
})
|
||||||
|
self.assertEqual(parse_codecs('mp4a.40.5,avc1.42001e'), {
|
||||||
|
'vcodec': 'avc1.42001e',
|
||||||
|
'acodec': 'mp4a.40.5',
|
||||||
|
})
|
||||||
|
self.assertEqual(parse_codecs('avc3.640028'), {
|
||||||
|
'vcodec': 'avc3.640028',
|
||||||
|
'acodec': 'none',
|
||||||
|
})
|
||||||
|
self.assertEqual(parse_codecs(', h264,,newcodec,aac'), {
|
||||||
|
'vcodec': 'h264',
|
||||||
|
'acodec': 'aac',
|
||||||
|
})
|
||||||
|
|
||||||
def test_escape_rfc3986(self):
|
def test_escape_rfc3986(self):
|
||||||
reserved = "!*'();:@&=+$,/?#[]"
|
reserved = "!*'();:@&=+$,/?#[]"
|
||||||
unreserved = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_.~'
|
unreserved = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_.~'
|
||||||
|
@ -2060,6 +2060,42 @@ def mimetype2ext(mt):
|
|||||||
}.get(res, res)
|
}.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):
|
def urlhandle_detect_ext(url_handle):
|
||||||
getheader = url_handle.headers.get
|
getheader = url_handle.headers.get
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user