mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-08-24 16:21:40 +02:00
[downloader/fragment] Allow persisting extra state when a download is interrupted
This commit is contained in:
parent
5873d4ccdd
commit
4d49884c58
@ -77,7 +77,10 @@ def _read_ytdl_file(self, ctx):
|
|||||||
assert 'ytdl_corrupt' not in ctx
|
assert 'ytdl_corrupt' not in ctx
|
||||||
stream, _ = sanitize_open(self.ytdl_filename(ctx['filename']), 'r')
|
stream, _ = sanitize_open(self.ytdl_filename(ctx['filename']), 'r')
|
||||||
try:
|
try:
|
||||||
ctx['fragment_index'] = json.loads(stream.read())['downloader']['current_fragment']['index']
|
ytdl_data = json.loads(stream.read())
|
||||||
|
ctx['fragment_index'] = ytdl_data['downloader']['current_fragment']['index']
|
||||||
|
if 'extra_state' in ytdl_data['downloader']:
|
||||||
|
ctx['extra_state'] = ytdl_data['downloader']['extra_state']
|
||||||
except Exception:
|
except Exception:
|
||||||
ctx['ytdl_corrupt'] = True
|
ctx['ytdl_corrupt'] = True
|
||||||
finally:
|
finally:
|
||||||
@ -90,6 +93,8 @@ def _write_ytdl_file(self, ctx):
|
|||||||
'index': ctx['fragment_index'],
|
'index': ctx['fragment_index'],
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
if 'extra_state' in ctx:
|
||||||
|
downloader['extra_state'] = ctx['extra_state']
|
||||||
if ctx.get('fragment_count') is not None:
|
if ctx.get('fragment_count') is not None:
|
||||||
downloader['fragment_count'] = ctx['fragment_count']
|
downloader['fragment_count'] = ctx['fragment_count']
|
||||||
frag_index_stream.write(json.dumps({'downloader': downloader}))
|
frag_index_stream.write(json.dumps({'downloader': downloader}))
|
||||||
|
Loading…
Reference in New Issue
Block a user