[InfoExtractor] Support groups in _`search_regex()`, etc

This commit is contained in:
dirkf 2023-05-03 13:08:58 +01:00
parent cb9366eda5
commit 1e8ccdd2eb
1 changed files with 5 additions and 4 deletions

View File

@ -1005,6 +1005,8 @@ class InfoExtractor(object):
if group is None:
# return the first matching group
return next(g for g in mobj.groups() if g is not None)
elif isinstance(group, (list, tuple)):
return tuple(mobj.group(g) for g in group)
else:
return mobj.group(group)
elif default is not NO_DEFAULT:
@ -1020,10 +1022,9 @@ class InfoExtractor(object):
Like _search_regex, but strips HTML tags and unescapes entities.
"""
res = self._search_regex(pattern, string, name, default, fatal, flags, group)
if res:
return clean_html(res).strip()
else:
return res
if isinstance(res, tuple):
return tuple(map(clean_html, res))
return clean_html(res)
def _get_netrc_login_info(self, netrc_machine=None):
username = None