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要与修饰方法名称一样,否则不会生效 

 

三、一个我使用到的案例

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

 

 

参考:

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

 

--end

 

你可能感兴趣的:(python高级编程——Decorator(修饰器)用法)