python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > 使用requests下载文件

python 如何使用requests下载文件

作者:阿宅gogo

这篇文章主要介绍了python 如何使用requests下载文件,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教

使用requests下载文件

1、获取token,或者session

如不需要可忽略

login_url = "http://xxxx/api/auth/login"
login_data = {"username":"test3","password":"123456"}
login_res = requests.post(url=login_url,data = login_data)
token = login_res.json()["data"]["token"]

2、获取下载路径

如果请求后直接返回文件内容,可直接进行第三步

batch_url = "http://xxxx/api/models/batch"
batch_data = {"ids":"[4]","version_number":"[309]"}
headers = {"Authorization":"bearer %s" % token}
batch_res = requests.get(url=batch_url,params=batch_data,headers=headers)

3、根据下载路径拼接下载url

完成文件下载以及写入

file_path = batch_res.json()['data']['file_path']
file_name = batch_res.json()['data']['file_name']
down_url = "http://xxxx/api/report/down"
down_data = {"type":2,
             "file_path":file_path,
             "file_name":file_name,
             "token":token
             }
down_res = requests.get(url=down_url,params=down_data)
with open(file_name,"wb") as code:
    code.write(down_res.content)

备注:

第二步返回json数据,包含路径、文件名,实际是文件生成过程,第三步下载在服务端生成的文件,有时第三步无法在页面F12查看到,需要使用抓包工具获取

用requests.get下载文件

不知道大家有没有遇到这样的问题

就是url源不稳定,时不时下载到空文件,我终于想到了一个解决的好办法,分享给大家。

def downloadfile(url,filename=None):
  if(not filename):							#如果参数没有指定文件名
    filename=os.path.basename(url)			#取用url的尾巴为文件名
  leng=1
  while(leng==1):
    torrent=requests.get(url,headers=headers)
    leng=len(list(torrent.iter_content(1024)))  #下载区块数
    if(leng==1):								#如果是1 就是空文件 重新下载
      print(filename,'下载失败,重新下载')
      sleep(1)
    else:
      print(path,'下载完成')
  with open(filename,'wb') as f:				
    for chunk in torrent.iter_content(1024):	#防止文件过大,以1024为单位一段段写入
      f.write(chunk)

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文