diff options
| author | Ajay Ramachandran <dev@ajay.app> | 2022-10-18 12:51:57 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-18 22:21:57 +0530 |
| commit | 63c547d71ceae6be181948b4b6ce4180b16f4209 (patch) | |
| tree | 9e08a01f9e7203135985661fe0605808bbc8c249 /yt_dlp/postprocessor/sponsorblock.py | |
| parent | 814bba3933ca36a79c68ac737b805cf25c407521 (diff) | |
| download | hypervideo-pre-63c547d71ceae6be181948b4b6ce4180b16f4209.tar.lz hypervideo-pre-63c547d71ceae6be181948b4b6ce4180b16f4209.tar.xz hypervideo-pre-63c547d71ceae6be181948b4b6ce4180b16f4209.zip | |
[SponsorBlock] Support `chapter` category (#5260)
Authored by: ajayyy, pukkandan
Diffstat (limited to 'yt_dlp/postprocessor/sponsorblock.py')
| -rw-r--r-- | yt_dlp/postprocessor/sponsorblock.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/yt_dlp/postprocessor/sponsorblock.py b/yt_dlp/postprocessor/sponsorblock.py index d79ed7ae7..befff0e1f 100644 --- a/yt_dlp/postprocessor/sponsorblock.py +++ b/yt_dlp/postprocessor/sponsorblock.py @@ -14,6 +14,10 @@ class SponsorBlockPP(FFmpegPostProcessor): POI_CATEGORIES = { 'poi_highlight': 'Highlight', } + NON_SKIPPABLE_CATEGORIES = { + **POI_CATEGORIES, + 'chapter': 'Chapter', + } CATEGORIES = { 'sponsor': 'Sponsor', 'intro': 'Intermission/Intro Animation', @@ -23,7 +27,7 @@ class SponsorBlockPP(FFmpegPostProcessor): 'filler': 'Filler Tangent', 'interaction': 'Interaction Reminder', 'music_offtopic': 'Non-Music Section', - **POI_CATEGORIES, + **NON_SKIPPABLE_CATEGORIES } def __init__(self, downloader, categories=None, api='https://sponsor.ajay.app'): @@ -68,12 +72,13 @@ class SponsorBlockPP(FFmpegPostProcessor): def to_chapter(s): (start, end), cat = s['segment'], s['category'] + title = s['description'] if cat == 'chapter' else self.CATEGORIES[cat] return { 'start_time': start, 'end_time': end, 'category': cat, - 'title': self.CATEGORIES[cat], - '_categories': [(cat, start, end)] + 'title': title, + '_categories': [(cat, start, end, title)], } sponsor_chapters = [to_chapter(s) for s in duration_match] @@ -89,7 +94,7 @@ class SponsorBlockPP(FFmpegPostProcessor): url = f'{self._API_URL}/api/skipSegments/{hash[:4]}?' + urllib.parse.urlencode({ 'service': service, 'categories': json.dumps(self._categories), - 'actionTypes': json.dumps(['skip', 'poi']) + 'actionTypes': json.dumps(['skip', 'poi', 'chapter']) }) for d in self._download_json(url) or []: if d['videoID'] == video_id: |