管窥python闭包

什么是闭包?
来看看wiki的解释

在计算机科学中,闭包(英语:Closure),又称词法闭包(Lexical Closure)或函数闭包(function closures),是引用了自由变量的函数。这个被引用的自由变量将和这个函数一同存在,即使已经离开了创造它的环境也不例外。所以,有另一种说法认为闭包是由函数和与其相关的引用环境组合而成的实体。闭包在运行时可以有多个实例,不同的引用环境和相同的函数组合可以产生不同的实例。

这个解释比较学术话,我的理解是引用了函数体外部变量的函数,叫做函数闭包,举个栗子:

def func():
    a = 1
    def closure():
        print(a)
    return closure()   

对于func这个函数来说,closure就是一个闭包,因为closure引用了func函数体内部变量a,脱离了func谈论closure是没有意义的。

在python,函数闭包这一概念通过python装饰器落地,成为python的一大语法糖。
先来看一下使用实例

def decorator(func):
    print("this is decorator")
    return func
    
@decorator
def func():
    print("this is func")

输出

this is decorator
this is func

@+decorator(函数名)构成了装饰器,从结果来看decorator函数要先于func函数执行,从闭包的概念上来看,可以认为funcdecorator
的闭包,python将这种使用行为标准化为装饰器。这里有疑问,func并没有使用decorator的变量啊(这个示例里面decorator中并没有定义变量)?
别着急,我们先来看一下func带参数的情况。

def decorator(func):
    print("this is decorator")
    def wrapper(func_param):
        return func(func_param)
    return wrapper

@decorator
def func(a):
    print(a)

func("this is func param a")

输出

this is decorator
this is func param a

decorator带参数的情况又该如何?

def decorator(decorator_param):
    print(decorator_param)
    def wrapper(func):
        print("this is wrapper")
        def sub_wrapper(*args, **kwargs):
            print("this is sub_wrapper")
            return func(*args, **kwargs)
        return sub_wrapper
    return wrapper

@decorator("this is decorator param")
def func(a):
    print(a)

func("this is func param")

输出

this is decorator param
this is wrapper
this is sub_wrapper
this is func param

可以看到,为了传递参数,需要层层闭包来实现,写法上是变得更加复杂,但是带来的好处也是很可观的:减少了重复代码

装饰器能做什么?
以上实例无一例外反映了装饰器(如上例中的decorator函数)会先于被装饰函数(也就是上例中的的func函数)执行
这有点类似于中间件的概念,比如你想在每个函数执行前打印一下当前时间,写一个装饰器就可以很容易实现,就像下面这样

from datetime import datetime

def decorator(decorator_param):
    print(datetime.now().strftime('%Y-%m-%d %H:%M:%S'))
    def wrapper(func):
        print("this is wrapper")
        def sub_wrapper(*args, **kwargs):
            print("this is sub_wrapper")
            return func(*args, **kwargs)
        return sub_wrapper
    return wrapper

@decorator("this is decorator param")
def func(a):
    print(a)

func("this is func param")

so easy!

ps:
在golang中,defer函数会在当前函数退出前执行,正好和python中的装饰器功能相对应。可以对比理解。

参考:

  • 闭包
  • 装饰器

你可能感兴趣的:(管窥python闭包)