Python中优雅处理可选文件打开与关闭的方法？

方案1：使用contextlib.nullcontext（Python 3.7+推荐） #

利用Python标准库的nullcontext，可以在无需打开文件时提供一个空上下文，统一用with语句管理资源，既避免重复判断，又自动处理文件关闭：

from contextlib import nullcontext

# 根据路径是否存在选择上下文：有路径则打开文件，否则返回None
with open(args.optional_input_file_path) if args.optional_input_file_path is not None else nullcontext(None) as fh:
    my_function(foo, bar, fh)

这段代码中，with语句会自动处理文件的关闭操作；当路径为None时，nullcontext(None)会让fh直接取None，完全符合你传入IO对象或None的需求，且仅调用一次my_function。

方案2：兼容Python 3.6及以下版本 #

如果你的Python版本较低，没有nullcontext，可以用try-finally封装资源清理逻辑，同时用函数简化文件句柄的初始化：

def get_file_handle(file_path):
    return open(file_path) if file_path is not None else None

fh = get_file_handle(args.optional_input_file_path)
try:
    my_function(foo, bar, fh)
finally:
    if fh is not None:
        fh.close()

这里用函数get_file_handle统一处理文件句柄的创建，避免重复写if判断；finally块确保文件句柄在使用后被正确关闭，同样只需要调用一次my_function。

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