From 3a0a65377a22294cd1d2b068784d024f154688eb Mon Sep 17 00:00:00 2001 From: Chunxiao Li Date: Wed, 25 Nov 2020 17:39:53 +0800 Subject: [PATCH] Update try_download.py --- pyatmos/utils/try_download.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/pyatmos/utils/try_download.py b/pyatmos/utils/try_download.py index 7d436bc..a20569b 100644 --- a/pyatmos/utils/try_download.py +++ b/pyatmos/utils/try_download.py @@ -6,7 +6,7 @@ def tqdm_request(url,dir_to,file,desc): ''' Try to download files from a remote server by request. ''' - block_size = 1024 + block_size = 1024*10 bar_format = "{l_bar}%s{bar}%s{r_bar}" % (Fore.BLUE, Fore.RESET) for idownload in range(5): try: @@ -19,14 +19,14 @@ def tqdm_request(url,dir_to,file,desc): for chunk in res.iter_content(block_size): pbar.update(len(chunk)) local_file.write(chunk) + pbar.close() + res.close() break except: sleep(2) if idownload == 4: remove(dir_to + file) print('No response, skip this file.') - finally: - pbar.close() - local_file.close() - res.close() + finally: + local_file.close()