对象清理似乎是我在编程期间遇到的一个非常常见的问题。迄今为止,我一直使用此处with
推荐的声明
今天我有另一个想法,这对我来说似乎更优雅(因为它不需要最终用户的 with 声明)。这个想法是对某种类型的对象使用 try-finally 装饰器(具有清理方法)。
只是想知道这种做法是否有任何问题,或者是否有更好的方法。我不喜欢我的许多类需要使用 with 语句进行初始化,但我也想确保我的对象正确关闭。这是一个简短的例子。
def cleanme(func):
def _decorator(self, *args, **kwargs):
try:
func(self, *args, **kwargs)
finally:
self._cleanup()
return _decorator
class IObject(object):
def __init__(self):
self.file_name = "some_file.txt"
self._file_object = None
self._cleaned = True
@cleanme
def run(self):
self._connect()
while True:
# do some things over a long period
pass
def _connect(self):
self._file_object = open(self.file_name)
self._cleaned = False
def _cleanup(self):
if not self._cleaned:
self._file_object.close()
self._cleaned = True