Python学习之如何使用装饰器 @decorator
接上一篇文章写的python学习之装饰器是什么?
建议大家在看这篇文章之前还是点击上面这篇文章看一下,先了解一下什么是装饰器之后再看。
猿人学python之我的学习笔记分享:如何使用装饰器 @decorator
上面代码中使用装饰器的方法看起来有点复杂,其实真正的装饰器的Python语法是这样的:
装饰器的Python语法
@decorator_funcdef say_hi(): print 'Hi!'
@ 符合是装饰器的语法糖,在定义函数say_hi时使用,避免了再一次的赋值语句。
上面的语句等同于:
def say_hi(): print 'Hi!'say_hi = decorator_func(say_hi)
装饰器的顺序
@a@b@cdef foo(): print('foo')# 等同于:foo = a(b(c(foo)))
带参数函数的装饰器
def decorator_func(some_func): def wrapper_func(*args, **kwargs): print("Wrapper function started") some_func(*args, **kwargs) print("Wrapper function ended") return wrapper_func@decorator_func def say_hi(name): print ("Hi!" + name)
上面代码中,say_hi函数带有一个参数。通常情况下,不同功能的函数可以有不同类别、不同数量的参数,在写wrapper_func的时候,我们不确定参数的名称和数量,可以通过*args 和 **kwargs 来引用函数参数。
带参数的装饰器
不仅被装饰的函数可以带参数,装饰器本身也可以带参数。参考下面的例子:
def use_logging(level): def decorator(func): def wrapper(*args, **kwargs): if level == "warn": logging.warn("%s is running" % func.__name__) return func(*args) return wrapper return decorator@use_logging(level="warn")def foo(name='foo'): print("i am %s" % name)
简单来说,带参数的装饰器就是在没有参数的装饰器外面再嵌套一个参数的函数,该函数返回那个无参数装饰器即可。
类作为装饰器
前面我们提到装饰器是可调用对象。在Python里面,除了函数,类也是可调用对象。使用类装饰器,优点是灵活性大,高内聚,封装性。通过实现类内部的__call__
方法,当使用 @ 语法糖把装饰器附加到函数上时,就会调用此方法。
class Foo(object): def __init__(self, func): self._func = funcdef __call__(self): print ('class decorator runing') self._func() print ('class decorator ending')@Foodef say_hi(): print('Hi!') say_hi()# Output:# class decorator running# Hi!# class decorator ending
functools.wraps
使用装饰器极大地复用了代码,但是他有一个缺点就是原函数的元信息不见了,比如函数的docstring、__name__、参数列表,先看看下面例子:
def decorator_func(some_func): def wrapper_func(*args, **kwargs): print("Wrapper function started") some_func(*args, **kwargs) print("Wrapper function ended") return wrapper_func@decorator_func def say_hi(name): '''Say hi to somebody''' print ("Hi!" + name) print(say_hi.__name__) # Output: wrapper_funcprint(say_hi.__doc__) # Output: None
可以看到,say_hi函数被wrapper_func函数取代,它的__name__ 和 docstring 也自然是wrapper_func函数的了。
不过不用担心,Python有functools.wraps,wraps本身也是一个装饰器,它的作用就是把原函数的元信息拷贝到装饰器函数中,使得装饰器函数也有和原函数一样的元信息。
from functools import wrapsdef decorator_func(some_func): @wraps(func) def wrapper_func(*args, **kwargs): print("Wrapper function started") some_func(*args, **kwargs) print("Wrapper function ended") return wrapper_func@decorator_func def say_hi(name): '''Say hi to somebody''' print ("Hi!" + name) print(say_hi.__name__) # Output: say_hiprint(say_hi.__doc__) # Output: Say hi to somebody
类的内置装饰器
类属性@property
静态方法@staticmethod
类方法@classmethod
通常,我们需要先实例化一个类的对象,再调用其方法。
若类的方法使用了@staticmethod或@classmethod,就可以不需要实例化,直接类名.方法名()来调用。
从使用上来看,@staticmethod不需要指代自身对象的self或指代自身类的cls参数,就跟使用普通函数一样。@classmethod不需要self参数,但第一个参数必须是指代自身类的cls参数。如果在@staticmethod中要调用到这个类的一些属性方法,只能直接类名.属性名,或类名.方法名的方式。
而@classmethod因为持有cls参数,可以来调用类的属性,类的方法,实例化对象等。
总结
通过认识Python的函数,我们逐步弄清了装饰器的来龙去脉。装饰器是代码复用的好工具,在编程过程中可以在适当的场景用多多使用。
- 点赞
- 收藏
- 关注作者
评论(0)