Python装饰器

清风穆云 2021-08-09 12:05:14
Python 腾讯云 装饰


装饰器前缘

Python装饰器对于Python初学者可能是一个比较抽象的概念,在了解装饰器之前,我们必须先熟悉Pythond方法的一个特性:Python中的方法可以像普通变量一样当做参数传递给另外一个方法,我们来看一个例子:

def add(x,y):
return x+y
def sub(x,y):
return x-y
def apply(func,x,y):
return func(x,y)
  • 上面两个方法中,我们定义了两个方法 add()sub()用于进行加减法运算。另外还定义了一个方法apply(),这个方法比较特殊,它的参数可以接收方法作为参数,其中第一个参数func参数就是方法类型的参数,
  • x,y两个参数是普通参数,这两个参数将传递到func()方法中,最终执行func(x,y)返回结果。下面我们看这个apply方法的使用。
 print(apply(add,3,4)) #执行结果: 7
print(apply(sub,3,4)) #执行结果:-1

从上面执行结果可以看出,通过调用apply()方法,我们可以实现加减法运算。从执行结果上来看,和单独去调用这2个方法执行是一样的效果,只不过从执行方式上我们把这两个方法作为参数传到另外一个方法去执行了。

问题思考

基于上面的例子,如果我们现在新增一个需求:要求addsub方法执行时打印日志,那么我们该如何处理,常规思维我们可能会直接在两个方法中加入日志打印语句如下所示:

def add(x,y):
logging.info('add logging is run' )
return x+y
def sub(x,y):
logging.info('sub logging is run' )
return x-y
  • 当方法比较少时这样处理也没啥问题,但是当我们新增其他方法也需要加入日志打印时,还这样一个个加入就会显得代码很冗余,这显然违背人生苦短,我用Python的Python初衷。那么如何解决这个问题呢?
  • 上面的案例中apply()方法我们曾经使用它来执行加减法方法的运算,它就像一个公共工具来承载两个方法的执行,那么我们现在需要在方法中增加公共内容,其实就可以直接在这个方法里面改造。
def apply(func,x,y):
logging.info('%s logging is run' %func.__name__)
return func(x,y)
print(apply(add,3,4))
print(apply(sub,3,4))
  • 在上面apply方法中我们增加了一个日志语句,这样以后每次执行addsub方法时就会执行日志打印语句,从而减少在add,sub方法中直接写入日志打印语句,这样就可以精简代码,提高效率。
  • 看似一切都比较完美了,但是还是有一个问题:每次执行都得通过apply这个方法去执行,可能会违背业务逻辑,比如我封装某个方法要调用加法运算,却每次得通过apply这个中转方法来执行,会对代码的可读性造成影响。
  • 那么有没有有一种方法既可以使addsub方法都具有日志打印功能,而且可以直接调用原方法就能实现的两全其美的方法呢?有!这个时候装饰器就开始闪亮登场了!

初识装饰器

首先我们来看一个例子,然后来解释什么是装饰器。

import logging
logging.basicConfig(level=logging.INFO,
format='%(asctime)s %(filename)s[line:%(lineno)d]%(levelname)s%(message)s')
def add(x,y):
return x+y
def logging_tool(func): #日志装饰器
def wrap(x,y):
logging.info(' %s logging is run' % func.__name__)
return func(x,y)
return wrap
add=logging_tool(add)
print(add(3,4))
  • 从上面的例子我们可以看到定义了一个方法logging_tool 参数可以接收方法 在这个方法里面还额外定义了一个方法wrap 参数为2个普通参数x,y
  • wrap参数内部定义了日志打印语句,然后是返回传入方法和参数,执行原方法的运算结果。
  • 最后在logging_tool这个方法里面返回wrap方法,这个方法即支持了日志打印功能和原方法add的求和功能。
  • add=logging_tool(add) 其实就是一个装饰赋值的过程,logging_tool就是一个装饰器,经过logging_tool方法装饰之后,新的add的方法便具有日志打印功能和求和运算功了,我们再次调用装饰之后的add方法add(3,4)便可得到如下结果:
