[Python知识库]一个例子看懂 Python 装饰器

我们开发 Python 代码，会经常碰到装饰器。

Python 中装饰器通常用来装饰函数、或者类的方法。

通常被装饰后的函数， 会在原有的函数基础上，增加一点功能。

1.用一个简单的栗子引入概念

栗子只求方便理解，未必是解决问题的最佳思路

假如你有一段之前写的代码，上面有些内容你想进行修改
你想在输出内容前面加以个 ：现在时间是

import time
def oldfunction():
    print()
    return f"{time.asctime()}，吴彦祖在看我的博客"
print(oldfunction())

如果我们需要在所有这样的函数 返回字符串前面 都加上开头: 现在时间是

这时候，我们完全可以 不去修改原来的函数 ， 而是 使用装饰器，像这样

import time

# 定义一个装饰器函数
def xiushi(function):
    # 注意在这个函数中什么都没有执行，我只是定义了一个cat函数
    def cat():
        x = function()
        return f'现在时间是： {x}'
    return cat() # 到这一步时，cat被执行了，cat中又会执行xiushi函数的参数函数function

def oldfunction():
    print()
    return f"{time.asctime()}，吴彦祖在看我的博客"

# 装饰 oldfunction
oldfunction = xiushi(oldfunction)
print(oldfunction)

首先执行的是

oldfunction = xiushi(oldfunction)

把 oldfunction函数 当作参数放到 xiushi 函数中去

然后执行修饰函数，会返回 cat()

所以返回的是执行 ca() 后的结果

在 cat 中完成了对 oldfunction 的修饰

2.语法糖

import time
# 定义一个装饰器函数
def xiushi(function):
    def cat():
        x = function()
        return f'现在时间是： {x}'

    return cat()

@xiushi
def oldfunction():
    print()
    return f"{time.asctime()}，吴彦祖在看我的博客"

print(oldfunction)

以 @xiushi开头后面接装饰器函数 的这种写法，是一种 语法糖 ，也就是便捷写法。

当Python解释器执行 完 下面语句的时候，

@xiushi
def oldfunction():
    ....

就等于执行了这样的一条语句

oldfunction = xiushi(oldfunction)

这样写起来比较方便快捷。

可能有的读者会问，干嘛要这么麻烦？
我们直接改这个函数的内容，不是更简单吗？

import time
def oldfunction():
    print()
    return f"{现在时间是，time.asctime()}，吴彦祖在看我的博客"
print(oldfunction())

装饰器经常被用在库和框架中， 给别的开发者使用。

这些库的开发者预料到 使用者 开发的函数可能需要 一些增强的功能。

但是 这些库的开发者 没法去改使用者的代码， 就可以把这些增强的部分做在 装饰器函数中。

这样使用者，只需要在他们的函数前面上@xxx 就使用了这些增强的功能了。