Python的修饰器@

Posted 2020-07-28

  修饰器是一个很著名的时机模式，经常用于有切面需求的场景，如插入日志、性能测试、事务处理等。修饰器能够很好地解决这些问题，有了修饰器我们能抽离出大量函数中与函数功能本身无关的雷同代码并继续使用。也就是说，修饰器的作用就是为已经存在的函数对象添加额外的功能。

1.修饰器入门：

1.1.需求的由来：

修饰器的定义很抽象，先来看一个例子：

def foo():
    print "in foo()"
    
foo()

这个函数的功能是打印出一窜字符窜。如果想要测试执行这个函数用了多长时间，我们可以这样做：

import time
def foo():
    start = time.clock()
    print "in foo()"
    end = time.clock()
    print "used:",end-start
    
foo()

这样能够很好的达到目的。但是想测试一个模块的所有函数的执行时间呢，就得把所有函数中都加入如上时间差的计算方法，这样不太现实。

1.2.以不变应万变

为了不改变原来的函数，我们可以定义一个函数timeit，将foo()的引用传递给他，然后在timeit中调用fool并进行计时，这样我们就不用修改foo函数而达到目的了。

import time

def foo():
    print "in foo()"
    
def timeit(func):
    start = time.clock()
    func()
    end = time.clock()
    print "used:",end - start
    
timeit(foo)

这样看上去逻辑没有问题，而且可以正常的工作。但却修改了调用部分的代码，原本是foo()调用，现在却成了timeit(foo)，如果foo在很多处都被调用了，就需要在很多处修改代码。

1.3最大限度的少改动

如果不改动调用的代码，也就意味着调用foo()需要产生timeit(foo)的效果。我们可以这样做，把timeit(foo)的返回值付给foo，然后直接调用foo()，就不用修改源代码了。

import time

def foo():
    print "in foo()"
    
def timeit(func):
    def wrapper():
        start = time.clock()
        func()
        end = time.clock()
        print "used:",end - start
    
    return wrapper
    
foo = timeit(foo)
foo()

这样，我们只需在定义foo以后和调用foo之前，加上foo=timeit(foo)，就可以达到目的了。这就是修饰器，看起来像foo被timeit修饰了。

2.正规的修饰器

上面的代码，看似没法再精简了，python于是提供了一个特殊的语法来降低字符输入量：

import time

def timeit(func):
    def wrapper():
        start = time.clock()
        func()
        end = time.clock()
        print "used:",end - start
        
    return wrapper
    
@timeit
def foo():
    print "in foo()"
    
foo()

在第12行的@timeit，效果和foo=timeit(foo)一样，而且看上去更有修饰器的感觉。

这就是Python中修饰器的原理

本文出自 “黑色时间” 博客，请务必保留此出处http://blacktime.blog.51cto.com/11722918/1791486