python基础之函数进阶之函数作为返回值/装饰器

mi0e

April 6, 2021

1205 views

No comments

2594 words

Python

**因为装饰器需要用到返回函数的知识，所以在这里将返回函数和装饰器合并讲解。**
**什么是返回函数？**

我们知道，一个函数中return可以返回一个或者多个值，但其实，return不仅可以返回值，还可以返回函数。

实例：

def col(*arg):
 def sum():
 res_sum=0
 for i in arg:
 res_sum=res_sum+i
 return res_sum
 return sum()
 a=col(1,2,3,4,5)
 print(a)#结果：15
 #a=col(1,2,3,4,5) ==〉
 #即a=sum,并且arg=(1,2,3,4,5)也被传入sum中了
 #a()==sum()
并且因为sum()定义在col()函数中，所以sum()继承了col()函数的局部变量和参数，这就是闭包。（比如，col()的arg参数就被sum()所继承）

下面来看一个我在检验上面加红的句话时所碰到的一个问题：

#还是用上面这段代码。稍稍修改一下
    def col(*arg):
        res_sum=0         #注意：将这句话移动到这里了
        def sum():
            for i in arg:
                res_sum=res_sum+i
            return res_sum
        return sum()
    a=col(1,2,3,4,5)
    print(a())       
    
    结果报错：local variable 'res_sum' referenced before assignment
    #既然内部函数可以引用外部函数的变量，为什么res_sum没有被内#部函数所引用？
为啥会报错？我当时很是疑惑，后来终于弄明白，错在这句：res_sum=res_sum+i

这句导致内部函数修改了外部函数的局部变量res_sum，这时，Python认为res_sum是内部函数的局部变量，而res_sum=res_sum+i之前并没有事先定义res_sum，所以当然会发生这种错误。

所以，需要这样进行修改：

def col(*arg):
        res_sum=0         
        def sum():
            for i in arg:
                a=res_sum+i     #在内部函数中定义一个新的变量a
            return a
        return sum
    a=col(1,2,3,4,5)
    print(a())

----------
**什么是装饰器？**

def login():
        pass
    @login       #关键字@
    def open():
        pass
上例就是一个装饰器的例子。

**装饰器的作用：**
当我们定义了open函数后（可以帮助我们打开某个文件），过了一段时间，发现我们需要在进行open之前进行验证用户，如何在不修改open函数的条件下将login函数和open函数进行结合，即：运行open之前先运行login？ 这里就要用到装饰器。

下面我们来看一个真正的装饰器的实例：

def login(fun):
        def real_login():
            print('please input your password')
            return fun()
        return real_login
    @login
    def open():
        print('hello world')
    open()
@login相当于open=login(open)   也就是将open函数偷梁换柱了一番。

当我们运行open()时：

open=login(open)　　#login(open)中的参数及局部变量：fun=open(原)　　#open=login(open)的返回值！
    　　#即：open=real_login　　
    open()=real_login()
    　　#因为real_login定义在login之中，所以继承其参数和局部变量
    　　#所以整体来看open()==login(open)+real_login()+open()

实现一个可以接受任意参数的装饰器：

def outer(func):
        def inner(*args,**kwargs):
            print('start')
            r=func(*args,**kwargs)    # 这里func(*args,**kwargs)相当于f(a,b)
            print('end')
            return r
        return inner
    
    @outer
    def f(a,b):
        print(a+b)
    @outer
    def f2(a,b,c):
        print(a+b+c)
    
    f(1,2)
    f2(1,2,3)
    #结果：
    start
    3
    end
    start
    6
    end

Last modification：April 6, 2021