Question

我有一个复杂的库，用户可以向其中添加功能。然后，该库在程序中使用，该程序随后接受输入。问题是，函数没有按照我想要的方式进行处理。让我说明一下（我简化了代码以说明要点）：

import functools

def register_command(func):
    def wrapper(*args, **kwargs):
        func(*args, **kwargs)
    return wrapper

def execute_command(func, *args, **kwargs)):
    return functools.partial(func, *args, **kwargs))

'''DECORATORS: EASY WAY FOR THE USER TO ADD NEW FUNCTIONS TO THE LIBRARY'''
@register_command
def some_math_function(variable):
    return variable + 1

'''THIS IS THE PROGRAM WHICH USES THE ABOVE LIBRARY & USER-CREATED FUNCTIONS'''
input_var = input("input your variable: ")
response = execute_command(register_command, input_var)
print(response())

因此，如果用户要输入'10'，input_var = 10，我希望算法返回11。相反，我得到了这一点：

<function register_command.<locals>.wrapper at 0x110368ea0>

我要去哪里错了？

Answer 1

让我们看看装饰者首先做什么。装饰器接受一个对象（可以是函数，类或其他可能的对象），执行某些操作，然后返回替换对象，该对象被重新分配为原始对象的名称。

在这种情况下，register_function无用。它返回一个包装，调用未修改的原始包装。根据名称，您可能希望将功能记录在某些注册表中，例如字典：

function_registry = {}

def register_command(func):
    function_registry[func.__name__] = func
    return func

请记住，装饰器不必直接（或完全不）影响功能。只需返回输入函数就没有错。

现在，让我们来看看使用我们刚刚创建的注册表。您当前的execute_command不执行任何操作。它创建并返回一个可调用对象，如果您不带参数调用该对象，该对象将调用。 execute_command实际上并不调用结果。

您可能希望名称为execute_command的函数按名称查找命令并运行它：

def execute_command(name, *args, **kwargs)):
     return function_registry[name](*args, **kwargs)

所以现在您可以做类似的事情

@register_command
def some_math_function(variable):
    return variable + 1

这将使函数保持不变，但会在注册表中添加一个条目，将名称'some_math_function'映射到函数对象some_math_function。

您的程序成为

func_name = input('what function do you want to run? ')  # enter some_math_func
input_var = int(input("input your variable: "))
response = execute_command(func_name, input_var)
print(response())

此示例不执行任何类型检查，跟踪参数数量或以其他方式使您可以通过调用函数来发挥创造力，但是它将使您从装饰器入手。

因此，这是一个基本系统，用于将固定数量的输入转换为所需的类型。参数按提供的顺序转换。如果您没有提供足够的输入，则该函数负责引发错误。不支持关键字：

function_registry = {}

def register_function(*converters):
    def decorator(func):
        def wrapper(*args):
            real_args = (c(arg) for c, arg in zip(converters, args))
            return func(*real_args)
        function_registry[func.__name__] = wrapper
        return wrapper
    return decorator

@registrer_function(int)
def some_math_func(variable):
    return variable + 1

def execute_command(name, *args):
    return function_registry[name](*args)

command = input("command me: ")  # some_math_func 1
name, *args = command.split()
result = execute_command(name, *args)
print(result)

使用包装器返回函数对象

1 个答案: