首页 > > 网络编程 > 其它 >

python装饰器探究与参数的领取

2018-06-17 23:42:26来源：未知阅读 ()

首先上原文,

现在，假设我们要增强now()函数的功能，比如，在函数调用前后自动打印日志，但又不希望修改now()函数的定义，这种在代码运行期间动态增加功能的方式，称之为“装饰器”（Decorator）。
本质上，decorator就是一个返回函数的高阶函数。

Decorator本质是高阶函数?

不信邪的我试了下..

 1 def g():
 2     print("这里是G")
 3     return "G"
 4 
 5 @g
 6 def f():
 7     print("这里是F")
 8     return 1
 9 '''
10 --------------------------------------------
11 line 5, in <module>
12     @g
13 TypeError: g() takes 0 positional arguments but 1 was given
14 >>>
15 '''

运行结果在注释里

尴尬了...g被强制塞了个参数,那个参数应该是g"修饰的对象"

修改下继续.....

def g(f):
    print("这里是G")
    return "G"

@g
def f():
    print("这里是F")
    return 1
'''
--------------------------------------------
这里是G
>>> f()
Traceback (most recent call last):
  File "<pyshell#0>", line 1, in <module>
    f()
TypeError: 'str' object is not callable
'''

str对象不能被调用,,,这里的str只有一个,就是g的返回值

为了验证下,我把"G"改成了2

结果是

TypeError: 'int' object is not callable

OK了,大概意思就是,装饰器首先被"塞"一个参数,然后,返回值还要被调用一次,然而好像只有函数才能被调用,,所以,为了不报错,装饰器必须返回一个函数,装饰器必须是高阶函数......

我表示不服(￣へ￣),不就是函数吗,g的参数就是函数,,,骚操作上脑ing

def g(f):
    print("这里是G")
    return f

@g
def f():
    print("这里是F")
    return 1
'''
--------------------------------------------
这里是G
>>> f()
这里是F
'1'
'''

运行成功.但是....说好的"打印日志功能"呢,#没错"这里是G"就是我想要的日志

疑点: 打印的"这里是G"是第一行,是在输入"f()"之前发生的....

#单看g函数,它不算是"高阶函数"

作为成功的例子,它太失败了┐(￣ヘ￣)┌

##好吧,我投降了,负隅顽抗也不怎么有意思.....

闭包告诉我们一个道理,,,为了保证返回值一定是函数,最好的措施就是"在函数内部,现场造一个函数然后扔出去"

def g(f):
    print("这里是G")
    def h():
        print('这里是H')
        return "H"
    return h

@g
def f():
    print("这里是F")
    return 1
'''
--------------------------------------------
这里是G
>>> f()
这里是H
'H'
>>> f()
这里是H
'H'
>>> 
'''

f函数不执行了,,是的没错,我还多试了一次的

另外,两条日志只有里面的能用......(下文会解释的)

看了一下书,h函数返回f()的话,f函数就能被执行了,,,,,个人感觉装饰器应该叫"劫持器"

def g(f):
    print("这里是G")
    def h():
        print('这里是H')
        return f()
    return h

@g
def f():
    print("这里是F")
    return 1
'''
--------------------------------------------
这里是G
>>> f()
这里是H
这里是F
'1'
>>> f
<function g.<locals>.h at 0x0000020CBDBB6C80>
'''

按书上的思路解释下

'''
@g
def f():
    pass
>>>f()
等价于
>>>g(f) ()
g函数执行,返回
>>>h ()
h函数执行(打印日志)
>>>f()
f执行,返回1
>>>1
'''

加上参数,

def g(f):
    print("这里是G")
    def h(*args,**kw):
        print('这里是H')
        return f(*args,**kw)
    return h

@g
def f(*args,**kw):
    print("这里是F")
    return 1
'''
>>>f(*args,**kw)
等价于
>>>g(f) (*args,**kw)
g函数执行,返回
>>>h (*args,**kw)
h函数执行(打印日志)
>>>f(*args,**kw)
f执行,返回1
>>>1
'''

