调用修饰的外部函数

Question

我正在寻找一种方法让脚本执行一些我不知道名称并在另一个模块上定义的函数（假设由另一个用户提供），但用我定义的函数进行修饰。

例如，外部脚本是：

    from MyScript import preprocess, process, postprocess 

    @preprocess(foo, bar)
    def external_provider_random_name():
        return "foo"

    @process(foobar)
    def external_provider_random():
        return "bar"

    @postprocess(foo, bar)
    def external_provider():
        return "foobar"

然后我会在我的应用程序中导入它并执行函数（我不知道它们的名字），并能够使用修饰函数返回的内容。

我想到装饰器的原因是成为flask似乎暴露了类似的行为： 用户可以通过使用$@app.route(my/route) 装饰函数来定义新路由。然后在路由上执行 http 调用时执行此函数，并且应用程序可以使用该函数返回的任何内容（一个 json）将其返回给用户。

但是，这种行为似乎与装饰器的实际定义不符，我不知道如何实现这一点。我在这里遗漏了什么还是只使用装饰器的解决方案？

Answer 1

你只需要从你的包装器中取出传入函数的副本。您可以将函数引用存储在全局名称、list、dict 或您选择的任何其他方式中。

这是一个 MyScript.py 的示例，它将其函数引用存储在 dict 中：

funcs = { }

def preprocess(*args):
    def wrapper(fn):
        funcs['preprocess'] = fn
        return fn
    return wrapper

def process(*args):
    def wrapper(fn):
        funcs['process'] = fn
        return fn
    return wrapper

def postprocess(*args):
    def wrapper(fn):
        funcs['process'] = fn
        return fn
    return wrapper

所有初始化完成后，就可以调用函数了：

MyScript.funcs['preprocess']()

Answer 2

您似乎想要创建某种函数注册表。这是一个简单的示例实现，它使用字典来跟踪注册的函数。

class Tracker(dict):
    def register(self, identifier):
        def wrapper(function):
            self[identifier] = function
            return function
        return wrapper


tracker = Tracker()

@tracker.register('A')
def function_A():
    print('Blah')

tracker['A']() # Blah

如果要注册一个已经定义好的函数，可以使用：

tracker.register('A')(function_A)