python学习--装饰器

闭包函数

• 函数内创建一个函数，这种函数只可以在外部函数的作用域内被正常调用，在外部函数的作用域之外调用会报错。
• 如果内部函数里引用了外部函数里定义的对象，那么此时内部函数就被称为闭包函数。
• 闭包可以将其自己的代码和作用域以及外部函数的作用结合在一起。

def count():
    a = 1
    b = 1
    def sum():
        c = 1
        return a + c
    return sum
a = count()
print(a())

装饰器

python装饰器本质上就是一个函数，它可以让其他函数在不需要做任何代码变动的前提下增加额外的功能，装饰器的返回值也是一个函数对象（函数的指针）

• 实质： 是一个函数
• 参数：是你要装饰的函数名（并非函数调用）
• 返回：是装饰完的函数名（也非函数调用）
• 作用：为已经存在的对象添加额外的功能
• 特点：不需要对对象做任何的代码上的变动

案例
比如有一个方法

def ma():
    print('马')

新增一个牵走马的功能

def ma():
    print('马')
    print('牵走马')

此时，鸡和羊也要这个新功能，我们可以写一个可以共用的方法

def qian(func):
    func()
    print('牵走', func.__name__)
def yang():
    print('羊')
qian(yang)

这个方法已经破坏了原有的代码逻辑结构
此时我们使用装饰器

无参数装饰器

def qian(func):
    def wrapper():
        func()
        print('牵走:',func.__name__)
    return wrapper
def yang():
    print('羊')
qy = qian(yang)
qy()

qian就是装饰器，把真正的函数包裹起来
可以使用@使用装饰器

def qian(func):#装饰器的定义#自身不传参数
    def wrapper():
        func()
        print('牵走:',func.__name__)
    return wrapper

@qian
def yang():
    print('羊')

@qian
def ma():
    print('马')
yang()
ma()

带参数装饰器

对装饰器进行装饰，给内部传入值

def qian(person):#装饰器的定义#传参数
    def dec(func):
        def wrapper():
            func()
            print(person,'牵走:',func.__name__)
        return wrapper
    return dec

@qian(person='lily')
def yang():
    print('羊')

@qian('二柱')
def ma():
    print('马')
yang()
ma()

类装饰器

当用@使用装饰器时，会调用__call__方法

class qian(object):
    def __init__(self,func):
        self.func = func

    def __call__(self):
        self.func()
        print('牵走：',self.func.__name__)

@qian
def yang():
    print('羊')
@qian
def ma():
    print('马')
yang()
ma()

wraps装饰器

上述装饰器中

yang()#等价于 ma = qian(ma),改变了函数名称
ma()
print(yang.__name__) #wrapper

wraps装饰器可以把函数的元信息拷贝到装饰器函数中

from functools import wraps
def qian(func):
    @wraps(func)
    def wrapper():
        func()
        print('牵走:',func.__name__)
    return wrapper

@qian
def yang():
    print('羊')
@qian
def ma():
    print('马')
yang()
ma()
print(yang.__name__)#yang