可以看出,(*args,**kw)本改被h函数拿走,所以,观察h函数,h把它的参数原封不动的交给了f

机智的我动了歪主意

def g(f):
    print("这里是G")
    def h():#h没要求参数
        print('这里是H')
        return f
    return h

@g
def f(*args,**kw):
    print("这里是F")
    return 1
'''
>>>f()(*args,**kw)
等价于
>>>g(f) ()(*args,**kw)
g函数执行,返回
>>>h ()(*args,**kw)
h函数执行,h拿的空参数
(打印日志)
>>>f(*args,**kw)
f执行,返回1
>>>1
'''

可是新的问题来了,后面f调用的时候得多加个空括号,否则

'''
>>>f(*args,**kw)
等价于
>>>g(f) (*args,**kw)
g函数执行,返回
>>>h (*args,**kw)
h函数执行(打印日志)返回f
>>>f
这是一个函数对象
'''

以上告诉我们一个道理"函数执行不执行取决于后面有没有括号"

举个例子

def m(a):
    print(a)
    return m

print(m(1)(2)(3)(4)(5)(6)(7)(8)(9)(10))
'''
-----------------------------
1
2
3
4
5
6
7
8
9
10
<function m at 0x000002832BDB10D0>

烧脑时刻

f = a.b.c()()[0]()[d()()[e]]
a模块的b类的c方法是高阶函数,最终返回一个列表,列表里有个函数
函数又返回一个字典............

下面开始,,装饰器的三层嵌套

它要解决的问题是装饰器需要额外的参数怎么办

就是g(m)变成了g(canshu)(m)多了一个括号,那么就得多加一层函数,,否则就会出错

>>> 1()
    1()
TypeError: 'int' object is not callable

.修改后的

def i(canshu):
    def g(f):
        print("这里是G")
        def h(*args,**kw):
            print('这里是H')
            return f(*args,**kw)
        return h
    return g

@i('canshu')
def f(*args,**kw):
    print("这里是F")
    return 1
'''
--------------------------------------------
这里是G
>>> f()
这里是H
这里是F
'1'
>>> f
<function i.<locals>.g.<locals>.h at 0x00000283A6886D08>
'''

解释一下

'''
>>>f(*args,**kw)
等价于
>>>i('canshu')(f)(*args,**kw)
i函数返回g
>>>g(f)(*args,**kw)
g函数执行,返回h
>>>h(*args,**kw)
h函数执行(打印日志)返回f(*args,**kw)
>>>f(*args,**kw)
f执行返回1


>>>f
等价于
>>>i('canshu')(f)
i函数返回g
>>>g(f)
g函数执行,返回h
>>>h
这是个函数对象
'''

所以

>>> f.__name__
'h'

真是劫持器呀,__name__都变了...

至于为什么每次最开始都有一个

应该可以这样理解

执行@i('canshu')语句时候装饰器被执行了一次

'''
>>>i('canshu')(f)
i函数返回g
>>>g(f)
g函数执行,打印了一句,返回h
>>>h
函数对象
'''

验证

def i(canshu):
    def g(f):
        print(f.__name__)
        print('这里是G')
        def h(*args,**kw):
            print('这里是H')
            return f(*args,**kw)
        return h
    return g

@i('canshu')
def f(*args,**kw):
    print("这里是F")
    return 1
'''
--------------------------------------------
f
这里是G
>>> f()
这里是H
这里是F
'1'
'''

综上,新启示

装饰器只有最内层的内容能起"日志"的作用,其它层的内容会在定义时执行
只要设计得当,一个函数后面可以跟好多括号,函数层数和括号数得尽量一致

另外,"劫持器"还可以用于调试,当两个函数不知道哪个更好时

def i(f_1):
    def g(f):
        def h(*args,**kw):
            print('%s把%s替换了'%(f_1.__name__,f.__name__))
            return f_1(*args,**kw)
        return h
    return g

def f_1(*args,**kw):
    print("这里是F_1")
    return 2

@i(f_1)
def f(*args,**kw):
    print("这里是F")
    return 1
'''
--------------------------------------------
>>> f()
f_1把f替换了
这里是F_1
2
'''