Python文件存储教程：如何将文件保存到目录

创建目录（如果不存在）

在保存文件之前，通常需要确保目标目录存在。Python的os模块提供了创建目录的方法：

import os

# 定义要保存文件的目录
save_dir = "documents/reports"

# 如果目录不存在，则创建它（包括所有中间目录）
os.makedirs(save_dir, exist_ok=True)

print(f"目录 '{save_dir}' 已创建或已存在")

保存文本文件

使用Python内置的open()函数可以轻松保存文本文件：

def save_text_file(directory, filename, content):
    # 创建完整文件路径
    file_path = os.path.join(directory, filename)
    
    # 使用'w'模式写入文本（会覆盖现有文件）
    with open(file_path, 'w', encoding='utf-8') as file:
        file.write(content)
    
    print(f"文本文件已保存至: {file_path}")
    return file_path

# 示例用法
report_content = """销售报告
日期: 2023-11-15
产品: Python教程
销量: 1200
收入: $24,000
"""
save_text_file("documents/reports", "sales_report.txt", report_content)

保存二进制文件

对于图片、PDF等二进制文件，需要使用二进制写入模式('wb')：

def save_binary_file(directory, filename, data):
    file_path = os.path.join(directory, filename)
    
    # 使用'wb'模式写入二进制数据
    with open(file_path, 'wb') as file:
        file.write(data)
    
    print(f"二进制文件已保存至: {file_path}")
    return file_path

# 示例用法：保存图片
# 通常从网络或其它来源获取二进制数据
image_data = b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x01\x00...'  # 简化的图片数据
save_binary_file("documents/images", "chart.png", image_data)

错误处理与最佳实践

文件操作时可能会遇到各种异常，使用try-except块处理错误是良好的编程实践：

def safe_save_file(directory, filename, content, is_binary=False):
    try:
        os.makedirs(directory, exist_ok=True)
        file_path = os.path.join(directory, filename)
        
        mode = 'wb' if is_binary else 'w'
        encoding = None if is_binary else 'utf-8'
        
        with open(file_path, mode, encoding=encoding) as file:
            file.write(content)
            
        print(f"文件保存成功: {file_path}")
        return True
        
    except OSError as e:
        print(f"保存文件时出错: {e}")
        return False
    except Exception as e:
        print(f"发生未知错误: {e}")
        return False

# 安全地保存文件
safe_save_file("documents/reports", "quarterly.txt", "第四季度报告...")
safe_save_file("documents/images", "graph.png", image_data, is_binary=True)

最佳实践建议：

始终使用os.path.join()构建文件路径，确保跨平台兼容性
使用with语句自动处理文件打开和关闭
为文本文件指定编码（推荐UTF-8）
检查并处理可能出现的异常
验证用户输入的文件路径，防止路径遍历攻击
处理大文件时使用分块读写

实际应用示例

下面是一个完整示例，展示如何从网络下载图片并保存到本地目录：

import os
import requests

def download_and_save_image(url, save_dir, filename):
    try:
        # 创建目录（如果不存在）
        os.makedirs(save_dir, exist_ok=True)
        
        # 下载文件内容
        response = requests.get(url)
        response.raise_for_status()  # 检查请求是否成功
        
        # 构建保存路径
        file_path = os.path.join(save_dir, filename)
        
        # 保存二进制文件
        with open(file_path, 'wb') as file:
            file.write(response.content)
            
        print(f"图片已保存至: {file_path}")
        return file_path
        
    except Exception as e:
        print(f"下载或保存图片失败: {e}")
        return None

# 使用示例
image_url = "https://example.com/sample-image.jpg"
download_and_save_image(image_url, "downloads/images", "sample.jpg")