具有原始文件名的Python下载文件

import requests
import os

def download_file(url):
    response = requests.get(url)
    if response.status_code == 200:
        # 从响应头中获取文件的原始名称
        filename = get_filename_from_headers(response.headers)
        if filename:
            # 保存文件到本地，并使用原始名称命名
            with open(filename, 'wb') as file:
                file.write(response.content)
            print("文件下载成功！")
        else:
            print("无法获取文件名！")
    else:
        print("文件下载失败！")

def get_filename_from_headers(headers):
    # 从Content-Disposition响应头中获取文件名
    content_disposition = headers.get('Content-Disposition')
    if content_disposition:
        filename = content_disposition.split('filename=')[1].strip('"')
        return filename
    return None

# 示例使用腾讯云对象存储COS作为文件下载的推荐产品
download_file('https://example.com/file.pdf')