面试中经常被问到 python 装饰器,在这里做一个完整的总结。
装饰器
装饰器其实就是一个闭包,把一个函数当参数然后返回一个替代版本函数。
下面来实现一个简单的修饰器
def outer(func):
def inner():
print('call %s():', func.__name__)
return func()
return inner
decorated = outer(foo) # 2
decorated()
print(decorated.__name__)
# 输出如下:
call %s(): foo
2016.04.08
inner
outer
是一个装饰器,它接收一个函数作为参数,并返回一个函数。
我们可以认为变量 decorated
时函数 foo
的一个装饰版本,一个加强版本。我甚至可以用装饰版本完全取代原先的函数 foo
,对此我们只需使用 foo = outer(foo)
简单的赋值给新的foo
即可。
使用 @
标识符将装饰器应用到函数
把@outer
放到foo()
函数的定义处,相当于执行了语句:
foo = outer(foo)
需要明白的是,这样的做法和先前简单的用包装方法替代原有方法是一模一样的, python只是加了一些语法糖让装饰的行为更加的直接明确和优雅一点。
*args **kwargs
*args 使用方法有如下两点:
- 定义方法的时候标志这个方法能够接受任意的位置参数
- 调用方法的时候额外的参数可以从一个可迭代的列表中取得
分别举两个例子:
定义函数时使用 *args
def test(a, b, *args):
print(a, b, args)
test(a, b, 3, 4, 5)
# 输出结果如下:
a b (3, 4, 5)
调用函数时使用 *args
def add(x, y):
return x + y
lst = [1,2]
add(lst[0], lst[1]) # 1
3
add(*lst) # 2
3
#1处的代码和#2处的代码所做的事情其实是一样的,在#2处,python为我们所做的事其实也可以手动完成。**
代表着键值对的字典,和 *
所代表的意义相差无几。
更通用的装饰器
有了 *args
我们就能实现一个能够应用在任何方法上的装饰器。比如,如果我们要实现一个能应用在任何方法上的类似于计数器的装饰器,不需要改变原有方法的任何逻辑。这意味着装饰器能够接受拥有任何签名的函数作为自己的被装饰方法,同时能够用传递给它的参数对被装饰的方法进行调用。
def logger(func):
def inner(*args, **kwargs): #1
print "Arguments were: %s, %s" % (args, kwargs)
return func(*args, **kwargs) #2
return inner
请注意我们的函数inner,它能够接受任意数量和类型的参数并把它们传递给被包装的方法,这让我们能够用这个装饰器来装饰任何方法
@logger
def foo1(x, y=1):
return x * y
@logger
def foo2():
return 2
foo1(5, 4)
Arguments were: (5, 4), {}
20
foo1(1)
Arguments were: (1,), {}
1
foo2()
Arguments were: (), {}
2