[postprocessor] Fix chapters if duration is not extracted (#6037)

Authored by: bashonly
pull/6476/head
bashonly 2 years ago committed by GitHub
parent 6f4fc5660f
commit 01ddec7e66
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -302,6 +302,11 @@ class FFmpegPostProcessor(PostProcessor):
None) None)
return num, len(streams) return num, len(streams)
def _fixup_chapters(self, info):
last_chapter = traverse_obj(info, ('chapters', -1))
if last_chapter and not last_chapter.get('end_time'):
last_chapter['end_time'] = self._get_real_video_duration(info['filepath'])
def _get_real_video_duration(self, filepath, fatal=True): def _get_real_video_duration(self, filepath, fatal=True):
try: try:
duration = float_or_none( duration = float_or_none(
@ -678,6 +683,7 @@ class FFmpegMetadataPP(FFmpegPostProcessor):
@PostProcessor._restrict_to(images=False) @PostProcessor._restrict_to(images=False)
def run(self, info): def run(self, info):
self._fixup_chapters(info)
filename, metadata_filename = info['filepath'], None filename, metadata_filename = info['filepath'], None
files_to_delete, options = [], [] files_to_delete, options = [], []
if self._add_chapters and info.get('chapters'): if self._add_chapters and info.get('chapters'):
@ -1040,6 +1046,7 @@ class FFmpegSplitChaptersPP(FFmpegPostProcessor):
@PostProcessor._restrict_to(images=False) @PostProcessor._restrict_to(images=False)
def run(self, info): def run(self, info):
self._fixup_chapters(info)
chapters = info.get('chapters') or [] chapters = info.get('chapters') or []
if not chapters: if not chapters:
self.to_screen('Chapter information is unavailable') self.to_screen('Chapter information is unavailable')

@ -23,6 +23,7 @@ class ModifyChaptersPP(FFmpegPostProcessor):
@PostProcessor._restrict_to(images=False) @PostProcessor._restrict_to(images=False)
def run(self, info): def run(self, info):
self._fixup_chapters(info)
# Chapters must be preserved intact when downloading multiple formats of the same video. # Chapters must be preserved intact when downloading multiple formats of the same video.
chapters, sponsor_chapters = self._mark_chapters_to_remove( chapters, sponsor_chapters = self._mark_chapters_to_remove(
copy.deepcopy(info.get('chapters')) or [], copy.deepcopy(info.get('chapters')) or [],

Loading…
Cancel
Save