如何写一个Python万能装饰器,既可以装饰有参数的方法,也可以装饰无参数方法,或者有无返回值都可以装饰
需求:定义一个装饰器,既可以装饰有参数的方法,又可以装饰没有参数的方法,既可以装饰有返回值的方法,又可以装饰没有返回值的方法
代码如下:
def decorate(test): #定义一个装饰器函数 def bold(*args,**kwargs): #*args可以接收任意多个参数,**kwargs可以接受任意多个字典参数 print('style="font-weight:blod"') # 替换成你自己的代码 res = test(*args,**kwargs) #这样写,即使被装饰的test函数没有参数或者没有返回值都没关系,不会报错 return res #test函数有返回值就返回,没有就返回None return bold
以上的代码就完成了python万能装饰器的写法,(*args,**kwargs)可以接收任意个数任意类型的参数,有没有返回值都可以return。修改print这句话为自己想要的功能,就是自己的万能装饰器了。这样一来,就可以装饰任意的函数了。
测试一下上面的代码:
@decorate #装饰一个没有任何参数没有返回值的方法 def test1(): print("test1:没有参数,没有返回值") @decorate #装饰一个有一个参数的方法 def test2(name): print("test2:带参数%s"%name) @decorate #装饰一个有返回值的方法 def test3(): strs = "test3:带返回值" return strs test1()
#结果为:
#style="font-weight:blod" #test1:没有参数,没有返回值 test2("ss")
#结果为:
#style="font-weight:blod" #test2:带参数ss a = test3() print(a)
#结果为:
#style="font-weight:blod" #test3:带返回值
每天进步一点点~