【Python】闭包 | 装饰器

闭包

在函数嵌套的前提下，内部函数使用到了外部函数的变量，并且外部函数返回了内部函数，我们把这个使用外部函数变量的内部函数称为闭包。

# 闭包
def account_init(init_amount=0):
    def atm(num, flag=True):
        nonlocal init_amount
        if flag:
            init_amount += num
            print(f"存款：+{num}，账户余额：{init_amount}")
        else:
            init_amount -= num
            print(f"取款：-{num}，账户余额：{init_amount}")

    return atm


fn = account_init()
fn(300)                 # 存款：+300，账户余额：300
fn(200)                 # 存款：+200，账户余额：500
fn(100, False)            # 取款：-100，账户余额：400

需要使用nonlocal关键字修饰外部函数的变量才可在内部函数中修改它。

闭包注意事项

优点

• 无需定义全局变量即可实现通过函数，持续的访问、修改某个值
• 闭包使用的变量所用于在函数内，难以被错误的调用修改

缺点：

• 由于内部函数持续引用外部函数的值，所以会导致一部分内存空间不被释放，一直占用内存

装饰器

装饰器其实也是一种闭包，其功能就是在不破坏目标函数原有的代码和功能的前提下，为目标函数增加新功能。

装饰器的一般写法（闭包写法）

定义一个闭包函数，在闭包函数内部：

• 执行目标函数
• 并完成功能的添加

# 装饰器的一般写法（闭包写法）
def outer(func):
    def inner():
        print('开始睡觉')
        func()
        print('开始起床')

    return inner


def sleep():
    import random
    import time
    print('睡眠中。。。')
    time.sleep(random.randint(1, 5))


fn = outer(sleep)
fn()

# -----------执行结果-----------
# 开始睡觉
# 睡眠中。。。
# 开始起床

装饰器的快捷写法（语法糖）

使用@outer定义在目标函数之上。

# 装饰器的快捷写法（语法糖）
def outer(func):
    def inner():
        print('开始睡觉')
        func()
        print('开始起床')

    return inner


@outer
def sleep():
    import random
    import time
    print('睡眠中。。。')
    time.sleep(random.randint(1, 5))


sleep()

# -----------执行结果-----------
# 开始睡觉
# 睡眠中。。。
# 开始起床