欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

python 如何使用requests下载文件

程序员文章站 2022-06-09 22:34:49
目录使用requests下载文件1、获取token,或者session2、获取下载路径3、根据下载路径拼接下载url用requests.get下载文件不知道大家有没有遇到这样的问题使用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)

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