blackteay/DrissionPage
1
0
Fork 0
mirror of https://gitee.com/g1879/DrissionPage.git synced 2024-12-10 04:00:23 +08:00
Code Issues Projects Releases Wiki Activity

微调download的show_msg参数

Browse Source
This commit is contained in:
g1879 2020-06-03 10:26:53 +08:00
parent 3edac9f427
commit 800a4239ef
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
DrissionPage/session_page.py
View File

@ -149,6 +149,7 @@ class SessionPage(object):
if show_msg: if show_msg:
print_txt = file_full_name if file_name == file_full_name else f'{file_name} -> {file_full_name}' print_txt = file_full_name if file_name == file_full_name else f'{file_name} -> {file_full_name}'
print(print_txt) print(print_txt)
print(f'Downloading to: {goal_path}')
# -------------------开始下载------------------- # -------------------开始下载-------------------
# 获取远程文件大小 # 获取远程文件大小
file_size = int(r.headers['Content-Length']) if 'Content-Length' in r.headers else None file_size = int(r.headers['Content-Length']) if 'Content-Length' in r.headers else None
@ -158,12 +159,11 @@ class SessionPage(object):
full_path = Path(f'{goal_path}\\{file_full_name}') full_path = Path(f'{goal_path}\\{file_full_name}')
try: try:
with open(str(full_path), 'wb') as tmpFile: with open(str(full_path), 'wb') as tmpFile:
print(f'Downloading to: {goal_path}')
for chunk in r.iter_content(chunk_size=1024): for chunk in r.iter_content(chunk_size=1024):
if chunk: if chunk:
tmpFile.write(chunk) tmpFile.write(chunk)
# 如表头有返回文件大小，显示进度 # 如表头有返回文件大小，显示进度
if file_size: if show_msg and file_size:
downloaded_size += 1024 downloaded_size += 1024
rate = downloaded_size / file_size if downloaded_size < file_size else 1 rate = downloaded_size / file_size if downloaded_size < file_size else 1
print('\r {:.0%} '.format(rate), end="") print('\r {:.0%} '.format(rate), end="")