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

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