Qubes/core-admin-client
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

qvm-template: Remove downloaded file if the download is interrupted.

Browse Source
This commit is contained in:
WillyPillow 2020-07-29 19:31:25 +08:00
parent 88ee572cac
commit 8aa9ab9e89
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
qubesadmin/tools/qvm_template.py
View File

@ -724,12 +724,16 @@ def download(args, app, path_override=None,
done = True done = True
break break
except ConnectionError: except ConnectionError:
os.remove(target_suffix)
if attempt + 1 < args.retries: if attempt + 1 < args.retries:
print('\'%s\' download failed, retrying...' % spec, print('\'%s\' download failed, retrying...' % spec,
file=sys.stderr) file=sys.stderr)
except:
# Also remove file if interrupted by other means
os.remove(target_suffix)
raise
if not done: if not done:
print('\'%s\' download failed.' % spec, file=sys.stderr) print('\'%s\' download failed.' % spec, file=sys.stderr)
os.remove(target_suffix)
sys.exit(1) sys.exit(1)
def remove(args, app): def remove(args, app):