如何在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):
        # ...