python - 用装饰函数填充字典

Question

我正在创建一个基本上运行一堆测试的应用程序，每个测试都是一个单独的函数。我有一个像这样的字典形式的测试/功能列表。

test = {
    "test 1": f1,
    "test 2": f2,
    "test 3": f3
}

我可以这样执行函数:

for testn, _ in test.items():
    test[testn]()

但是我想做这样的事情:

tests = {}

@task("test 1")
def f1():
   ...

@task("test 2")
def f1():
   ...

def run():
    for testn, _ in tests:
        tests[test]()

我想知道这怎么可能。任何帮助，将不胜感激。

Answer 1

您可以创建一个简单的装饰器来填充您的函数注册表测试:

tests = {}
def task(name):
   def wrapper(funct):
      tests[name] = funct
      return funct
   return wrapper

@task("test 1")
def f1():
  ...

@task("test 2")
def f2():
  ...

def run():
  for testn, _ in tests.items():
    tests[test]()

现在，在调用 run 之前，在遍历字典时，可以观察到结构是正确的:

print({a:b.__name__ for a, b in tests.items()})

输出:

{'test 2': 'f2', 'test 1': 'f1'}

但是请注意，tests 确实包含所有函数实例:

{'test 2': <function f2 at 0x10de0e5f0>, 'test 1': <function f1 at 0x10de0e578>}

这个装饰器稍微偏离了接收包装函数、截取参数并通过仅返回传递给它的原始函数对象来调用包装函数的正常装饰器模式。这节省了必须实现另一个包装器来拦截参数的空间，并允许在函数绑定(bind)到其对象时立即添加注册表。

另外，请注意 tests 是一个字典，因此在不首先调用 dict.items() 并引发 ValueError 的情况下尝试通过解包对其进行迭代: 太多值无法解压 错误。

编辑:如果函数变成方法，那么解决方案是:

class Tests:
    def __init__(self):
       self.tests = {}
    def task(self, name):
       def wrapper(f):
          self.tests[name] = f
       return wrapper

test = Tests()
@test.task('test 1')
def f1():
   pass

@test.task('test 2')
def f2():
   pass

for a, b in test.tests.items():
   pass

print({a:b.__name__ for a, b in test.tests.items()})

输出:

{'test 2': 'f2', 'test 1': 'f1'}