在ContextDecorator中有没有方法访问函数的属性/参数？

Question

我试图用Python的contextlib.ContextDecorator班级。在

有没有办法在上下文管理器中访问修饰函数的参数？在

下面是我正在做的一个例子：

from contextlib import ContextDecorator

class savePen(ContextDecorator):
    def __enter__(self):
        self.prevPen = self.dc.GetPen()     # AttributeError
        return self

    def __exit__(self, *exc):
        self.dc.SetPen(self.prevPen)
        return False

鉴于上述情况，这：

^{pr2}$

应等同于：

def func(dc, param1, param2):
    prevPen = dc.GetPen()
    # do stuff, possibly changing the pen style
    dc.SetPen(prevPen)

---

我在文档中搜寻了contextlib，没有发现任何有用的东西。在

有人知道如何从ContextDecorator类中访问修饰函数的属性吗？在

编辑1:

正如@chepner在this response中所说，ContextDecorator是sugar for

def func(dc, param1, param2):
    with savePen():
        ...

它不能访问函数的参数。在

但是，在本例中，with savePen()中运行的任何内容都可以访问函数参数dc、param1和{}。这使我认为我应该能够使用ContextDecorator访问它们。在

例如，这是有效的：

def func(dc, param1, param2):
    with savePen():
        print(param1)

Answer 1

我用我以前在Python2中做过的东西，建造了自己的装饰工。在

我没有使用上下文管理器，而是使用了try...finally结构。在

以下是我的想法（我去掉了所有能使docstring正确的东西）：

class savePen(object):
    def __init__(self, func):
        self.func = func

    def __call__(self, *args, **kwargs):
        """This provides support for functions """
        dc = args[0]
        prevPen = dc.GetPen()
        try:
            retval =  self.func(*args, **kwargs)
        finally:
            dc.SetPen(prevPen)
        return retval

    def __get__(self, obj, objtype):
        """ And this provides support for instance methods """
        @functools.wraps(self.func)
        def wrapper(*args, **kwargs):
            dc = args[0]
            prevPen = dc.GetPen()
            try:
                retval = self.func(obj, *args, **kwargs)
            finally:
                dc.SetPen(prevPen)
            return retval
        return wrapper