7
2019-07-06 09:08:56,325 decorator_blog.py[line:10]INFO add logging is run

装饰器语法糖

在python中相信使用过unittestpytest单元测试框架的同学对于 @符号一定不会陌生, 比如我们经常在unittest中会看到如下类似的用法:

@unittest.skip("skip Test2")
class Test2(unittest.TestCase):
def setUp(self):
print("Test2 start")
@classmethod
def setUpClass(cls):
print("Class module start test>>>>>>>>")

上面的@符号其实就是一个装饰器符号,它可以代替上面的add=logging_tool(add)赋值过程。那么我们使用语法糖@符号来改造代码如下:

import logging
logging.basicConfig(level=logging.INFO,
format='%(asctime)s %(filename)s[line:%(lineno)d]%(levelname)s%(message)s')
def logging_tool(func): #日志装饰器
def wrap(x,y):
logging.info(' %s logging is run' % func.__name__)
return func(x,y)
return wrap
@logging_tool
def add(x,y):
return x+y
print(add(3,4))
  • 从上面的例子我们可以看出使用@装饰符号更加简洁明了,既不影响add的方法定义结构,也支持了日志打印功能,提高代码复用效率,当有新的方法需要日志打印功能时,同样可以继续在方法顶部加上@logging_tool即可。
  • 装饰器本质上是一个Python函数或类,它可以让其他函数或类在不需要做任何代码修改的前提下增加额外的功能,装饰器的返回值也是一个函数/类对象。
  • 它经常用于有切面需求的场景,比如:插入日志、性能测试、事务处理、缓存、权限校验等场景,装饰器是解决这类问题的绝佳设计。
  • 有了装饰器,我们就可以抽离出大量与函数功能本身无关的代码到装饰器中并继续重用。简单来说:装饰器的作用就是让已经存在的对象添加额外的功能。

*args **kwargs

  • 上面装饰器中我们是根据add的参数也在装饰器wrap方法中定义了两个参数,但是如果新的业务方法有三个参数或者更多那么该如何处理,难道定义多个不同的装饰器吗?显然这样是不合理的,这里我们可以使用不定参数的处理方式,关于*args**kwargs
  • 我们将上面的装饰器loging_tool进行参数改造代码如下:
def logging_tool(func): #日志装饰器
def wrap(*args,**kwargs):
logging.info(' %s logging is run' % func.__name__)
return func(*args,**kwargs)
return wrap
@logging_tool
def add_multi(x,y,z):
return x+y+z
print(add_multi(1,2,3))

通过上面的改造,不过被装饰的业务参数是多少,我们都可以接收处理,这样极大的提高了装饰器的灵活性。

装饰器参数

  • 装饰器还有更大的灵活性,例如带参数的装饰器,在上面的装饰器调用中,该装饰器接收唯一的参数就是执行业务的方法add。如果现在我要根据日志不同级别来分别打印日志,那么该如何处理?先看如下代码:
def logging_tool(level): #日志装饰器
def decorator(func):
def wrap(*args,**kwargs):
if level=='info':
logging.info(' %s info logging is run' % func.__name__)
elif level=='warn':
logging.info(' %s warn logging is run' % func.__name__)
else:
logging.debug(' %s debug logging is run' % func.__name__)
return func(*args,**kwargs) #返回原业务方法结果
return wrap #返回增加装饰器的功能
return decorator #返回整个结果
@logging_tool(level="warn")
def add_multi(x,y,z):
return x+y+z
print(add_multi(1,2,3))

执行结果:

6
2019-07-06 10:39:25,698 decorator_blog.py[line:22]INFO add_multi warn logging is run
  • 从上面例子我们可以看出,首先将装饰器增加了参数level,然后增加了一个方法decorator用来接收业务方法,在wrap内部根据装饰器参数level来进行判断输入不同级别的日志。
  • 在调用装饰器时,同样也会传入参数level=“warn”

functools.wraps

