如何在Python中的“ with”语句下调用整个函数？

Question

我有以下示例函数：

def Run(self):
    with self._resource() as r:
        # a lot of code uses |r|
        pass
    # end of 'with' statement
# end of function body

我不想失去，因为额外的缩进内整体功能体的视觉空间with声明。

我也不想在类范围之外调用_resource() -它以某些方式破坏了封装，即，这不是一个好方法：

with obj._resource() as r:
    obj.Run(r)

有什么漂亮的方法可以运行相同的代码而不损失可视空间？

Answer 1

如果只需要处理此功能，则非常简单：

class Foo(object):
    def Run(self):
        with self._resource() as r:
            return self._RunWithResource(r)

    def _RunWithResource(self, r):
        # ...

如果要重复图案，装饰器可能会有所帮助。 或多或少：

from functools import wraps
def with_resource(f):
    @wraps
    def wrapper(self, *a, **kw):
        with self._resource() as r:
            return f(self, r, *a, **kw)
    return wrapper

class Foo(object):
    @with_resource
    def Run(self, r):
        # ...