python - 如何在 python 中将实例变量设置为 fileHandle？

Question

我想定义一个将 fileHandle 附加到类的类。例如，我可以定义一个带有实例变量的类作为文件处理程序对象，但当该对象最终被取消引用时，它是不安全的。

class CustomLoggingClass(object):
    def __init__(self, *args, **kwargs):
        self.fileHandle = open("logFile.json", "w+")

如何以安全的方式实现此操作，以便正确关闭 logFile.json？

Answer 1

最简单的方法可能是使用contextlib.contextmanager装饰器创建一个辅助上下文管理器函数，然后将其与with语句结合使用。例如:

class CustomLoggingClass(object):
    def __init__(self, *args, **kwargs):
        self.fileHandle = open("logFile.json", "w+")

    def close(self):
        self.fileHandle.close()

import contextlib

@contextlib.contextmanager
def cm_logger():
    logger = CustomLoggingClass()
    yield logger
    logger.close()

if __name__ == '__main__':
    with cm_logger as logger:
        # do stuff with logger, a CustomLoggingClass instance
        pass

with 语句和上下文管理器在 PEP 343 中讨论。 .