hongshaorou/ATMOS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update try_download.py

Browse Source
This commit is contained in:
Chunxiao Li 2020-11-25 17:39:53 +08:00
parent a1a723020b
commit 3a0a65377a
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pyatmos/utils/try_download.py
View File

@ -6,7 +6,7 @@ def tqdm_request(url,dir_to,file,desc):
''' '''
Try to download files from a remote server by request. 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) bar_format = "{l_bar}%s{bar}%s{r_bar}" % (Fore.BLUE, Fore.RESET)
for idownload in range(5): for idownload in range(5):
try: try:
@ -19,6 +19,8 @@ def tqdm_request(url,dir_to,file,desc):
for chunk in res.iter_content(block_size): for chunk in res.iter_content(block_size):
pbar.update(len(chunk)) pbar.update(len(chunk))
local_file.write(chunk) local_file.write(chunk)
pbar.close()
res.close()
break break
except: except:
sleep(2) sleep(2)
@ -26,7 +28,5 @@ def tqdm_request(url,dir_to,file,desc):
remove(dir_to + file) remove(dir_to + file)
print('No response, skip this file.') print('No response, skip this file.')
finally: finally:
pbar.close()
local_file.close() local_file.close()
res.close()