diff options
author | Nayil Mukhametshin <66028747+nlscc@users.noreply.github.com> | 2020-07-30 14:38:56 +0200 |
---|---|---|
committer | Nayil Mukhametshin <66028747+nlscc@users.noreply.github.com> | 2020-07-30 14:38:56 +0200 |
commit | 8899c8d5821ac06f0dd09fbb2a7a0577fa9d5eb2 (patch) | |
tree | 9addb5a837ec69f24d0e28ec43fca48709a89c98 /samloader | |
parent | fix #3 (broken checkupdate command) (diff) | |
download | samloader-8899c8d5821ac06f0dd09fbb2a7a0577fa9d5eb2.tar samloader-8899c8d5821ac06f0dd09fbb2a7a0577fa9d5eb2.tar.gz samloader-8899c8d5821ac06f0dd09fbb2a7a0577fa9d5eb2.tar.bz2 samloader-8899c8d5821ac06f0dd09fbb2a7a0577fa9d5eb2.tar.lz samloader-8899c8d5821ac06f0dd09fbb2a7a0577fa9d5eb2.tar.xz samloader-8899c8d5821ac06f0dd09fbb2a7a0577fa9d5eb2.tar.zst samloader-8899c8d5821ac06f0dd09fbb2a7a0577fa9d5eb2.zip |
Diffstat (limited to 'samloader')
-rw-r--r-- | samloader/main.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/samloader/main.py b/samloader/main.py index e8fca4f..a4896bb 100644 --- a/samloader/main.py +++ b/samloader/main.py @@ -38,15 +38,17 @@ def checkupdate(model, region): @click.argument("version") @click.argument("model") @click.argument("region") -@click.argument("outfile") -def download(version, model, region, outfile): +@click.argument("out") +def download(version, model, region, out): client = fusclient.FUSClient() path, filename = getbinaryfile(client, version, region, model) print("Downloading file {} ...".format(path+filename)) initdownload(client, filename) r = client.downloadfile(path+filename) length = int(r.headers["Content-Length"]) - with open(outfile, "wb") as f: + if os.path.isdir(out): + out = os.path.join(out, filename) + with open(out, "wb") as f: for chunk in progress.bar(r.iter_content(chunk_size=0x10000), expected_size=(length/0x10000)+1): if chunk: f.write(chunk) |