python 带参数装饰器

在前面一文《python里为什么需要使用装饰器（decorator）》里，我们学习了为什么需要装饰器，知道装饰器就是为了不修改原来函数的代码，又达到增加功能的作用。其实为了装饰器更通用化，那么装饰器是否也可以带参数呢？其实是可以的，这样更加通用化了，达到共享极点。在前面也学习《为什么要使用闭包（closures）》一文，知道参数可以嵌套函数里实现隐藏，并且实现全局参数的功能，与函数一起绑定。因此只需要结合这两个知识点，就可以产生通用带参数的装饰器了。原来不带参数的代码如下：

[python] view plain copy

1. #python 3.6  
2. #定义一个新的函数  
3. def printStar(func):  
4.     def f():  
5.         print('*************************************')  
6.         return func()  
7.     return f  
8.  
9. @printStar  
10. def add():      
11.     return 1 + 1  
12.  
13. @printStar  
14. def sub():      
15.     return 2 -1  
16.       
17. print(add())  
18.   
19. print(sub())  

现在再对装饰器函数进一步修改，再嵌套一层函数，实现闭包的功能，代码就修改如下：

[python] view plain copy

1. #python 3.6  
2. #定义一个新的函数  
3. def title(show = ''):  
4.     def printStar(func):  
5.         def f():  
6.             print(show,'*************************************')  
7.             return func()  
8.         return f  
9.     return printStar  
10.  
11. @title('add')  
12. def add():      
13.     return 1 + 1  
14.  
15. @title('sub')  
16. def sub():      
17.     return 2 -1  
18.       
19. print(add())  
20.   
21. print(sub())  

输出如下：

add *************************************
2
sub *************************************
1

经过带参数的装饰器的修改，这时控制输出的内容，就更方便了。可以根据不同的函数名称来进行修改输出提示。

前面发现被装饰的函数都没有带参数，这是为了简单起见，现在来学习一下如果被装饰的函数也参数，怎么样修改呢？
可以看下面的代码：

[python] view plain copy

1. #python 3.6  
2. #定义一个新的函数  
3. def title(show = ''):  
4.     def printStar(func):  
5.         def f(a, b):  
6.             print(show,'*************************************')  
7.             return func(a, b)  
8.         return f  
9.     return printStar  
10.  
11. @title('add')  
12. def add(a, b):      
13.     return a + b  
14.  
15. @title('sub')  
16. def sub(a, b):      
17.     return a - b  
18.       
19. print(add(1, 1))  
20.   
21. print(sub(2, 1))  

由两段代码相比较，可以发现函数传送参数是修改def f(a, b)，就是第三层的函数。