2.带参装饰器

在上面的装饰器调用中，比如@decorator_demo
该装饰器默认它后面的函数是唯一的参数
装饰器的语法允许我们调用decorator时，提供其他参数
比如@decorator(a)，这样，就为装饰器的编写和使用提供了更大的灵活性

#带参装饰器

def pre_str(pre=""):
    def decorator(old_function):
        def new_function(a,b):
            print(pre + "input",a,b)
            return old_function(a,b)
        return new_function
    return decorator
    
#装饰square_sum()
@pre_str("^_^")
def square_sum(a,b):
    return a**2 + b**2   
    
#装饰square_diff()
@pre_str("T_T")
def square_diff(a,b):
    return a**2 - b**2

if __name__ == '__main__':
    print(square_sum(3,4))
    print(square_diff(3,4))

^_^input 3 4
25
T_Tinput 3 4
-7

上面的pre_str是一个带参装饰器，它实际上是对原有装饰器的一个函数封装
并返回一个装饰器，我们可以将它理解为一个含有环境参量的闭包
当我们使用@pre_str(“^_“)调用的时候，py能够发现这一层的封装
并把参数传递到装饰器的环境中，该调用相当于：

square_sum = pre_str("^_^")(square_sum)

根据参数不同，带参装饰器会对函数进行不同的加工，进一步提高了装饰器的适用范围
还是以网站的用户验证为例子，装饰器负责验证的功能，装饰了处理HTTP请求的函数
可能有的关键HTTP请求需要管理员权限，有的只需要普通用户权限
因此，我们可以把“管理员”和“用户”作为参数，传递给验证装饰器
对于那些负责关键HTTP请求的函数，我们可以把“管理员”参数传给装饰器
对于负责普通HTTP请求的函数，我们可以把“用户”参数传给它们的装饰器
这样，同一个装饰器就可以满足不同的需求了

3.装饰类

在上面的例子中，装饰器接收一个函数，并返回一个函数
从而起到加工函数的效果，装饰器还拓展到了类
一个装饰器可以接收一个类，并返回一个类，从而起到加工类的效果

def decorator_class(SomeClass):
    class NewClass(object):
        def __init__(self,age):
            self.total_display = 0
            self.wrapped = SomeClass(age)
        def display(self):
            self.total_display += 1
            print('total display',self.total_display)
            self.wrapped.display()
    return NewClass

@decorator_class
class Bird:
    def __init__(self,age):
        self.age = age
    def display(self):
        print('my age is',self.age)
        
if __name__ == '__main__':
    eagle_lord = Bird(5)
    for i in range(3):
        eagle_lord.display()

total display 1
my age is 5
total display 2
my age is 5
total display 3
my age is 5

在装饰器decorator_class中，我们返回了一个新类NewClass
在新类的构造器中，我们用一个属性self.wrapped记录了原来类生成的对象
并附加了新的属性total_display，用于记录调用display()的次数
我们也同时更改了display方法，通过装饰
我们的Bird类可以调用display()的次数
无论是装饰函数，还是装饰类，装饰器的核心作用都是名称绑定
虽然装饰器出现较晚，但在各个py项目中的使用却很广泛
即便不需要自定义装饰器，你也很有可能会在自己的项目中调用其他库中的装饰器
因此，py程序员需要掌握这一语法