2Shirt/WizardKit
2Shirt/WizardKit
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases 26 Packages Wiki Activity Actions

Avoid crash when downloading tools

Browse source
This commit is contained in:
2Shirt 2021-04-30 02:08:50 -06:00
parent 447e93ca3a
commit e488b2c89c
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
scripts/wk/kit/tools.py
View file

@ -27,6 +27,8 @@ CACHED_DIRS = {}
def download_file(out_path, source_url, as_new=False, overwrite=False): def download_file(out_path, source_url, as_new=False, overwrite=False):
"""Download a file using requests, returns pathlib.Path.""" """Download a file using requests, returns pathlib.Path."""
out_path = pathlib.Path(out_path).resolve() out_path = pathlib.Path(out_path).resolve()
if as_new:
out_path = out_path.with_suffix(f'{out_path.suffix}.new')
cursor_left = '\u001B[14D' cursor_left = '\u001B[14D'
print(f'Downloading...{cursor_left}', end='', flush=True) print(f'Downloading...{cursor_left}', end='', flush=True)
@ -36,8 +38,6 @@ def download_file(out_path, source_url, as_new=False, overwrite=False):
# Create destination directory # Create destination directory
out_path.parent.mkdir(parents=True, exist_ok=True) out_path.parent.mkdir(parents=True, exist_ok=True)
if as_new:
out_path = out_path.with_suffix(f'{out_path.suffix}.new')
# Request download # Request download
response = requests.get(source_url, stream=True) response = requests.get(source_url, stream=True)