微信号:Reboot51

介绍:专注于互联网运维开发分享、交流,让更多的运维工程师更加专注于自动化,为国内公有云开发、监控、运维贡献自己的力量.这里聚集着国内一线互联网工程师,乐于分享与交流 .

Python无参装饰器

2017-02-27 12:09 kk


需求:想要在test_func函数前后执行一些代码  

1.第一步(定义函数,将调用原函数,使用新函数替换)

def test_func():
    return 'test_func'


def test_wrapper_func():    print 'before'    rt = test_func()
   print 'after'    return rt
   

print
test_wrapper_func()

2.第二步(原函数可以定义参数)

def test_func():
    return 'test_func'


def test_wrapper_func(*args, **kwargs):    print 'before'    rt = test_func(*args, **kwargs)
   print 'after'    return rt
   
   
print test_wrapper_func()

3.第三步(定义工厂函数,返回wapper函数)

def test_func():
    return 'test_func'


def test_wrapper_func(*args, **kwargs)):    print 'before'    rt = test_func(*args, **kwargs))
   print 'after'    return rt

def
test_wrapper():
   return test_wrapper_func test_wrapper_func_temp = test_wrapper()

print test_wrapper_func_temp == test_wrapper_func
print test_wrapper_func_temp()
print test_wrapper_func()

4.第四步(去掉共有的wapper函数,放在工厂函数内部)

def test_func():
    return 'test_func'


def test_wrapper():    def wrapper(*args, **kwargs):        print 'before'        rt = test_func(*args, **kwargs)
       print 'after'        return rt    
   return wrapper test_wrapper_func = test_wrapper()
print test_wrapper_func()

5.第五步(可以为所有函数添加同一功能前后执行代码)

def test_func()
    return 'test_func'

def test_wrapper(func):
   def wrapper(*args, **kwargs):        print 'before'        rt = func(*args, **kwargs)        print 'after'        return rt
   return wrapper test_wrapper_func = test_wrapper(test_func) print test_wrapper_func()

6.第六步(使用装饰器)

def test_wrapper(func):
    def wrapper(*args, **kwargs):
        print 'before'
        rt = func(*args, **kwargs)
       print 'after'        return rt    
   return wrapper
   

@test_wrapper
def test_func():
   return 'test_func'


print test_func()

7.第七步(修正多个装饰器时,使用函数fun名称问题)

from functools import wraps

def test_wrapper(func):    
   
   @wraps(func)
   def wrapper(*args, **kwargs):        print 'before'        rt = func(*args, **kwargs)
       print 'after'        return rt    
   return wrapper
   
   
@test_wrapper
def test_func():
   return 'test_func'


print test_func()




戳原文,更有料!
 
Reboot 更多文章 用graphite diamond做监控 Python爬虫入门 丨如何在豆瓣中获取自己喜欢的TOP N电影信息 Python爬虫入门 丨如何在豆瓣中获取自己喜欢的TOP N电影信息 用graphite diamond做监控 用graphite diamond做监控
猜您喜欢 猜数游戏用Python应该这样写 21天混入数据科学家队伍(下) 谈谈编码风格与编码规范 国内物联网平台的发展、技术架构演进暨物联网解决方案发布 译文:Swift面试问题及答案 - part1