58、Python之函数高级：不定参数的函数，写出更加通用的装饰器

引言

上一篇文章中，我们见到引入了Python中的装饰器，通过一个简单的案例实现了一个初步的装饰器，但是，这个装饰器其实是有些缺陷。这一篇文章中，我们对上一篇文章中的装饰器进行一个优化升级，从而写出更加通用的装饰器。

本文的主要内容有：

1、简陋装饰器的缺陷

2、关于函数参数的更加通用的写法

3、更加规范通用的装饰器实现方式

简陋装饰器的缺陷

我们之所以需要使用装饰器，很多时候就是因为需要对很多现有功能，动态添加新的功能，实现更高级别的代码复用。

但是，装饰器是作用于函数的，其增强的目标是函数。但是，根据实际的需要，每个函数的参数及返回值可能是多种多样的。怎么能够写出一个装饰器，能够对各种函数都能进行装饰呢？

首先，回看一下上一篇文章中，关于统一添加登录功能的装饰器的写法：

# 扩展1：新增登录功能
def login(user):print(f"用户[{user}]登录成功")# 扩展2：装饰器：对传入的函数进行登录功能的动态添加
def login_wrap(func):def inner(user):login(user)func(user)return inner@login_wrap
def read(user):print(f"用户[{user}]查看系统相关信息")@login_wrap
def write(user):print(f"用户[{user}]修改系统相关信息")

这个装饰器最大的问题，显然是只能封装只有一个参数而且没有返回值的函数。如果函数有返回值或者函数参数个数不是一个，就会出现问题，比如，尝试通过@login_wrap封装下面这个函数：

def send_msg(from_user, to_user, content):print(f"{from_user}对{to_user}说：{content}")if __name__ == '__main__':send_msg('张三', '李四', '天气真好，万里无云，不远处飘着朵朵白云')

正常执行结果：

如果尝试使用@login_wrap进行封装：

from m1 import login_wrap@login_wrap
def send_msg(from_user, to_user, content):print(f"{from_user}对{to_user}说：{content}")if __name__ == '__main__':send_msg('张三', '李四', '天气真好，万里无云，不远处飘着朵朵白云')

执行结果：

只能接收一个参数，却传入了3个参数。

关于函数参数的更加通用的写法

只要装饰器嵌套的内部函数能够接收任意不定长参数，并且返回任意返回值即可（这里的任意，是指包装函数是什么样，装饰器装饰之后也应当保持一样）。

首先，返回值的通用化处理，是比较简单的，只需要将被包装函数的返回值进行原样返回即可（如果函数没有返回值，实际是返回None）。

由于，返回值的处理比较简单，这里就不进行代码的演示了。

比较头痛的是任意函数参数的实现。其实，我们在Python内置模块的函数定义中，总能看到这种任意函数参数的写法，比如：

再比如：

其实，我们在前面的文章《一颗星，两颗星，满天都是小星星》中，已经介绍过*在函数定义中的写法。

我们可以定义一个这样的函数，然后看下，不同形式的参数传递，这种函数的形参写法是如何接收参数传递的。

直接看代码：

def test_args(*args, **kwargs):print(f"args: {args}")print(f"kwargs: {kwargs}")if __name__ == '__main__':test_args()test_args(1, 2, 3)test_args('张三', to='李四', msg='你好')

运行结果：

可以看到，一个*的形参args会把所有位置参数接收，以一个元组的形式进行存储；两个*的形参kwargs会把所有关键字参数进行接收，以一个字典的形式进行存储。

只需要通过：def xxxx(*args, **kwargs)这种方式，就可以让函数接收任意参数了。

更加规范通用的装饰器实现方式

任意函数的参数形式以及任意函数返回值都已经可以搞定了，那么我们就可以把前面的装饰器进行调整优化了，让它变得更加通用。

直接看代码：

# 扩展1：新增登录功能
def login(*args, **kwargs):print(f"用户[{args[0]}]登录成功")# 扩展2：装饰器：对传入的函数进行登录功能的动态添加
def login_wrap(func):def inner(*args, **kwargs):login(args[0])return func(*args, **kwargs)return inner@login_wrap
def read(user):print(f"用户[{user}]查看系统相关信息")@login_wrap
def write(user):print(f"用户[{user}]修改系统相关信息")@login_wrap
def send_msg(from_user, to_user, content):print(f"{from_user}对{to_user}说：{content}")if __name__ == '__main__':read('张三')write('李四')send_msg('张三', '李四', '天气真好，万里无云，不远处飘着朵朵白云')

执行结果：