首页 » CentOS 7 » 文件、目录管理 » 正文

Python装饰器

插队插队

def foo(func):
    print("start dingdingding..")
    def wapper():
        print("wo shi GEGE")
        func()
    return wapper

@foo
def brother():
    print("wo shi didi")

brother()

'''
第一步执行:@foo  = brother = foo(brother)
2 foo(func) = foo(brother)         func = brother
3 print("start dingdingding..")
4 def wapper():
5 return wapper        也就是说调用def foo(func) 最终返回的是 wapper  那也就是相当于了第一步  @foo  = brother = foo(brother)  也就是brother = wapper
6 brother()        如上 brother加括号 相当于 wapper加括号
7 执行 def wapper()      
8 print("wo shi GEGE")
9 func()  在如上我们就知道  func = brother
10 执行 def brother():
11 print("wo shi didi")
'''

示例一

# -*- coding:gbk -*-
'''示例1: 使用语法糖@来装饰函数,相当于“myfunc = deco(myfunc)
但发现新函数只在第一次被调用,且原函数多调用了一次'''
 
def deco(func):
    print("before myfunc() called.")
    func()
    print("  after myfunc() called.")
    return func
 
@deco
def myfunc():
    print(" myfunc() called.")
 
myfunc()

这里@deco这一句,和myfunc = deco(myfunc)其实是完全等价的,只不过是换了一种写法而已

将@deco 替换为 myfunc = deco(myfunc) 程序首先调用deco(myfunc),得到的返回结果赋值给了myfunc (注意:在Python中函数名只是个指向函数首地址的函数指针而已) 而deco(myfunc)的返回值就是函数myfunc()的地址 这样其实myfunc 没有变化,也就是说,最后的两次myfunc()函数调用,其实都没有执行到deco()。 有同学就问了,明明打印了deco()函数里面的内容啊,怎么说没有调用到呢。这位同学一看就是没有注意听讲,那一次打印是在@deco 这一句被执行的。大家亲自动手试一下就会发现” myfunc() called.” 这句打印输出了三次。多的那次就是@deco这里输出的,因为@deco 等价于myfunc = deco(myfunc),这里已经调用了deco()函数了。

第二步 :确保装饰器被调用

怎么解决装饰器没有被调用的问题呢

# -*- coding:gbk -*-
'''示例2: 使用内嵌包装函数来确保每次新函数都被调用,
内嵌包装函数的形参和返回值与原函数相同,装饰函数返回内嵌包装函数对象'''

import time

def deco(func):
    def _deco():
        print("before myfunc called")
        func()
        print("after myfunc alled")
    return _deco

@deco       ##myfunc = deco(myfunc) = _deco
def myfunc():
    print("myfunc called at {0}".format(time.strftime('%Y.%m.%d-%H:%M:%S')))

myfunc()

这里其实不需要我解释了,还是按照第一步中的方法做替换就可以了。还是啰嗦几句吧。。 @deco 替换为 myfunc = deco(myfunc) 程序首先调用deco(myfunc),得到的返回结果赋值给了myfunc ,这样myfunc 就变成了指向函数_deco()的指针 以后的myfunc(),其实是调用_deco()

第三步 :对带参数的函数进行装饰

破案过程和第一步、第二步完全一致,不再重复了

# -*- coding:gbk -*-
'''示例5: 对带参数的函数进行装饰,
内嵌包装函数的形参和返回值与原函数相同,装饰函数返回内嵌包装函数对象'''
 
def deco(func):
    def _deco(a, b):
        print("before myfunc() called.")
        ret = func(a, b)
        print("  after myfunc() called. result: %s" % ret)
        return ret
    return _deco
 
@deco
def myfunc(a, b):
    print(" myfunc(%s,%s) called." % (a, b))
    return a + b
 
myfunc(1, 2)

编写一个decorator,能在函数调用的前后打印出’begin call’和’end call’的日志。

def log(func):
    def wrapper(*args,**kw):
        print ('begin call')
        func(*args,**kw)
        print ('end call')
    return wrapper
 
@log
def f():
    print ('this is a app!')
 
f()

输出结果:

begin call
this is a app!
end call
赞 (1)

发表评论