函数装饰器

一切皆对象

• 在Python中，变量的本质都是对象，字符串变量是一个str类型的对象，而函数同样是一个类型为 function的变量。

def fun():
 def fun():
    print('函数fun')


fun()
#函数fun

#我们可以将一个函数赋值给一个变量
f = fun

f()
#函数fun

#它的本质是一个类型为function的对象
print(type(f))
#<class 'function'>

在函数中再定义一个函数

def fun1():
    print('这里是fun1')

    def fun2():
        print('这里是fun2')

    fun2()


fun1()
# 这里是fun1
# 这里是fun2

fun2()
# 这里调用fun2会报 NameError 的错误 

# 这里说明fun2只能在函数中的调用，不能在函数外被调用。

从函数中返回一个函数

• 在上面例子中我们得知，嵌套函数（在函数中定义的函数）只能在定义它的函数体中被调用。那么有没有办法可以在函数的外部执行调用呢？

def fun1():
    def fun2():
        print('这里是fun2')
    #将嵌套函数作为函数的返回值
    return fun2


# 我们可以通过调用fun1函数，获得fun2函数的对象
f = fun1()

f()
# 这里是fun2
# fun1()() 与上述调用等同

将函数作为参数传入另一个函数

def fun1(fun):
    print('在执行fun函数前做些事情')
    # 调用传入的函数fun
    fun()


def fun2():
    print('fun2函数执行中')


# 将fun2函数作为参数传入fun1函数
fun1(fun2)
# 在执行fun函数前做些事情
# fun2函数执行中

编写第一个装饰器

# 定义一个装饰器
def decorator(fun):
    def wrapper():
        print("在执行 fun函数 前做一些事")
        fun()
        print("在执行 fun函数 后做一些事")
    # 返回被装饰后的 fun函数
    return wrapper


def hello():
    print('hello python')


hello()
# hello python

# 使用 decorator函数 装饰 hello函数
hello_with_decorator = decorator(hello)

hello_with_decorator()
# 在执行 fun函数 前做一些事
# hello python
# 在执行 fun函数 后做一些事

• 糖语法：在上面的例子中，装饰器的使用过于繁琐，不利于阅读。为了更贴近装饰器的作用（在不改变函数体的前提下，对函数的功能进行扩展），我们通过糖语法来使用装饰器。

# 定义一个装饰器
def decorator(fun):
    def wrapper():
        print("在执行 fun函数 前做一些事")
        fun()
        print("在执行 fun函数 后做一些事")
    # 返回被装饰后的 fun函数
    return wrapper

# 糖语法
@decorator
def hello():
    print('hello python')

hello()
# 在执行 fun函数 前做一些事
# hello python
# 在执行 fun函数 后做一些事

装饰带参数的函数

• 在上面的例子中装饰器所装饰的函数是没有参数的，而在大多数情况下函数是否有参数是不确定的，为了满足普适性，下面继续完善我们的装饰器。

# 定义一个装饰器
def decorator(fun):
    # 使用不定长参数
    def wrapper(*args, **kwargs):
        print("在执行 fun函数 前做一些事")
        fun(*args, **kwargs)
        print("在执行 fun函数 后做一些事")
    # 返回被装饰后的 fun函数
    return wrapper

# 糖语法
@decorator
def hello(content: str):
    print(content)


hello("hello python")
# 在执行 fun函数 前做一些事
# hello python
# 在执行 fun函数 后做一些事

带参数的装饰器

• 在一些比较复杂的应用场景中，为了满足开发需求，装饰器需要传入一些参数，那如何定义带参数的装饰器呢？

# 带参数的装饰器
def tag(name: str='<a>'):
    def decorator(f):
        # 使用不定长参数
        def wrapper(*args, **kwargs):
            content = f(*args, **kwargs)
            return '{tag}{content}{tag}'.format(tag=name, content=content)
        # 返回被装饰后的 fun函数
        return wrapper
    return decorator


@tag('<a>')
def hello(content: str = 'hello python'):
    return content


print(hello("hello python"))
# <a>hello python<a>