0
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-26 02:55:17 +00:00

[downloader/external] Simplify finished progress hook reporting and add elapsed time (closes #10876)

This commit is contained in:
Sergey M․ 2018-03-24 16:29:03 +07:00
parent 2ea212628e
commit f0298f653e
No known key found for this signature in database
GPG key ID: 2C393E0F18A9236D

View file

@ -1,9 +1,10 @@
from __future__ import unicode_literals
import os.path
import re
import subprocess
import sys
import re
import time
from .common import FileDownloader
from ..compat import (
@ -30,6 +31,7 @@ def real_download(self, filename, info_dict):
tmpfilename = self.temp_name(filename)
try:
started = time.time()
retval = self._call_downloader(tmpfilename, info_dict)
except KeyboardInterrupt:
if not info_dict.get('is_live'):
@ -41,21 +43,20 @@ def real_download(self, filename, info_dict):
self.to_screen('[%s] Interrupted by user' % self.get_basename())
if retval == 0:
if filename == '-':
self._hook_progress({
'filename': filename,
'status': 'finished',
})
else:
status = {
'filename': filename,
'status': 'finished',
'elapsed': time.time() - started,
}
if filename != '-':
fsize = os.path.getsize(encodeFilename(tmpfilename))
self.to_screen('\r[%s] Downloaded %s bytes' % (self.get_basename(), fsize))
self.try_rename(tmpfilename, filename)
self._hook_progress({
status.update({
'downloaded_bytes': fsize,
'total_bytes': fsize,
'filename': filename,
'status': 'finished',
})
self._hook_progress(status)
return True
else:
self.to_stderr('\n')