使用装饰器极大地复用了代码,但是他有一个缺点就是原函数的元信息不见了,比如函数的docstring__name__、参数列表,先看例子:

def logging_tool(func): #日志装饰器
def wrap(x,y):
'''logging tool wrap'''
logging.info(' %s logging is run' % func.__name__)
return func(x,y)
return wrap
@logging_tool
def add(x,y):
'''
add function
:return:
'''
return x+y
print(add.__name__)
print(add.__doc__)

执行结果:

wrap
logging tool wrap

通过上面的例子我们不难发现,原本add方法的docstring__name__、属性都被装饰器里面的wrap方法的属性替换了,显然这个不是我们愿意看到的。不过,我们可以通过 @functools.wraps 将原函数的元信息拷贝到装饰器里面的 func 函数中,使得装饰器里面的 func 和原函数有一样的元信息。

import functools
def logging_tool(func): #日志装饰器
@functools.wraps(func) #保留方法的元信息
def wrap(x,y):
'''logging tool wrap'''
logging.info(' %s logging is run' % func.__name__)
return func(x,y)
return wrap
@logging_tool
def add(x,y):
'''add function'''
return x+y
print(add.__name__)
print(add.__doc__)

执行结果:

add
add function

我们可以从上面的例子看到,使用 @functools.wraps 后将方法的元信息得到了保留。

类装饰器

装饰器不仅可以是函数,还可以是类,相比函数装饰器,类装饰器具有灵活度大、高内聚、封装性等优点。使用类装饰器主要依靠类的call方法,当使用 @ 形式将装饰器附加到函数上时,就会调用此方法。

class Foo():
def __init__(self,func):
self._func=func
def __call__(self, *args, **kwargs):
print('类装饰器开始执行')
self._func(*args,**kwargs)
print('类装饰器执行结束')
@Foo
def bar():
print('hello world')
bar()

执行结果

类装饰器开始执行
hello world
类装饰器执行结束

内置装饰器

Python内置了一些装饰器,如:@property, @classmethod, @staticmethod 接下来我们一一阐述这些内置装饰器的用法。

@property

property装饰器是负责把一个方法变成属性,

class Student(object):
@property
def score(self):
return self._score
@score.setter
def score(self, value):
if not isinstance(value, int):
raise ValueError('score must be an integer!')
if value < 0 or value > 100:
raise ValueError('score must between 0 ~ 100!')
self._score = value
s=Student()
s.score=90
print(s.score) #执行结果 90
s.score=101
print(s.score) #执行结果:score must between 0 ~ 100!
s.score='hello'
print(s.score) #执行结果:score must be an integer!

从上面的例子我们可以看出,使用装饰器@property我们将score方法变成了一个类的属性,另外又通过@score.setter变成一个可以赋值的属性,如果不定义的话score就只是一个只读的属性。

@classmethod

@classmethod 类方法:定义备选构造器,第一个参数是类本身(参数名不限制,一般用cls) 我们结合大家熟悉的unittest测试框架看下面这个例子

import unittest
class Test(unittest.TestCase):
@classmethod #类方法装饰器
def setUpClass(cls):
print("test class start .....")
@classmethod
def tearDownClass(cls):
print("test class end ....")
def setUp(self):
print("Test case is testing")
def tearDown(self):
print("test case is end!")
def test_case(self):
print("test_case is runing!")
if __name__=='__main__':
unittest.main()

执行结果

test class start .....
Test case is testing
----------------------------------------------------------------------
test_case is runing!
Ran 1 test in 0.000s
test case is end!
test class end ....
OK
  • 上面案例中我们使用@classmethod来装饰setUpClasstearDownClass两个方法,这两个方便便成为类方法,也就是Test类的所有test类型用例执行时都会执行setUpClasstearDownClass两个方法里面的内容。
  • 这样在实际应用场景中比如进行数据库初始化连接断开操作就非常方便了,不用每个用例方法单独去调用或者单独写初始化方法。

@staticmethod

