谈谈Python中的decorator装饰器,如何更优雅的重用代码

时间:2022-12-18 12:19:21

众所周知,Python本身有很多优雅的语法,让你能用一行代码写出其他语言很多行代码才能做的事情,比如:

最常用的迭代(eg: for i in range(1,10)), 列表生成式(eg: [ x*x for x in range(1,10) if x % 2 ==  0])

map()能让你把函数作用于多个元素, reduce()能让你把多个元素的结果按照你预想的方式组合在一起,filter()能让你快速筛选出复合条件的数据

以上具体用法可以参考https://www.liaoxuefeng.com/wiki/0014316089557264a6b348958f449949df42a6d3a2e542c000/0014317793224211f408912d9c04f2eac4d2af0d5d3d7b2000

而我们这次要讨论的装饰器decorator,可以在不改变现有函数的前提下更有效率的重用代码

比如我们在实际工作当中,经常需要添加try...except来捕获异常,但是一个个加也太麻烦了,此时我们就可以用decorator装饰器来实现

比如我们有以下原始函数

def hello():
    print("Hello, world!")

def bye():
    print("Bye, world!")

正常情况下,如果都需要捕获异常的话,需要加两次try...except来做:

def main():
    try:
        hello()
    except Exception as e:
        print('except:', e)
    ....
    ....
    try:
        bye()
    except Exception as e:
        print('except:', e)
    ....
    ....

但是当我们有装饰器decorator的时候,一切都会变得特别优雅而简单,首先定义好我们的装饰器:

import functools

def decorator_try(func):
    @functools.wraps(func)
    def wrapper(*arg,**kw):
        try:
            func(*arg, **kw)
        except Exception as e:
            print('except:', e)
    return wrapper

然后只需要在原来的hello()和bye()函数定义之前添加一行语法就可以:

@decorator_try
def hello():
    print("Hello, world!")

@decorator_try
def bye():
    print("Bye, world!")

然后执行的时候任何东西都不用加

def main():
    hello()
    ....
    ....
    bye()

结果为:

>>> hello()
Hello, world!
>>> hello(1,2)
except: hello() takes 0 positional arguments but 2 were given

具体的关于decorator装饰器的语法解释可以参考https://www.liaoxuefeng.com/wiki/0014316089557264a6b348958f449949df42a6d3a2e542c000/0014318435599930270c0381a3b44db991cd6d858064ac0000

另外要注意装饰器定义中 func(*arg, **kw) 和 return wrapper的区别,注意看一个是带参数,一个不带参数与括号,带参数表示执行这个函数,不带参数和括号代表把定义的函数作为一个参数传递了过去,这对理解decorator的语法是至关重要的。因为:

@decorator_try放到hello()函数的定义前,相当于执行了语句:

hello = decorator_try(hello)

 如果想更深入的了解decorator装饰器,推荐一篇博文https://www.cnblogs.com/zh605929205/p/7704902.html