mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-21 20:46:36 -05:00
[postprocessor] Fix chapters if duration is not extracted (#6037)
Authored by: bashonly
This commit is contained in:
parent
6f4fc5660f
commit
01ddec7e66
2 changed files with 8 additions and 0 deletions
|
@ -302,6 +302,11 @@ def get_stream_number(self, path, keys, value):
|
|||
None)
|
||||
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):
|
||||
try:
|
||||
duration = float_or_none(
|
||||
|
@ -678,6 +683,7 @@ def _options(target_ext):
|
|||
|
||||
@PostProcessor._restrict_to(images=False)
|
||||
def run(self, info):
|
||||
self._fixup_chapters(info)
|
||||
filename, metadata_filename = info['filepath'], None
|
||||
files_to_delete, options = [], []
|
||||
if self._add_chapters and info.get('chapters'):
|
||||
|
@ -1040,6 +1046,7 @@ def _ffmpeg_args_for_chapter(self, number, chapter, info):
|
|||
|
||||
@PostProcessor._restrict_to(images=False)
|
||||
def run(self, info):
|
||||
self._fixup_chapters(info)
|
||||
chapters = info.get('chapters') or []
|
||||
if not chapters:
|
||||
self.to_screen('Chapter information is unavailable')
|
||||
|
|
|
@ -23,6 +23,7 @@ def __init__(self, downloader, remove_chapters_patterns=None, remove_sponsor_seg
|
|||
|
||||
@PostProcessor._restrict_to(images=False)
|
||||
def run(self, info):
|
||||
self._fixup_chapters(info)
|
||||
# Chapters must be preserved intact when downloading multiple formats of the same video.
|
||||
chapters, sponsor_chapters = self._mark_chapters_to_remove(
|
||||
copy.deepcopy(info.get('chapters')) or [],
|
||||
|
|
Loading…
Reference in a new issue