aboutsummaryrefslogtreecommitdiffstats
path: root/youtube_dl/postprocessor/metadatafromtitle.py
diff options
context:
space:
mode:
authorUnknown <blackjack4494@web.de>2020-09-02 23:33:41 +0200
committerUnknown <blackjack4494@web.de>2020-09-02 23:33:41 +0200
commit3ca3f77f9ce9dd504dc6af4ef605c245c31ff860 (patch)
tree6bd9c9352327148a78b8c46227c8d526f1447b03 /youtube_dl/postprocessor/metadatafromtitle.py
parent4cd6add62b54721eeb3bf76bd9c0b4d676dc4d68 (diff)
downloadhypervideo-pre-3ca3f77f9ce9dd504dc6af4ef605c245c31ff860.tar.lz
hypervideo-pre-3ca3f77f9ce9dd504dc6af4ef605c245c31ff860.tar.xz
hypervideo-pre-3ca3f77f9ce9dd504dc6af4ef605c245c31ff860.zip
[skip travis] adding automerge support
basically copying content of youtube_dl folder to youtube_dlc and excluding the youtube_dl folder when compiling
Diffstat (limited to 'youtube_dl/postprocessor/metadatafromtitle.py')
-rw-r--r--youtube_dl/postprocessor/metadatafromtitle.py48
1 files changed, 48 insertions, 0 deletions
diff --git a/youtube_dl/postprocessor/metadatafromtitle.py b/youtube_dl/postprocessor/metadatafromtitle.py
new file mode 100644
index 000000000..f5c14d974
--- /dev/null
+++ b/youtube_dl/postprocessor/metadatafromtitle.py
@@ -0,0 +1,48 @@
+from __future__ import unicode_literals
+
+import re
+
+from .common import PostProcessor
+
+
+class MetadataFromTitlePP(PostProcessor):
+ def __init__(self, downloader, titleformat):
+ super(MetadataFromTitlePP, self).__init__(downloader)
+ self._titleformat = titleformat
+ self._titleregex = (self.format_to_regex(titleformat)
+ if re.search(r'%\(\w+\)s', titleformat)
+ else titleformat)
+
+ def format_to_regex(self, fmt):
+ r"""
+ Converts a string like
+ '%(title)s - %(artist)s'
+ to a regex like
+ '(?P<title>.+)\ \-\ (?P<artist>.+)'
+ """
+ lastpos = 0
+ regex = ''
+ # replace %(..)s with regex group and escape other string parts
+ for match in re.finditer(r'%\((\w+)\)s', fmt):
+ regex += re.escape(fmt[lastpos:match.start()])
+ regex += r'(?P<' + match.group(1) + '>.+)'
+ lastpos = match.end()
+ if lastpos < len(fmt):
+ regex += re.escape(fmt[lastpos:])
+ return regex
+
+ def run(self, info):
+ title = info['title']
+ match = re.match(self._titleregex, title)
+ if match is None:
+ self._downloader.to_screen(
+ '[fromtitle] Could not interpret title of video as "%s"'
+ % self._titleformat)
+ return [], info
+ for attribute, value in match.groupdict().items():
+ info[attribute] = value
+ self._downloader.to_screen(
+ '[fromtitle] parsed %s: %s'
+ % (attribute, value if value is not None else 'NA'))
+
+ return [], info