前言

如何給裝飾器的參數(shù)傳參，這個問題曾經(jīng)困擾我好久，雖然Python版本的更新，現(xiàn)在這個問題終于解決了，特此記錄。

[[346777]]

疑問

首先我有一個這樣的裝飾器文件路徑helper/log_helper.py

import traceback 
from functools import wraps 
 
from loguru import logger 
 
 
def my_logger(count): 
    def step1(foo): 
        @wraps(foo) 
        def step2(*args, **kwargs): 
            try: 
                result = foo(*args, **kwargs) 
                logger.info(f"{result=},{count=}") 
            except Exception: 
                logger.exception(traceback.format_exc()) 
 
        return step2 
 
    return step1 

然后我有個文件需要引用這個裝飾器demo.py

from helper.log_helper import my_logger 
 
 
class Demo: 
    @my_logger(count=2) 
    def main(self): 
        return "in main function" 
 
 
if __name__ == '__main__': 
    d = Demo() 
    d.main() 

輸出結(jié)果如下

2020-10-16 11:43:12.001 | INFO     | helper.log_helper:step2:18 - result='in main function',count=2 

這個裝飾器的作用很簡單，就是獲取當(dāng)前函數(shù)的返回值，和傳入的count值。

好，現(xiàn)在問題來了?

如果給裝飾器的參數(shù)傳值呢，也就是說我的count=2，是通過傳值的形式。你想到可能是這樣

from helper.log_helper import my_logger 
 
COUNT=2 
class Demo: 
    @my_logger(count=COUNT) 
    def main(self): 
        return "in main function" 
 
 
if __name__ == '__main__': 
    d = Demo() 
    d.main() 

ok，這樣確實可以，我們還可以使用再簡化一步

from functools import partial 
from helper.log_helper import my_logger 
 
COUNT=2 
my_logger = partial(my_logger,count=2) 
 
 
class Demo: 
    @my_logger() 
    def main(self): 
        return "in main function" 
 
 
if __name__ == '__main__': 
    d = Demo() 
    d.main() 

暫時來看我們搞定了傳參數(shù)的問題，這時候我們想如果外界調(diào)用了Demo類的main方法，并且向指定count的值怎么辦呢?

我們知道外界調(diào)用Demo類傳參的唯一途徑就是向__init__里進(jìn)行傳參數(shù)，按照這個思路我們只能這么寫了，

class Demo: 
    def __init__(self): 
        count =2 
    @my_logger(count=self.count) 
    def main(self): 
        return "in main function" 

但是這樣并不可以，我們得到錯誤信息

NameError: name 'self' is not defined 

在裝飾器中無法使用self.形式的參數(shù),難道這個問題解決不了么?

問題解決

在Python3.7之前確實沒什么可行的方案。

我們知道在Python3.7的時候引入了dataclasses，我們可以通過它來簡化__init__。

改下我們的代碼

from functools import partial 
 
from helper.log_helper import my_logger 
from dataclasses import dataclass 
 
@dataclass() 
class Demo: 
    count: int = 2 
    logger: my_logger = partial(my_logger, count) 
 
    @logger() 
    def main(self): 
        return "in main function" 
 
 
if __name__ == '__main__': 
    d = Demo() 
    d.main() 

如果使用Python3.8那么可以直接忽略掉dataclass

class Demo: 
    count: int = 2 
    logger: my_logger = partial(my_logger, count) 
 
    @logger() 
    def main(self): 
        return "in main function" 

這樣我們就成功的解決了這個問題，突然想起來之前遇到的這個難題，現(xiàn)在算是解決了，希望對你有幫助。