如何在装饰器中使用上下文管理器以及如何将在装饰器中创建的对象传递给装饰函数

Question

我有一个测试 class 需要在最后做一些清理工作。为了确保用户不会忘记这样做，我想向 class 添加一个上下文管理器。我还有一个装饰器，我想在装饰器内部使用这个上下文管理器创建一个 test class 的对象并将其传递给装饰函数。有可能吗？

这就是我想要做的：

class test:
    def __init__(self, name):
        self._name = name
        print "my name is {0}".format(name)

    def exit():
        print "exiting"

    @contextmanager
    def testcm(self):
        print "inside cm"
        try:
            yield self
        finally:
            self.exit()

    def randomtest(self, str):
        print "Inside random test {0}".format(str)


def decorate(name):
    def wrapper(testf):
        def testf_wrapper(test):
            with test(name).testcm() as testobj:
                return testf(testobj)
            return testf_wrapper
        return wrapper
    return decorate

@decorate("whatever")
def testf(testobj):
    testobj.randomtest("randomness")

函数 testf 接受测试 class 对象 - testobj 并用它做一些事情。之后，由于上下文管理器，testcm 确保调用清理函数。

所以有两个问题：

如何在装饰器中使用上下文管理器，据我所知，装饰器必须 return 一个函数，但是如果我 return 该函数（如代码中所示上面），上下文管理器将如何调用清理？
如何将装饰器中创建的对象传递给装饰函数，如果像上面代码那样传递，装饰函数将如何调用？

Answer 1

您的示例程序中有几个错误，我迷失在所有 test/testf/testobj 冗余中。请允许我直接回答您的问题。

How do I use a context manager inside a decorator?

就像您在其他任何地方使用上下文管理器一样。考虑这个程序，它使用装饰器在调用函数时透明地将 str 转换为 file：

def opener(func):
    def wrapper(name):
        with open(name) as input_file:
            func(input_file)
    return wrapper

@opener
def first_line(fd):
    print fd.readline()

first_line('/etc/passwd')

如您所见，装饰函数在装饰函数的调用周围使用上下文管理器。

How do I pass an object created in decorator to the decorated function, if I pass it like in above code, how would I call the decorated function?

就像将对象传递给任何函数一样。请参阅上面的示例。装饰器创建一个 file 对象并将其传递给装饰函数。

为了完整性，这里是您的示例程序，错误已修复：

from contextlib import contextmanager

class test:
    def __init__(self, name):
        self._name = name
        print "my name is {0}".format(name)

    def exit(self):
        print "exiting"

    @contextmanager
    def testcm(self):
        print "inside cm"
        try:
            yield self
        finally:
            self.exit()

    def randomtest(self, str):
        print "Inside random test {0}".format(str)


def decorate(name):
    def wrapper(testf):
        def testf_wrapper():
            with test(name).testcm() as testobj:
                return testf(testobj)
        return testf_wrapper
    return wrapper

@decorate("whatever")
def testf(testobj):
    testobj.randomtest("randomness")


testf()

如何在装饰器中使用上下文管理器以及如何将在装饰器中创建的对象传递给装饰函数

How to use a context manager inside a decorator and how to pass an object created in decorator to decorated function

python

python-decorators

contextmanager