ChatGPT生成文件下载全指南：从API调用到本地存储实战

1次阅读

没有评论

共计 2500 个字符，预计需要花费 7 分钟才能阅读完成。

在日常开发中，我们经常使用 ChatGPT 生成各种文件，比如自动生成的报告、数据分析结果、甚至是代码片段。但很多新手开发者会遇到这样的问题：

直接从网页复制粘贴，格式容易混乱
生成的文件链接有时会过期
大文件下载过程中网络中断导致前功尽弃
不知道如何批量处理多个生成文件

这是最原始的方式，适合小段文本内容，但存在明显缺陷：

格式容易丢失
无法处理结构化数据（如 CSV、JSON）
对大文件不友好

通过 API 可以获取原始文件数据，优势包括：

保持原始格式
支持各种文件类型
可以程序化处理
适合批量操作

首先确保已安装 Python 3.8+ 和 openai 库：

pip install openai

import openai
import os

# 1. 身份认证设置
openai.api_key = '你的 API 密钥'  # 替换为实际 API 密钥

# 2. 文件下载函数
def download_chatgpt_file(file_id, save_path):
    try:
        # 获取文件元数据
        file_info = openai.File.retrieve(file_id)

        # 3. 下载文件内容
        file_content = openai.File.download(file_id)

        # 4. 本地存储
        with open(save_path, 'wb') as f:
            f.write(file_content)

        print(f"文件已保存到: {save_path}")
        return True
    except Exception as e:
        print(f"下载失败: {str(e)}")
        return False

# 使用示例
file_id = 'file-abc123'  # 替换为实际文件 ID
save_path = './downloads/report.txt'  # 保存路径
os.makedirs(os.path.dirname(save_path), exist_ok=True)  # 创建目录

download_chatgpt_file(file_id, save_path)

def download_large_file(file_id, save_path, chunk_size=1024*1024):
    try:
        # 获取文件大小
        file_info = openai.File.retrieve(file_id)
        file_size = file_info['bytes']

        # 分块下载
        with open(save_path, 'wb') as f:
            for offset in range(0, file_size, chunk_size):
                # 设置下载范围
                headers = {'Range': f'bytes={offset}-{min(offset+chunk_size-1, file_size-1)}'
                }

                # 重试机制
                for attempt in range(3):
                    try:
                        file_content = openai.File.download(file_id, headers=headers)
                        f.write(file_content)
                        break
                    except Exception as e:
                        if attempt == 2:
                            raise
                        print(f"下载中断，重试 {attempt+1}/3")

        print(f"大文件下载完成: {save_path}")
        return True
    except Exception as e:
        print(f"大文件下载失败: {str(e)}")
        return False

import hashlib

def verify_file_integrity(file_path, expected_md5):
    with open(file_path, 'rb') as f:
        file_hash = hashlib.md5(f.read()).hexdigest()

    if file_hash == expected_md5:
        print("文件完整性验证通过")
        return True
    else:
        print(f"文件损坏！预期 MD5: {expected_md5}, 实际 MD5: {file_hash}")
        return False

OpenAI API 有调用频率限制，建议：

在代码中添加延时
捕获速率限制异常
考虑使用指数退避策略

根据文件扩展名自动处理不同格式：

def get_file_extension(content_type):
    mapping = {
        'text/plain': '.txt',
        'text/csv': '.csv',
        'application/json': '.json'
        # 添加更多 MIME 类型映射
    }
    return mapping.get(content_type, '.bin')

建议实现自动重试逻辑：

from time import sleep

def safe_download(file_id, save_path, max_retries=3):
    for attempt in range(max_retries):
        try:
            return download_chatgpt_file(file_id, save_path)
        except Exception as e:
            if attempt == max_retries - 1:
                raise
            sleep(2 ** attempt)  # 指数退避