带有 *args 的函数周围的装饰器

Question

我尝试学习 python 装饰器，因为这些东西非常有用。我了解简单的装饰器是如何工作的，但我试图围绕需要 *args 的函数制作装饰器，但它不起作用。当然，我错过了一些元素。我一直在浏览互联网寻找答案，但我找不到答案。

内部函数将列表中的所有整数相加，包装器应该检查列表中的所有元素是否都是整数。

def wrapper(func):
    def inner(*args):
        for i in range(0, len(args[0])): #It does not iven get there.
            if not isinstance(i, int):
                return 'Invaild values.'
        else:       
            return func(*args)
    return inner

def add(*args):
    result = 0
    for i in range(0, len(args[0])):
        result += args[0][i]
    return result


def main():
    numbers = [1, '2', 3, 4]
    print(add(numbers))
    numbers = [1, 2, 3, 4]
    print(add(numbers))


if __name__ == '__main__':
    main()

我得到的错误。

Traceback (most recent call last):
  File "wrapper_of_function.py", line 23, in <module>
    main()
  File "wrapper_of_function.py", line 20, in main
    print(add(numbers))
  File "wrapper_of_function.py", line 15, in add
    result += args[0][i]
TypeError: unsupported operand type(s) for +=: 'int' and 'str'

Answer 1

你测试i 是否是int 的实例，什么时候你应该测试args[0][i] 是否是int 的实例：

def wrapper(func):
    def inner(*args):
        for i in range(len(args[0])): #It does not iven get there.
            if not isinstance(args[0][i], int):
                return 'Invalid values.'       
        return func(*args)
    return inner

@wrapper
def add(*args):
    result = 0
    for i in range(0, len(args[0])):
        result += args[0][i]
    return result


def main():
    numbers = [1, '2', 3, 4]
    print(add(numbers))
    numbers = [1, 2, 3, 4]
    print(add(numbers))


if __name__ == '__main__':
    main()

Answer 2

也许你打算实际使用你的装饰器

def wrapper(func):
    def inner(*args):
        for i in args[0]:
            if not isinstance(i, int):
                raise Exception('a non integer was encountered')
        else:       
            return func(*args)
    return inner

@wrapper
def add(*args):
    total = 0
    for i in args[0]:
        total += i
    return total


def main():
    numbers = [1, '2', 3, 4]
    try:
        print(add(numbers))
    except Exception as e:
        print 'invalid input %r - %s' % (numbers, e)
    numbers = [1, 2, 3, 4]
    print(add(numbers))


if __name__ == '__main__':
    main()

请注意，此实现的一个弱点是wrapped 必须了解传递给它的内容才能正常工作。它需要知道检查第 0 个元素。对于其他目的，这不会是一个非常有用的功能。即使您使用*args，您仍然只将一个参数传递给add。如果您确实将多个参数传递给它所装饰的任何东西，您可能会发现 wrapped 通常更有用。

这样做

def wrapper(func):
    def inner(*args):
        for i in args:
            if not isinstance(i, int):
                raise Exception('a non integer was encountered')
        else:       
            return func(*args)
    return inner

@wrapper
def add(*args):
    total = 0
    for i in args:
        total += i
    return total


def main():
    numbers = [1, '2', 3, 4]
    try:
        print(add(*numbers))
    except Exception as e:
        print 'invalid input %r - %s' % (numbers, e)
    numbers = [1, 2, 3, 4]
    print(add(*numbers))


if __name__ == '__main__':
    main()

Answer 3

抛出的类型错误是因为您无法像 [1,'2',3,4] 中那样添加 Int 和 Str。您的装饰者应该在系统抛出错误之前执行检查，但您并没有真正使用它。

此外，还有一些错误。

def wrapper(func):
    def inner(*args):
        # 1, you don't use args other than args[0], what's the point of using list arguments?
        for i in range(0, len(args[0])): #It does not iven get there.
            # 2, you are running isinstance on array indexes, you should run this on values.
            if not isinstance(i, int):
                # 3, it's a bad practice to return different types in different scenarios. Return a special value and print the error message, or just raise a exception
                return 'Invaild values.'
        # 4, else keyword is redundant. the if statement is inside the loop.
        else:       
            return func(*args)
    return inner

# 5, you didn't use wrapper at the first place.
def add(*args):
    result = 0
    # 6, only using the first argument rather than the list
    for i in range(0, len(args[0])):
        result += args[0][i]
    return result

我为你重写了这段代码，希望它能按你的需要工作。

def wrapper(func):
    def inner(*args):
        for arg in args:
            if not isinstance(arg, int):
                raise TypeError('Invalid values')
        return func(*args)
    return inner

@wrapper
def add(*args):
    result = 0
    for arg in args:
        result += arg
    return result

检查是否有效

Answer 4

你应该在add函数上使用@wrapper：

def wrapper(func):
    def inner(args):
        for item in args:
            if not isinstance(item, int):
               return  'Invalid value found'
        return func(args)
    return inner

@wrapper    
def add(args):
    return sum(args)


def main():
    numbers = [1, '2', 3, 4]
    print(add(numbers))
    numbers = [1, 2, 3, 4]
    print(add(numbers))


if __name__ == '__main__':
    main()

Answer 5

@Zetys : 在 numbers = [1, '2', 3, 4] 中，第二个元素是一个字符串 所以它不能添加到result，因为它是int（结果= 0）。如果您希望 string 参数转换为 int 以防它们是数字，您应该使用 int(str) 函数。所以你的代码将是：

def add(*args):
    result = 0
    for i in range(0, len(args[0])):
        result += int(args[0][i])
    return result

顺便说一句，你没有在代码的任何地方使用wrapper 类，所以它不会运行:)