[python] 帶有參數並且傳遞參數的裝飾器

標籤：python   裝飾   decorator   參數   傳遞   

情境時這樣的，我有個一大堆任務，我要給這些任務計時，入庫。就需要一個帶有參數的裝飾器來記錄任務名稱, 在任務執行前和執行之後都需要記錄任務當時執行的時刻。

#-*- encoding=utf-8 -*-from functools import wrapsdef task_logging(taskname):    def func_wrapper(func):        @wraps(func)        def return_wrapper(*args, **wkargs):            # 函數通過裝飾起參數給裝飾器傳送參數            print ‘before task‘,taskname            # 裝飾器傳變數給函數            taskid = 1            summer, funcres = func(taskid, *args, **wkargs)            print ‘after task‘, taskid, summer             return funcres        return return_wrapper    return func_wrapper@task_logging("test")def testd(taskid):    print "testd runing",taskid    return "task summer success eg", []print testd()

執行的結果

[email protected]|snmp # python decorators_1.pybefore task testtestd runing 1after task 1 task summer success eg[]

把中間的print 替換成需要執行的動作就可以完成，響應的功能了，參數在裝飾器和被裝飾函數之間可以很好的傳遞。

[python] 帶有參數並且傳遞參數的裝飾器