diff options
| author | MMM <flashdagger@googlemail.com> | 2022-06-07 14:44:08 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-06-07 05:44:08 -0700 |
| commit | 36195c4461701ffe4245b126a222b784c7abd1ea (patch) | |
| tree | f37305feda7c6e1078fe30e7b7fecbf3411b1db7 | |
| parent | 65141660aba62fefb1901804aeb0484992243af7 (diff) | |
| download | hypervideo-pre-36195c4461701ffe4245b126a222b784c7abd1ea.tar.lz hypervideo-pre-36195c4461701ffe4245b126a222b784c7abd1ea.tar.xz hypervideo-pre-36195c4461701ffe4245b126a222b784c7abd1ea.zip | |
[dash] Show fragment count with `--live-from-start` (#3493)
Authored by: flashdagger
| -rw-r--r-- | yt_dlp/downloader/dash.py | 1 | ||||
| -rw-r--r-- | yt_dlp/downloader/fragment.py | 4 | ||||
| -rw-r--r-- | yt_dlp/extractor/youtube.py | 1 |
3 files changed, 6 insertions, 0 deletions
diff --git a/yt_dlp/downloader/dash.py b/yt_dlp/downloader/dash.py index d0a0b2893..a6da26f09 100644 --- a/yt_dlp/downloader/dash.py +++ b/yt_dlp/downloader/dash.py @@ -73,6 +73,7 @@ class DashSegmentsFD(FragmentFD): yield { 'frag_index': frag_index, + 'fragment_count': fragment.get('fragment_count'), 'index': i, 'url': fragment_url, } diff --git a/yt_dlp/downloader/fragment.py b/yt_dlp/downloader/fragment.py index d94cb4956..79161b809 100644 --- a/yt_dlp/downloader/fragment.py +++ b/yt_dlp/downloader/fragment.py @@ -242,6 +242,9 @@ class FragmentFD(FileDownloader): if s['status'] not in ('downloading', 'finished'): return + if not total_frags and ctx.get('fragment_count'): + state['fragment_count'] = ctx['fragment_count'] + if ctx_id is not None and s.get('ctx_id') != ctx_id: return @@ -450,6 +453,7 @@ class FragmentFD(FileDownloader): fatal, count = is_fatal(fragment.get('index') or (frag_index - 1)), 0 while count <= fragment_retries: try: + ctx['fragment_count'] = fragment.get('fragment_count') if self._download_fragment(ctx, fragment['url'], info_dict, headers): break return diff --git a/yt_dlp/extractor/youtube.py b/yt_dlp/extractor/youtube.py index 113b9aa07..d44f16bc0 100644 --- a/yt_dlp/extractor/youtube.py +++ b/yt_dlp/extractor/youtube.py @@ -2407,6 +2407,7 @@ class YoutubeIE(YoutubeBaseInfoExtractor): last_segment_url = urljoin(fragment_base_url, 'sq/%d' % idx) yield { 'url': last_segment_url, + 'fragment_count': last_seq, } if known_idx == last_seq: no_fragment_score += 5 |