Python using Regular Expression to convert a string

user_7771848 0
Viewed 356

I want to convert the duration variable from YouTube Data api?

PT1M6S --> 1:06
PT38S --> 0:38
PT58M4 --> 58:04

Here is my codes:

p['duration'] is the value from json data

duration = re.sub(r'^PT',r'',p['duration'])
duration = re.sub(r'M',r':',duration)
if (len(p['duration']) > 5 ):
    duration = re.sub(r'S',r'',duration)
else:
    duration = "0:" + re.sub(r'S',r'',duration)

Is there a simple way to do in one regex statement?

Thanks!

python regex
2 Answers

You can use

import re
strings = ['PT1M6S', 'PT38S', 'PT58M4']
rx = re.compile(r'^PT(?:(\d+)M)?(?:(\d+)S?)?$')
for s in strings:
  print(s, ' => ', rx.sub(lambda x: f"{(x.group(1) or '00').zfill(2)}:{(x.group(2) or '00').zfill(2)}", s))

See the online demo. Output:

PT1M6S  =>  01:06
PT38S  =>  00:38
PT58M4  =>  58:04

The regex demo is ^PT(?:(\d+)M)?(?:(\d+)S?)?$, it matches

  • ^PT - PT at the start of a string
  • (?:(\d+)M)? - an optional non-capturing group matching 1+ digits (capturing them into Group 1) and an M char
  • (?:(\d+)S?)? - an optional non-capturing group matching 1+ digits (capturing them into Group 2) and an optional S char
  • $ - end of string.
user_38329700

An alternative to using a regex is using parser from dateutil. It has an option fuzzy that you can use to convert your data to datetime. If you subtract midnight today from that, you get the value as a timedelta:

from dateutil import parser
from datetime import date
from datetime import datetime

lst = ['PT1M6S','PT38S', 'PT58M4']

for t in lst:
    print(parser.parse(t, fuzzy=True) - datetime.combine(date.today(), datetime.min.time()))

gives you

0:01:06
0:00:38
0:58:04
user_140157370
Related