mirror of
https://github.com/ytdl-org/youtube-dl
synced 2025-12-05 19:14:09 +00:00
[utils] Align parse_duration() behaviour with yt-dlp
* handle comma-separated long-form durations * support : as millisecond separator.
This commit is contained in:
parent
7a488f7fae
commit
43e3121020
@ -665,6 +665,8 @@ class TestUtil(unittest.TestCase):
|
|||||||
self.assertEqual(parse_duration('3h 11m 53s'), 11513)
|
self.assertEqual(parse_duration('3h 11m 53s'), 11513)
|
||||||
self.assertEqual(parse_duration('3 hours 11 minutes 53 seconds'), 11513)
|
self.assertEqual(parse_duration('3 hours 11 minutes 53 seconds'), 11513)
|
||||||
self.assertEqual(parse_duration('3 hours 11 mins 53 secs'), 11513)
|
self.assertEqual(parse_duration('3 hours 11 mins 53 secs'), 11513)
|
||||||
|
self.assertEqual(parse_duration('3 hours, 11 minutes, 53 seconds'), 11513)
|
||||||
|
self.assertEqual(parse_duration('3 hours, 11 mins, 53 secs'), 11513)
|
||||||
self.assertEqual(parse_duration('62m45s'), 3765)
|
self.assertEqual(parse_duration('62m45s'), 3765)
|
||||||
self.assertEqual(parse_duration('6m59s'), 419)
|
self.assertEqual(parse_duration('6m59s'), 419)
|
||||||
self.assertEqual(parse_duration('49s'), 49)
|
self.assertEqual(parse_duration('49s'), 49)
|
||||||
@ -683,6 +685,10 @@ class TestUtil(unittest.TestCase):
|
|||||||
self.assertEqual(parse_duration('PT1H0.040S'), 3600.04)
|
self.assertEqual(parse_duration('PT1H0.040S'), 3600.04)
|
||||||
self.assertEqual(parse_duration('PT00H03M30SZ'), 210)
|
self.assertEqual(parse_duration('PT00H03M30SZ'), 210)
|
||||||
self.assertEqual(parse_duration('P0Y0M0DT0H4M20.880S'), 260.88)
|
self.assertEqual(parse_duration('P0Y0M0DT0H4M20.880S'), 260.88)
|
||||||
|
self.assertEqual(parse_duration('01:02:03:050'), 3723.05)
|
||||||
|
self.assertEqual(parse_duration('103:050'), 103.05)
|
||||||
|
self.assertEqual(parse_duration('1HR 3MIN'), 3780)
|
||||||
|
self.assertEqual(parse_duration('2hrs 3mins'), 7380)
|
||||||
|
|
||||||
def test_fix_xml_ampersands(self):
|
def test_fix_xml_ampersands(self):
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
|
|||||||
@ -3931,38 +3931,46 @@ def parse_duration(s):
|
|||||||
return None
|
return None
|
||||||
|
|
||||||
s = s.strip()
|
s = s.strip()
|
||||||
|
if not s:
|
||||||
|
return None
|
||||||
|
|
||||||
days, hours, mins, secs, ms = [None] * 5
|
days, hours, mins, secs, ms = [None] * 5
|
||||||
m = re.match(r'(?:(?:(?:(?P<days>[0-9]+):)?(?P<hours>[0-9]+):)?(?P<mins>[0-9]+):)?(?P<secs>[0-9]+)(?P<ms>\.[0-9]+)?Z?$', s)
|
m = re.match(r'''(?x)
|
||||||
|
(?P<before_secs>
|
||||||
|
(?:(?:(?P<days>[0-9]+):)?(?P<hours>[0-9]+):)?
|
||||||
|
(?P<mins>[0-9]+):)?
|
||||||
|
(?P<secs>(?(before_secs)[0-9]{1,2}|[0-9]+))
|
||||||
|
(?:[.:](?P<ms>[0-9]+))?Z?$
|
||||||
|
''', s)
|
||||||
if m:
|
if m:
|
||||||
days, hours, mins, secs, ms = m.groups()
|
days, hours, mins, secs, ms = m.group('days', 'hours', 'mins', 'secs', 'ms')
|
||||||
else:
|
else:
|
||||||
m = re.match(
|
m = re.match(
|
||||||
r'''(?ix)(?:P?
|
r'''(?ix)(?:P?
|
||||||
(?:
|
(?:
|
||||||
[0-9]+\s*y(?:ears?)?\s*
|
[0-9]+\s*y(?:ears?)?,?\s*
|
||||||
)?
|
)?
|
||||||
(?:
|
(?:
|
||||||
[0-9]+\s*m(?:onths?)?\s*
|
[0-9]+\s*m(?:onths?)?,?\s*
|
||||||
)?
|
)?
|
||||||
(?:
|
(?:
|
||||||
[0-9]+\s*w(?:eeks?)?\s*
|
[0-9]+\s*w(?:eeks?)?,?\s*
|
||||||
)?
|
)?
|
||||||
(?:
|
(?:
|
||||||
(?P<days>[0-9]+)\s*d(?:ays?)?\s*
|
(?P<days>[0-9]+)\s*d(?:ays?)?,?\s*
|
||||||
)?
|
)?
|
||||||
T)?
|
T)?
|
||||||
(?:
|
(?:
|
||||||
(?P<hours>[0-9]+)\s*h(?:ours?)?\s*
|
(?P<hours>[0-9]+)\s*h(?:(?:ou)?rs?)?,?\s*
|
||||||
)?
|
)?
|
||||||
(?:
|
(?:
|
||||||
(?P<mins>[0-9]+)\s*m(?:in(?:ute)?s?)?\s*
|
(?P<mins>[0-9]+)\s*m(?:in(?:ute)?s?)?,?\s*
|
||||||
)?
|
)?
|
||||||
(?:
|
(?:
|
||||||
(?P<secs>[0-9]+)(?P<ms>\.[0-9]+)?\s*s(?:ec(?:ond)?s?)?\s*
|
(?P<secs>[0-9]+)(?:\.(?P<ms>[0-9]+))?\s*s(?:ec(?:ond)?s?)?\s*
|
||||||
)?Z?$''', s)
|
)?Z?$''', s)
|
||||||
if m:
|
if m:
|
||||||
days, hours, mins, secs, ms = m.groups()
|
days, hours, mins, secs, ms = m.group('days', 'hours', 'mins', 'secs', 'ms')
|
||||||
else:
|
else:
|
||||||
m = re.match(r'(?i)(?:(?P<hours>[0-9.]+)\s*(?:hours?)|(?P<mins>[0-9.]+)\s*(?:mins?\.?|minutes?)\s*)Z?$', s)
|
m = re.match(r'(?i)(?:(?P<hours>[0-9.]+)\s*(?:hours?)|(?P<mins>[0-9.]+)\s*(?:mins?\.?|minutes?)\s*)Z?$', s)
|
||||||
if m:
|
if m:
|
||||||
@ -3970,17 +3978,13 @@ def parse_duration(s):
|
|||||||
else:
|
else:
|
||||||
return None
|
return None
|
||||||
|
|
||||||
duration = 0
|
duration = (
|
||||||
if secs:
|
((((float(days) * 24) if days else 0)
|
||||||
duration += float(secs)
|
+ (float(hours) if hours else 0)) * 60
|
||||||
if mins:
|
+ (float(mins) if mins else 0)) * 60
|
||||||
duration += float(mins) * 60
|
+ (float(secs) if secs else 0)
|
||||||
if hours:
|
+ (float(ms) / 10 ** len(ms) if ms else 0))
|
||||||
duration += float(hours) * 60 * 60
|
|
||||||
if days:
|
|
||||||
duration += float(days) * 24 * 60 * 60
|
|
||||||
if ms:
|
|
||||||
duration += float(ms)
|
|
||||||
return duration
|
return duration
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user