在类 python3 中使用装饰器

Question

请帮助了解如何在类中正确使用装饰器。 例如，我想在执行类之前检查变量，如果它存在使用这个变量运行类。 如何做到这一点？

我不工作的例子：

import os,sys
import requests


def checkCredentials(function):
  KEY = os.environ['KEY']
  if KEY:
    print("exist")
    return KEY
  else:
    print("nope")
    sys.exit(0)


@checkCredentials
class StartUp(object):

  def __init__(self, argument1):
    self.argument1 = argument1

  def printstr(self):
    print(self.argument1)


if __name__ == '__main__':
  start = StartUp()

我的错误：

python3.6 ./main.py 
exist
Traceback (most recent call last):
  File "./main.py", line 26, in <module>
    start = StartUp()
TypeError: 'str' object is not callable

Answer 1

你应该阅读更多关于装饰器的内容，也许检查PEP 3129 或PEP 318。以下来自前 PEP 的示例说明了类装饰器的语义：

# Solution 1: Without class decorators
class A:
  pass
A = foo(bar(A))

# Solution 2: With class decorators
@foo
@bar
class A:
  pass

这两种解决方案产生相同的结果。在Solution 1 中，函数foo 和bar 被显式调用以修改初始类A 并生成修改后的变体。通过使用装饰器，可以更清楚地实现相同的效果：这是Solution 2。

让我们将此应用于您的代码示例。你的最后一行代码是

start = StartUp()

翻译成

start = checkCredentials(StartUp)()

但是，checkCredentials 要么不返回任何内容，要么返回 str，因此是 TypeError。我不确定你想用装饰器实现什么。如果您只想测试特定环境变量是否存在，则以下装饰器可能会产生所需的效果：

def checkCredentials(cls):
  KEY = os.environ['KEY']
  if KEY:
    print("exist")
    return cls
  else:
    print("nope")
    sys.exit(0)

请注意，如果设置了环境变量（快乐的代码路径），这将从装饰器返回原始类，否则终止执行（因此无需从装饰器返回任何内容）。