如何以Python风格且高效的方式在代码多处写入同一文件？

当需要在代码不同位置多次写入同一文件时，既要避免每次调用都打开/关闭文件的性能损耗，又要保证文件资源安全释放，推荐以下几种符合Python风格的方案：

方案1：自定义上下文管理器类（推荐） #

将文件句柄的管理逻辑封装到上下文管理器中，利用Python的with语句自动处理文件的打开与关闭，同时在整个上下文范围内保持文件打开状态，支持多次写入。

class FileWriter:
    def __init__(self, file_path, mode="a"):
        self.file_path = file_path
        self.mode = mode
        self._file = None

    def __enter__(self):
        # 进入上下文时打开文件
        self._file = open(self.file_path, self.mode)
        return self

    def write(self, content):
        # 封装写入方法，支持任意位置调用
        if self._file:
            self._file.write(f"{content}\n")

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 退出上下文时自动关闭文件，即使发生异常也会执行
        if self._file:
            self._file.close()

# 使用示例
if __name__ == "__main__":
    with FileWriter("myfile.txt", "a") as writer:
        writer.write("第一行写入内容")
        # 中间可以插入任意业务逻辑
        writer.write("第二行写入内容")
        writer.write("第三行写入内容")

优势 #

• 仅在进入上下文时打开一次文件，多次写入无重复IO开销
• 无需手动调用close()，__exit__方法会自动处理资源释放，异常场景下也能保证文件关闭
• 代码结构清晰，符合Python的上下文管理规范

方案2：面向对象场景的上下文管理器集成 #

如果你的代码本身是基于类的（比如原问题中的self场景），可以直接让业务类实现上下文管理器协议：

import datetime

class BusinessService:
    def __init__(self, log_file_path):
        self.log_file_path = log_file_path
        self._log_file = None

    def __enter__(self):
        self._log_file = open(self.log_file_path, "a", encoding="utf-8")
        return self

    def log_operation(self, operation_detail):
        # 业务方法中直接写入文件
        self._log_file.write(f"[{datetime.datetime.now()}] {operation_detail}\n")

    def do_business_logic1(self):
        # 业务逻辑1中调用写入
        self.log_operation("执行业务逻辑1")

    def do_business_logic2(self):
        # 业务逻辑2中调用写入
        self.log_operation("执行业务逻辑2")

    def __exit__(self, exc_type, exc_val, exc_tb):
        self._log_file.close()

# 使用示例
with BusinessService("operation.log") as service:
    service.do_business_logic1()
    service.do_business_logic2()

方案3：模块级资源管理（简单场景适用） #

如果是小型单脚本场景，可以在模块级别初始化文件句柄，配合try/finally保证关闭，但这种方式灵活性较低，需注意全局变量的使用：

import datetime

_log_file = None

def init_logger(file_path):
    global _log_file
    _log_file = open(file_path, "a", encoding="utf-8")

def log(message):
    if _log_file:
        _log_file.write(f"[{datetime.datetime.now()}] {message}\n")

def close_logger():
    global _log_file
    if _log_file:
        _log_file.close()

# 使用示例
if __name__ == "__main__":
    try:
        init_logger("simple.log")
        log("程序启动")
        # 其他代码逻辑
        log("程序执行中")
        log("程序结束")
    finally:
        close_logger()

注意事项 #

• 方案3需严格保证在finally中调用关闭方法，否则程序异常退出时可能导致文件未关闭、数据丢失
• 不建议在复杂多线程/多进程场景中使用全局文件句柄，需额外处理同步问题

内容的提问来源于stack exchange，提问作者Fontanka16