装饰器

装饰器原理剖析

装饰器的作用：在完全不改变函数的前提下，通过装饰的该函数的方法下实现在该函数的前面或者后面新增语句

装饰器的原理：

1、将被装饰的函数当做参数传递给你装饰器函数

2、将装饰器函数的返回值赋值给装饰器函数

装饰器1：仅仅实现最基本的功能

def outer(func):
    def inner():
        print("before")
        func()
        print("after")
    return inner
 
@outer
def f1():
    print("Hi,Everybody.")
 
f1()

上面就是最基本的装饰器的函数，outer函数为装饰器，f1为被装饰的函数；outer装饰函数f1，此时outer中的参数为args，此时args就是f1函数的内存地址，inner函数的内存地址就现在的，此时执行f1函数就是执行inner函数，而inner函数中就可以在旧的f1函数前后增加代码

装饰器2：处理被装饰的函数的返回值

如果原函数有返回值，怎么办呢？其实也很简单，因为上在上面的例子中，func就是被装饰的函数，也就是f1

def outer(func):
    def inner():
        print("before")
        ret = func()
        print("after")
        return ret
    return inner
 
@outer
def f1():
    print("Hi,Everybody.")
    return "这个是被装饰的函数的返回值"
 
ret = f1()
print(ret)

装饰器3：处理被装饰器的函数的参数

如果原函数有参数，该怎么办，其实这个也很简单，直接用万能参数即可

def outer(func):
    def inner(*args,**kwargs):
        print("before")
        ret = func(*args,**kwargs)
        print("after")
        return ret
    return inner
 
@outer
def f1(a):
    print(a)
    return "这个是被装饰的函数1的返回值"
 
 
@outer
def f2(a,b):
    print(a)
    print(b)
    return "这个是被装饰的函数2的返回值"
ret = f1("这个是被装饰的函数1的参数")
print(ret)
 
ret = f2("这个是被装饰的函数2的参数1","这个是被装饰的函数2的参数2")
print(ret)

解决查看函数信息的方法失效的问题

from functools import wraps

def deco(func):
    @wraps(func) #加在最内层函数正上方
    def wrapper(*args,**kwargs):
        return func(*args,**kwargs)
    return wrapper

@deco
def index():
    '''哈哈哈哈'''
    print('from index')

print(index.__doc__)
print(index.__name__)