Sirius/youtube-dl
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

[downloader/fragment] use the documented names for fragment progress_hooks fields

Browse source
This commit is contained in:
Remita Amine 2017-04-22 16:42:24 +01:00
parent fbf56be213
commit 3e0304fe6e
5 changed files with 14 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

18
youtube_dl/downloader/fragment.py
View file

@ -66,7 +66,9 @@ class FragmentFD(FileDownloader):
if not (ctx.get('live') or ctx['tmpfilename'] == '-'):
frag_index_stream, _ = sanitize_open(self.ytdl_filename(ctx['filename']), 'w')
frag_index_stream.write(json.dumps({
'frag_index': ctx['frag_index']
'download': {
'last_fragment_index': ctx['fragment_index']
},
}))
frag_index_stream.close()
@ -100,7 +102,7 @@ class FragmentFD(FileDownloader):
ytdl_filename = encodeFilename(self.ytdl_filename(ctx['filename']))
if os.path.isfile(ytdl_filename):
frag_index_stream, _ = sanitize_open(ytdl_filename, 'r')
frag_index = json.loads(frag_index_stream.read())['frag_index']
frag_index = json.loads(frag_index_stream.read())['download']['last_fragment_index']
frag_index_stream.close()
dest_stream, tmpfilename = sanitize_open(tmpfilename, open_mode)
@ -108,7 +110,7 @@ class FragmentFD(FileDownloader):
'dl': dl,
'dest_stream': dest_stream,
'tmpfilename': tmpfilename,
'frag_index': frag_index,
'fragment_index': frag_index,
# Total complete fragments downloaded so far in bytes
'complete_frags_downloaded_bytes': resume_len,
})
@ -120,8 +122,8 @@ class FragmentFD(FileDownloader):
state = {
'status': 'downloading',
'downloaded_bytes': ctx['complete_frags_downloaded_bytes'],
'frag_index': ctx['frag_index'],
'frag_count': total_frags,
'fragment_index': ctx['fragment_index'],
'fragment_count': total_frags,
'filename': ctx['filename'],
'tmpfilename': ctx['tmpfilename'],
}
@ -144,12 +146,12 @@ class FragmentFD(FileDownloader):
if not ctx['live']:
estimated_size = (
(ctx['complete_frags_downloaded_bytes'] + frag_total_bytes) /
(state['frag_index'] + 1) * total_frags)
(state['fragment_index'] + 1) * total_frags)
state['total_bytes_estimate'] = estimated_size
if s['status'] == 'finished':
state['frag_index'] += 1
ctx['frag_index'] = state['frag_index']
state['fragment_index'] += 1
ctx['fragment_index'] = state['fragment_index']
state['downloaded_bytes'] += frag_total_bytes - ctx['prev_frag_downloaded_bytes']
ctx['complete_frags_downloaded_bytes'] = state['downloaded_bytes']
ctx['prev_frag_downloaded_bytes'] = 0