[已解决]含参函数如何使用装饰器

九歌当下

我先定义一个闭包

1. def logger(func):
   
2.     def call_func():
   
3.         print("----------")
   
4.         func()
   
5.         print("----------")
   
6.     return call_func

复制代码

再给一个含参函数加上该闭包的装饰器

1. @logger
   
2. def sayhi(name):
   
3.     print("hi,{}".format(name))

复制代码

然后函数sayhi()就没法用了

1. sayhi("小甲鱼")
   
2. Traceback (most recent call last):
   
3.   File "<pyshell#38>", line 1, in <module>
   
4.     sayhi("小甲鱼")
   
5. TypeError: logger.<locals>.call_func() takes 0 positional arguments but 1 was given

复制代码

求帮助

最佳答案

月排行榜 / 总排行榜

isdkz

2023-3-5 14:10:07

一个函数在使用装饰器后会被替换成装饰器的返回值，

而装饰器的返回值 call_func 不接受任何参数，故而你给 sayhi 传参会报错，

所以就是要让 call_func 接收跟 sayhi 的参数个数一样的形参，并在 call_func 调用 sayhi 的时候把参数传递给 sayhi

故对你的代码修改如下：

1. def logger(func):
   
2.     def call_func(name):                         # 改了这里
   
3.         print("----------")
   
4.         func(name)                         # 改了这里
   
5.         print("----------")
   
6.     return call_func
   
7. 
   
8. @logger
   
9. def sayhi(name):
   
10.     print("hi,{}".format(name))
    
11. 
    
12. sayhi("小甲鱼")

复制代码

跳转到最佳答案楼层

isdkz

一个函数在使用装饰器后会被替换成装饰器的返回值，

而装饰器的返回值 call_func 不接受任何参数，故而你给 sayhi 传参会报错，

所以就是要让 call_func 接收跟 sayhi 的参数个数一样的形参，并在 call_func 调用 sayhi 的时候把参数传递给 sayhi

故对你的代码修改如下：

1. def logger(func):
   
2.     def call_func(name):                         # 改了这里
   
3.         print("----------")
   
4.         func(name)                         # 改了这里
   
5.         print("----------")
   
6.     return call_func
   
7. 
   
8. @logger
   
9. def sayhi(name):
   
10.     print("hi,{}".format(name))
    
11. 
    
12. sayhi("小甲鱼")

复制代码

九歌当下

isdkz 发表于 2023-3-5 14:10
一个函数在使用装饰器后会被替换成装饰器的返回值，

而装饰器的返回值 call_func 不接受任何参数，故而 ...

OK， 谢谢了

歌者文明清理员

1. def logger(func):
   
2.     def call_func(**args, **kwargs):                         # 改了这里
   
3.         print("----------")
   
4.         func(**args, **kwargs)                         # 改了这里
   
5.         print("----------")
   
6.     return call_func
   
7. 
   
8. @logger
   
9. def sayhi(name):
   
10.     print("hi,{}".format(name))
    
11. 
    
12. sayhi("小甲鱼")

复制代码

这样就兼容所有函数了

九歌当下

歌者文明清理员 发表于 2023-3-5 15:05
这样就兼容所有函数了

真是太厉害了，没有想到收集参数能这么用

歌者文明清理员

九歌当下 发表于 2023-3-5 16:33
真是太厉害了，没有想到收集参数能这么用

可惜了我的最佳答案

九歌当下

歌者文明清理员 发表于 2023-3-5 16:40
可惜了我的最佳答案

下次看了得晚点在设最佳答案