Python使用修饰器强制函数只接收关键参数

2017-01-05 董付国 Python小屋 Python小屋

前面发过一篇文章介绍了单独一个星号或斜线作为函数参数的含义,请查看:Python函数中单独一个星号或斜线作为形参的含义

如果需要强制函数的所有参数都必须以关键参数形式进行传值,可以在定义函数时把单独一个星号*作为函数第一个参数。例如


>>> def demo(*, a, b):

print(a, b)

>>> demo(a=1, b=2)

1 2

>>> demo(1, 2)

TypeError: demo() takes 0 positional arguments but 2 were given

也可以使用修饰器实现同样的功能,下面的代码首先定义了一个修饰器对函数的关键参数和位置参数进行检查,如果发现有位置参数与关键参数同名则抛出异常。


def mustBeKeywords(func):

    import inspect

    positions = inspect.getargspec(func).args

    def wrapper(*args, **kwargs):

        for pos in positions:

            if pos not in kwargs:

                raise Exception(pos+' must be keyword parameter')

            return func(*args, **kwargs)

    return wrapper


@mustBeKeywords

def demo(a, b, c):

    print(a, b, c)