hoelee/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2025-08-02 10:20:23 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

[MetadataFromField] Improve regex and add tests

Browse Source
This commit is contained in:
pukkandan
2021-04-21 11:12:04 +05:30
parent 8c54a3051d
commit a3faeb7de4
2 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
yt_dlp/postprocessor/metadatafromfield.py
View File

@@ -7,7 +7,7 @@ from ..compat import compat_str
class MetadataFromFieldPP(PostProcessor):
regex = r'(?P<in>.+):(?P<out>.+)$'
regex = r'(?P<in>.*?)(?<!\\):(?P<out>.+)$'
def __init__(self, downloader, formats):
PostProcessor.__init__(self, downloader)
@@ -17,10 +17,11 @@ class MetadataFromFieldPP(PostProcessor):
assert isinstance(f, compat_str)
match = re.match(self.regex, f)
assert match is not None
inp = match.group('in').replace('\\:', ':')
self._data.append({
'in': match.group('in'),
'in': inp,
'out': match.group('out'),
'tmpl': self.field_to_template(match.group('in')),
'tmpl': self.field_to_template(inp),
'regex': self.format_to_regex(match.group('out')),
})
@@ -68,6 +69,6 @@ class MetadataFromFieldPP(PostProcessor):
class MetadataFromTitlePP(MetadataFromFieldPP): # for backward compatibility
def __init__(self, downloader, titleformat):
super(MetadataFromTitlePP, self).__init__(downloader, ['title:%s' % titleformat])
super(MetadataFromTitlePP, self).__init__(downloader, ['%%(title)s:%s' % titleformat])
self._titleformat = titleformat
self._titleregex = self._data[0]['regex']