上一篇的教程可查看https://blog.csdn.net/qq_45812220/article/details/137815060?csdn_share_tail=%7B%22type%22%3A%22blog%22%2C%22rType%22%3A%22article%22%2C%22rId%22%3A%22137815060%22%2C%22source%22%3A%22qq_45812220%22%7D
本篇主题:如何对含有返回值的函数进行装饰
类似于之前对含有参数的函数进行装饰的思路一样,我们调用被装饰器装饰的函数时,实际上调用的是inner
函数,因此,必须要让inner
函数具有返回值。因此做一下简单的修改:
def addChina(func):
def inner(*args,**kwargs):
print("你好,来自中国的用户")
res = func(*args,**kwargs)
return res
return inner
def messageOne(a,b):
print("今天天气是",a,'\t','今天的空气质量为',b)
return a+' '+b
temp = messageOne('晴转多云','优')
print(temp)
装饰完毕以后,可以发现,我们可以正常调用messageOne
函数并赋值给变量temp
,最后也成功打印了返回值结果。代码的运行结果如下:
你好,来自中国的用户
今天天气是 晴转多云 今天的空气质量为 优
晴转多云 优
现在,我们的装饰器就比较通用了,既可以对任意长参数的函数进行装饰,也可以返回其对应的返回值