Python 3 文件下载(可直接调用，有进度条)

自己写的,貌似好用
借鉴了不少
copyfileobj()函数来自shutil库,为了进度条,去除了引用,把它放在了自己的代码里做了些更改~~(少引一个库不香吗)~~

下载函数(实现主要功能):download_file()

依赖:

python 3.10.0

requests 2.28.1

参数:

url: 被下载文件的url
path: 保存路径 需要文件名和扩展名(e.g. “/home/user/test.txt” or “./png/test.png”)
redown: 是否询问重新下载已存在文件(0:询问 1:不重新下载 2:重新下载)
return: 成功:0 失败:1

效果图:

应用举例:

将所有代码保存在downloader.py中
在其他文件中使用

from downloader import download_file

引用下载函数,
然后使用download_file()下载url

代码

import requests
import os_WINDOWS = os.name == 'nt'
COPY_BUFSIZE = 1024 * 1024 if _WINDOWS else 64 * 1024header = {'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, ''like Gecko) Chrome/103.0.5060.134 Safari/537.36 Edg/103.0.1264.71'}def copyfileobj(fsrc, fdst, size, length=0):"""copy data from file-like object fsrc to file-like object fdst:param fsrc: 被拷贝文件:param fdst: 目标文件:param size: 文件大小(B):param length: 分块大小:return:"""# Localize variable access to minimize overhead.if not length:length = COPY_BUFSIZEfsrc_read = fsrc.readfdst_write = fdst.writei = 0print('0.00% |', end='')while True:buf = fsrc_read(length)if not buf:breakfdst_write(buf)i += lengthsp = i / sizeif sp > 1:sp = 1print(f'\r', format(sp * 100, '.2f'), '% |', '▊' * int(40 * sp), sep='', end='')print('\n')def download_file(url, path, redown=0):"""文件下载:param url: 下载文件url:param path: 保存路径 需要文件名和扩展名:param redown: 是否询问重新下载(0:询问 1:默认不重新下载 2:默认重新下载):return: 成功:0 失败:1"""print(url.split('/')[-1], '=>', path.split('/')[-1])if os.path.exists(path):if redown == 0:if input('文件已存在,是否重新下载? (y/n)') != 'y':return 0elif redown == 1:return 0os.remove(path)with requests.get(url, stream=True, headers=header) as r:with open(path + '.tmp', 'wb') as f:copyfileobj(r.raw, f, int(r.headers['Content-Length']))try:os.rename(path + '.tmp', path)except FileExistsError:return 1return 0if __name__ == '__main__':print(download_file('https://mirrors.tuna.tsinghua.edu.cn/debian-cd/11.4.0/amd64/iso-cd/''debian-11.4.0-amd64-netinst.iso', 'test.iso'))

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！