python download不带文件名的文件/文件流

最新推荐文章于 2024-06-13 10:01:50 发布

allen sue

最新推荐文章于 2024-06-13 10:01:50 发布

阅读量1k

点赞数

分类专栏： # python

本文链接：https://blog.csdn.net/fish2009122/article/details/104747545

版权

python 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

需求：需要下载文件/文件流，文件为zip的压缩包，url不现实任何文件格式

下载文件

    def download_file():
        filepath = './data/'
        file_name = '{}{}.zip'.format(filepath, self.day)

        url = 'https://xx.com/download.do?custId=1103383695&taskId=10002779'
        header = {
            'accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3',
            'accept-encoding': 'gzip, deflate, br',
            'accept-language': 'zh-CN,zh;q=0.9',
            'cookie': COOKICE,
            'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/76.0.3809.87 Safari/537.36'
        }

        req = requests.get(url=url, headers=header)
        with open(file_name, 'wb') as f:
            f.write(req.content)

		# 解压缩
        import zipfile
        with zipfile.ZipFile(file_name) as zf:
            zf.extractall(path=filepath)

        import os
        os.remove(file_name)

下载文件流

    def download_file(url, method='get'):
        filepath = './data/'
        file_name = '{}{}.zip'.format(filepath, self.day)

        paras = {
    'access_token': token,
    'selectDate': yestoday
}

        if method == 'get':
            req = requests.get(url=url, params=paras, stream=True)
        else:
            req = requests.post(url=url, json=paras, stream=True)
  

        with open(file_name, 'wb') as f:
            for chunk in req.iter_content(chunk_size=1024):
                if chunk:
                    f.write(chunk)

		# 解压缩
        import zipfile
        try:
	        with zipfile.ZipFile(file_name) as zf:
    	        zf.extractall(path=filepath)
		
	        import os
	        os.remove(file_name)
	        
		except:
			print('the file is not a correct zip file')