[已解决]装饰器问题

dami27

想问一下为什么下面第一种情况是OK的，第二种就会报错呢

第一种

1. class Logging:
   
2.     def __init__(self, func):
   
3.         self.func = func
   
4.     def __call__(self):
   
5.         print(f"{self.func.__name__}()")
   
6.         self.func()
   
7. 
   
8.         
   
9. @Logging
   
10. def say():
    
11.     print("hello")
    
12. 
    
13. say()

复制代码

1. say()
   
2. hello

复制代码

第二种

1. class Logging:
   
2.     def __init__(self, func, level="INFO"):
   
3.         self.func = func
   
4.         self.level = level
   
5.     def __call__(self):
   
6.         print(f"[{self.level}: {self.func}()]")
   
7.         self.func()
   
8. 
   
9.         
   
10. @Logging(level="ERROR")
    
11. def say():
    
12.     print("hello")
    
13. 
    
14. say()

复制代码

1. Traceback (most recent call last):
   
2.   File "C:\Users\JiJing\Desktop\test.py", line 10, in <module>
   
3.     @Logging(level="ERROR")
   
4. TypeError: Logging.__init__() missing 1 required positional argument: 'func'

复制代码

最佳答案

月排行榜 / 总排行榜

tommyyu

2022-11-10 09:47:21

dami27 发表于 2022-11-10 09:44
此时因为 level 参数和 func 参数分批传入，python 就会报错。
：这句话不太理解，什么叫分批传入，是le ...

在初始化的时候，应该 Logging(func, level) 这样传参，但是因为装饰器的语法原因，你原来的那个

1. @Logging(level="ERROR")
   
2. def say():
   
3.     print("hello")

复制代码

这个函数的定义被 Python 解析成了

1. def say():
   
2.     print("hello")
   
3. say = Logging(level = 'ERROR')(say)

复制代码

此时 Python 就会报错。

跳转到最佳答案楼层

tommyyu

原来的代码相当于 say = Logging(level = "ERROR")(say)，此时因为 level 参数和 func 参数分批传入，python 就会报错。如果想要实现这种效果，可以这样写

1. class Logging:
   
2.     def __init__(self, func, level="INFO"):
   
3.         self.func = func
   
4.         self.level = level
   
5.     def __call__(self):
   
6.         print(f"[{self.level}: {self.func}()]")
   
7.         self.func()
   
8. 
   
9. def temp(level):
   
10.     def inner(func):
    
11.         nonlocal level
    
12.         return Logging(func, level)
    
13.     return inner
    
14. 
    
15. @temp(level = 'INFO')
    
16. def say():
    
17.     print('hello')
    
18. 
    
19. say()

复制代码

jackz007

1. class Logging:
   
2.     def __init__(self , level = "INFO"):
   
3.         self . level = level
   
4. 
   
5.     def __call__(self , func_name):
   
6.         def wrapper(* args , ** kwargs):
   
7.             print(self . level)
   
8.             return func_name(* args , ** kwargs)
   
9.         return wrapper
   
10. 
    
11. @Logging("ERROR")
    
12. def say():
    
13.     print("hello")
    
14. 
    
15. say()

复制代码

dami27

tommyyu 发表于 2022-11-9 16:14
原来的代码相当于 say = Logging(level = "ERROR")(say)，此时因为 level 参数和 func 参数分批传入，pytho ...

此时因为 level 参数和 func 参数分批传入，python 就会报错。
：这句话不太理解，什么叫分批传入，是level和func一起传入，python无法分辨的意思吗

tommyyu

dami27 发表于 2022-11-10 09:44
此时因为 level 参数和 func 参数分批传入，python 就会报错。
：这句话不太理解，什么叫分批传入，是le ...

在初始化的时候，应该 Logging(func, level) 这样传参，但是因为装饰器的语法原因，你原来的那个

1. @Logging(level="ERROR")
   
2. def say():
   
3.     print("hello")

复制代码

这个函数的定义被 Python 解析成了

1. def say():
   
2.     print("hello")
   
3. say = Logging(level = 'ERROR')(say)

复制代码

此时 Python 就会报错。

dami27

jackz007 发表于 2022-11-9 18:21

可是为啥这样也会报错？装饰器的函数引用不是不用特意传入的吗

1. class Logging:
   
2.     def __call__(self, func):
   
3.         def wrapper():
   
4.             print(f"{func.__name__}()")
   
5.             return func()
   
6.         return func
   
7. 
   
8.    
   
9. @Logging
   
10. def say():
    
11.     print("hello")

复制代码

1. Traceback (most recent call last):
   
2.   File "C:\Users\JiJing\Desktop\1.py", line 10, in <module>
   
3.     def say():
   
4. TypeError: Logging() takes no arguments

复制代码

jackz007

dami27 发表于 2022-11-10 09:47
可是为啥这样也会报错？装饰器的函数引用不是不用特意传入的吗

        装饰器类在带和不带参数时，参数的传递约定不同，你只要记住就可以了，不带参数时的被装饰函数是在 __init__() 时传入的。任何时候，装饰器类都不可以少了 __init__() 和 __call__()，一个也不能少！！！

1. class Logging:
   
2.     def __init__(self , func):
   
3.         self . func = func
   
4.     def __call__(self, * args , ** kwargs):
   
5.         print(f"{self . func . __name__}()")
   
6.         return func(* args , ** kwargs)
   
7. 
   
8. @Logging
   
9. def say():
   
10.     print("hello")

复制代码