一、概述
使用python3写了一个获取某网站文件的小脚本，使用了requests模块的get方法得到内容，然后通过文件读写的方式保存到硬盘 同时需要实现下载进度的显示
二、代码实现
安装模块
代码语言：javascript复制pip3 install requests tqdmtest.py 完整代码如下：
代码语言：javascript复制#!/usr/bin/env python3
# coding: utf-8
import os
from urllib.request import urlopen
import requests
from tqdm import tqdm
def download_from_url(url, dst):
"""
@param: url to download file
@param: dst place to put the file
:return: bool
"""
# 获取文件长度
try:
file_size = int(urlopen(url).info().get('Content-Length', -1))
except Exception as e:
print(e)
print("错误，访问url: %s 异常" % url)
return False
# 文件大小
if os.path.exists(dst):
first_byte = os.path.getsize(dst)
else:
first_byte = 0
if first_byte >= file_size:
return file_size
header = {"Range": "bytes=%s-%s" % (first_byte, file_size)}
pbar = tqdm(
total=file_size, initial=first_byte,
unit='B', unit_scale=True, desc=url.split('/')[-1])
# 访问url进行下载
req = requests.get(url, headers=header, stream=True)
try:
with(open(dst, 'ab')) as f:
for chunk in req.iter_content(chunk_size=1024):
if chunk:
f.write(chunk)
pbar.update(1024)
except Exception as e:
print(e)
return False
pbar.close()
return True
if __name__ == '__main__':
url = "https://dl.360safe.com/360/inst.exe"
download_from_url(url, "inst.exe")效果如下：
本文参考链接：
https://www.zhihu.com/question/41132103/answer/279584095
