自定义SafeOpen函数返回文件对象后，是否需手动关闭？能否改用with open？

好问题！咱们一步步来解答：

第一个问题：调用SafeOpen后需要手动close()吗？ #

是的，必须手动调用。你的SafeOpen函数只处理了文件打开阶段的IO异常，但并没有实现自动关闭文件的逻辑——它只是返回了一个普通的文件对象。如果不手动调用close()，文件句柄会一直被占用，直到Python的垃圾回收机制偶然触发，这在处理大量文件或者长期运行的程序里很容易导致资源泄漏。你例子里的写法是正确的，但确实有点繁琐。

第二个问题：能不能用with open的语法实现这个函数的功能？ #

当然可以！而且这才是更Pythonic、更安全的做法。with语句依赖上下文管理器（实现__enter__和__exit__方法的对象），我们可以把SafeOpen改造成一个上下文管理器，这样就能像原生open一样用with自动管理文件的打开和关闭了。

这里有两种常见的实现方式：

方式1：用contextlib.contextmanager装饰器（推荐，更简洁）

借助contextlib模块的装饰器，我们可以用生成器快速实现上下文管理器：

import sys
from contextlib import contextmanager

@contextmanager
def SafeOpen(filename, mode):
    infile = None
    try:
        infile = open(filename, mode)
        yield infile  # 将文件对象传递给with语句的as变量
    except IOError as error:
        print('Cannot open file due to error:', str(error))
        sys.exit(1)
    finally:
        # 不管with块内有没有异常，都会执行这里的关闭逻辑
        if infile is not None:
            infile.close()

调用的时候直接用with，完全不用手动close()：

with SafeOpen(filename, 'r') as infile:
    for line in infile:
        print(line)
# with块结束后，文件会自动关闭

方式2：用类实现上下文管理器

如果你想更底层地控制上下文逻辑，可以写一个类来实现__enter__和__exit__方法：

import sys

class SafeOpen:
    def __init__(self, filename, mode):
        self.filename = filename
        self.mode = mode
        self.infile = None

    def __enter__(self):
        # 进入with块时执行：打开文件
        try:
            self.infile = open(self.filename, self.mode)
            return self.infile
        except IOError as error:
            print('Cannot open file due to error:', str(error))
            sys.exit(1)

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 离开with块时执行：关闭文件
        if self.infile is not None:
            self.infile.close()

调用方式和原生open完全一致：

with SafeOpen(filename, 'r') as infile:
    for line in infile:
        print(line)

总结 #

• 你原来的SafeOpen函数必须手动close()文件对象；
• 改成上下文管理器后，用with语法可以自动处理文件关闭，既安全又省心，推荐用这种方式。

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