Python中的闭包与装饰器

闭包是什么

在Python中,闭包Closure可以理解为函数以及函数所处的环境的一个整体。

下面的例子展示了一个最简单的闭包的示例:

def closure():
    a = 1

    def func(x):
        return a + x
    return func


f1 = closure()
f2 = closure()
print(f1(1))    # 结果为2
print(f1(2))    # 结果为3

但对内部函数稍作修改,尝试改变变量a的值时

def func(x):
    a += 1
    return a + x
return func

会发生如下错误
UnboundLocalError: local variable 'a' referenced before assignment,原因在于变量a为不可变类型,已经被引用了,python解释器会阻止这样的操作。

那么,尝试使用可变类型,再看看结果如何。

def closure():
    a = []

    def func(x):
        a.append(x)
        return a
    return func


f1 = closure()
f2 = closure()
print(f1(1))    # 结果为[1]
print(f1(2))    # 结果为[1, 2]
print(f2(3))    # 结果为[3]
print(f2(4))    # 结果为[3, 4]

内部函数可以成功修改闭包区域的变量。

此外,创建了两个闭包的实例,通过结果可以看到,同一实例中闭包中的变量是共享的,而不同的闭包实例是不会相互影响的。

如果闭包区域中的变量为不可变类型,又想去改变它,可以使用nonlocal修饰该变量。

def closure():
    a = 1

    def func(x):
        nonlocal a
        a += x
        return a
    return func

装饰器

闭包最重要的应用,其实就是装饰器了。

最简单的装饰器

定义一个闭包函数,此函数接收一个参数,传入另一个函数名,在内部函数中调用传入的函数

def decorator(f):
    def func():
        print('decorator start...')
        f()
        print('decorator end...')

    return func


def test():
    print('test...')


if __name__ == '__main__':
    my_test = decorator(test)
    my_test()

这里显式地将test函数传入闭包函数decorator中,得到一个新的函数(实际上就是闭包函数中的内部函数func)。

而Python提供了语法糖,使用@decorator来修饰目标函数,使装饰器的使用更为方便。

def decorator(f):
    def func():
        print('decorator start...')
        f()
        print('decorator end...')

    return func


@decorator
def test():
    print('test...')


if __name__ == '__main__':
    test()

多个装饰器的执行顺序

def decorator1(f):
    def func():
        print('decorator 1 start...')
        f()
        print('decorator 1 end...')

    return func


def decorator2(f):
    def func():
        print('decorator 2 start...')
        f()
        print('decorator 2 end...')

    return func


@decorator1
@decorator2
def test():
    print('test...')


if __name__ == '__main__':
    test()

执行结果如下

decorator 1 start...
decorator 2 start...
test...
decorator 2 end...
decorator 1 end...

从上面的结果看,可以理解为函数test首先被decorator2装饰后得到了一个中间函数,这个中间函数又被decorator1再次装饰。

目标函数(被装饰的函数)带参数

如果被装饰的函数带有参数,只需要在装饰器内部函数上添加相同个数参数即可。

def decorator(f):
    def func(a, b):
        print('decorator start...')
        f(a, b)
        print('decorator end...')

    return func


@decorator
def test(a, b):
    print('test...', a, b)


if __name__ == '__main__':
    test(123, 456)

当然,除此以外,Python中的可变参数和关键字参数的特性能够使得这种场景变得更加通用。

对上面的代码稍作修改,这样目标函数不管有多少个参数都能在不修改装饰器函数的情况下支持了。

def decorator(f):
    def func(*args, **kwargs):
        print('decorator start...')
        f(*args, **kwargs)
        print('decorator end...')

    return func


@decorator
def test(a, b, c):
    print('test...', a, b, c)


if __name__ == '__main__':
    test(123, 456, 789)

装饰器带参数

如果装饰器函数本身带有参数,该如何处理呢?

原来装饰器函数的参数为目标函数的函数名,现在装饰器函数本身有参数,那只需要将内部函数再包一层,用两次闭包就能解决问题。

def decorator(x, y):
    def wrapper(f):
        def func(*args, **kwargs):
            print('装饰器参数:', x, y)
            print('decorator start...')
            f(*args, **kwargs)
            print('decorator end...')
        return func
    return wrapper


@decorator(x=11, y=22)
def test(a, b):
    print('test...', a, b)


if __name__ == '__main__':
    test(33, 44)

通用装饰器

同样地,如果装饰器的参数不确定,那么使用可变参数及关键字参数,就能使的装饰器满足所有场景了。

def decorator(*ar, **kw):
    def wrapper(f):
        def func(*args, **kwargs):
            print("装饰器参数:", ar, kw)
            print('decorator start...')
            f(*args, **kwargs)
            print('decorator end...')
        return func
    return wrapper


@decorator(x='11', y='22')
def test(a, b):
    print('test...', a, b)


if __name__ == '__main__':
    test(33, 44)

类装饰器

前面的装饰器都是在说装饰器函数,其实类也作为装饰器,但前提是类中定义了call魔法函数,因为装饰器约定:接收一个callable对象,返回一个callable对象。

class Test:
    def __init__(self, f):
        print('Test inited..')
        self.f = f

    def __call__(self, *args, **kwargs):
        print('Test called...')
        self.f(*args, **kwargs)


@Test
def test(a, b):
    print('test...', a, b)


if __name__ == '__main__':
    test(1, 2)
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,189评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,577评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,857评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,703评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,705评论 5 366
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,620评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,995评论 3 396
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,656评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,898评论 1 298
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,639评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,720评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,395评论 4 319
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,982评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,953评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,195评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 44,907评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,472评论 2 342

推荐阅读更多精彩内容