python - python中的装饰器与在函数上调用函数完全一样吗？

Question

我以为是这样

@f
def g():
   print 'hello'

完全一样

def g():
   print 'hello'
g=f(g)

但是，我有这段代码，它使用了 contextlib.contextmanager:

@contextlib.contextmanager
def f():
    print 1
    yield
    print 2
with f:
    print 3

有效并产生 1 3 2

当我试图把它变成

def f():
    print 1
    yield
    print 2
f=contextlib.contextmanager(f)
with f:
    print 3

我得到 AttributeError: 'function' object has no attribute '__exit__'

我错过了什么？ contextlib.contextmanager 中是否有一些黑魔法，或者我是否误解了装饰器的一般工作方式？

Answer 1

是的，装饰器与调用函数并赋值给返回值完全一样

在这种情况下，错误是因为你没有调用函数，所以正确的代码是

def f():
    print 1
    yield
    print 2

f=contextlib.contextmanager(f)
with f():
    print 3

我也不确定你是否测试过代码，因为你给出的装饰器代码会因为同样的原因而失败

@contextlib.contextmanager
def f():
    print 1
    yield
    print 2
with f:
    print 3

错误:

    with f:
AttributeError: 'function' object has no attribute '__exit__'