python高级编程——Decorator（修饰器）用法

 

一、基本应用：

 

在python里方法也是对象，它可以作为参数传递。

现在有个方法foo，我希望在它运行前后做些事情，那就可以使用decorator了。

 

# coding=utf-8

# 原方法
def foo():
    print 'running'

# 修饰方法
def outer(some_func):
    def inner():
        print "before some_func"
        some_func()
        print "after some_func"
    return inner

# 修饰foo方法
decorated = outer(foo) # 2

# 运行修饰后的方法
decorated()

输出：

before some_func
running
after some_func

可以看到我们在没有修改原方法foo()的前提下，修改了foo()运行之前和运行之后的行为。

 

二、使用@符号简化修饰器应用 

# coding=utf-8

# 修饰方法
def NewFoo(some_func):
    def inner():
        print "before some_func"
        some_func()
        print "after some_func"
    return inner

# 原方法
@NewFoo
def foo():
    print 'running'

# 运行修饰后的方法
foo()

修改后的代码更清晰简单了，而且foo方法名称都没有改变，行为就改变了。

这里注意：@NewFoo要与修饰方法名称一样，否则不会生效。通过修饰器，python实际调用是这样的：

foo = NewFoo(foo)
foo()

 

 

 

三、一个我使用到的案例

假设我们的开发code有多个分支，但测试代码希望只有一份，那就可以在测试代码上定义@dev_branchA_run_only、@dev_branchB_not_run这样的标记来适应多个分支的测试case。

 

 

参考：

http://simeonfranklin.com/blog/2012/jul/1/python-decorators-in-12-steps/

http://coolshell.cn/articles/11265.html

http://www.imooc.com/article/1257

 

--end