上一篇文章我们对装饰器有了初步了解,接下来开始是对装饰器的进一步实践
当你写了一个装饰器作用在某个函数上,但是这个函数的重要的元信息比如名字、文档字符串、注解和参数签名都丢失了。这到底是怎样的一种现象呢?直接上代码吧。
# -*- coding:UTF-8 -*-
def show_me(func):
def wrapper():
print \"It is in wrapper.\"
func()
return wrapper
@show_me
def func1():
\'\'\'
this is func1
\'\'\'
print \"running func1\"
if __name__ == \'__main__\':
func1()
print func1.__name__
print func1.__doc__
打印结果:
元数据
我们可以看到func1.__name__
输出为wrapper
,func1.__doc__
输出为None
,哇,全乱套了,怎么办。不要怕,Python
大法自有办法。
温馨提醒:任何时候你定义装饰器的时候,都应该使用 functools 库中的
@wraps
装饰器来注解底层包装函数。
也就是说,装饰器函数应该这样做补充:
from functools import wraps ##导入这个包
def show_me(func):
@wraps(func)
def wrapper():
print \"It is in wrapper.\"
func()
return wrapper
加上这个后,你可以再运行一下,应该会得到完整的元数据.
我们可以看到,上文的被装饰函数(也就是func
函数)是没有参数的,但是在实际应用中,我们大多数的函数都会有参数,这就要用到Python
中的可变参数了。那么对于被装饰函数带有参数的装饰器该怎么写呢,可见如下代码:
# -*- coding:UTF-8 -*-
from functools import wraps
def show_me(func):
@wraps(func)
def wrapper(*args, **kwargs):
print \"强哥好帅!\"
return func(*args, **kwargs)
return wrapper
@show_me
def func4(a, b=2):
print \"a = %s\"%a
print \"b = %s\"%b
if __name__ == \'__main__\':
func4(3, 8)
运行结果如下:
>>强哥好帅!
>>a = 3
>>b = 8
可变参数是个好东西,在Python
中(星号)*
和参数有许多有趣的用法,值得一看。
有时候,@
语法糖后面还可以带参数。比如某个函数功能只有拥有经理权限的人才能访问。可以编写这样的一个装饰器@has_permission(\"manager\")
,下面来举一个简单粗鲁的栗子:
# -*- coding:UTF-8 -*-
from functools import wraps
def has_permission(position=\"\"):
def decorator(func):
@wraps(func)
def wrapper(*args, **kwargs):
if position == \"manager\":
print \"我是经理我怕谁!\"
return func(*args, **kwargs)
else:
print \"You don not have permission to access!\"
return wrapper
return decorator
@has_permission(position=\"manager\")
def func5(name=\"\"):
print \"My name is : %s\"%name
if __name__ == \'__main__\':
func5(\"zhujq\")
结果为:
>>我是经理我怕谁!
>>My name is : zhujq
当position
改为其他时(比如“Boss”
),程序便只能输出:
You don not have permission to access!
Python
还提供了类装饰器与@staticmethod
,@classmethod
,@property
和这三个在面向对象编程中常用的装饰器,我们下次再讲吧~~