@staticmethod表示静态方法,静态方法和普通方法不同之处在于,静态方法属于类,无需实例化便可进行调用。

class C(object):
@staticmethod
def f():
print('hello world')
C.f() #运行结果hello world

上面我们通过@staticmehod定义了一个静态方法f() 在调用时可以直接调用,无需实例化。

本文分享自微信公众号 - QA一隅(sutune2020)

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间: 2020-10-12

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

版权声明
本文为[清风穆云]所创,转载请带上原文链接,感谢
https://cloud.tencent.com/developer/article/1858059

  1. 利用Python爬虫获取招聘网站职位信息
  2. Using Python crawler to obtain job information of recruitment website
  3. Several highly rated Python libraries arrow, jsonpath, psutil and tenacity are recommended
  4. Python装饰器
  5. Python实现LDAP认证
  6. Python decorator
  7. Implementing LDAP authentication with Python
  8. Vscode configures Python development environment!
  9. In Python, how dare you say you can't log module? ️
  10. 我收藏的有关Python的电子书和资料
  11. python 中 lambda的一些tips
  12. python中字典的一些tips
  13. python 用生成器生成斐波那契数列
  14. python脚本转pyc踩了个坑。。。
  15. My collection of e-books and materials about Python
  16. Some tips of lambda in Python
  17. Some tips of dictionary in Python
  18. Using Python generator to generate Fibonacci sequence
  19. The conversion of Python script to PyC stepped on a pit...
  20. Python游戏开发,pygame模块,Python实现扫雷小游戏
  21. Python game development, pyGame module, python implementation of minesweeping games
  22. Python实用工具,email模块,Python实现邮件远程控制自己电脑
  23. Python utility, email module, python realizes mail remote control of its own computer
  24. 毫无头绪的自学Python,你可能连门槛都摸不到!【最佳学习路线】
  25. Python读取二进制文件代码方法解析
  26. Python字典的实现原理
  27. Without a clue, you may not even touch the threshold【 Best learning route]
  28. Parsing method of Python reading binary file code
  29. Implementation principle of Python dictionary
  30. You must know the function of pandas to parse JSON data - JSON_ normalize()
  31. Python实用案例,私人定制,Python自动化生成爱豆专属2021日历
  32. Python practical case, private customization, python automatic generation of Adu exclusive 2021 calendar
  33. 《Python实例》震惊了,用Python这么简单实现了聊天系统的脏话,广告检测
  34. "Python instance" was shocked and realized the dirty words and advertisement detection of the chat system in Python
  35. Convolutional neural network processing sequence for Python deep learning
  36. Python data structure and algorithm (1) -- enum type enum
  37. 超全大厂算法岗百问百答(推荐系统/机器学习/深度学习/C++/Spark/python)
  38. 【Python进阶】你真的明白NumPy中的ndarray吗?
  39. All questions and answers for algorithm posts of super large factories (recommended system / machine learning / deep learning / C + + / spark / Python)
  40. [advanced Python] do you really understand ndarray in numpy?
  41. 【Python进阶】Python进阶专栏栏主自述:不忘初心,砥砺前行
  42. [advanced Python] Python advanced column main readme: never forget the original intention and forge ahead
  43. python垃圾回收和缓存管理
  44. java调用Python程序
  45. java调用Python程序
  46. Python常用函数有哪些?Python基础入门课程
  47. Python garbage collection and cache management
  48. Java calling Python program
  49. Java calling Python program
  50. What functions are commonly used in Python? Introduction to Python Basics
  51. Python basic knowledge
  52. Anaconda5.2 安装 Python 库(MySQLdb)的方法
  53. Python实现对脑电数据情绪分析
  54. Anaconda 5.2 method of installing Python Library (mysqldb)
  55. Python implements emotion analysis of EEG data
  56. Master some advanced usage of Python in 30 seconds, which makes others envy it
  57. python爬取百度图片并对图片做一系列处理
  58. Python crawls Baidu pictures and does a series of processing on them
  59. python链接mysql数据库
  60. Python link MySQL database