Python---装饰器

装饰器是一个非常著名的设计模式，常常被用于有切面需求的场景，较为经典的有插入日志、性能測试、事务处理等。装饰器是解决这类问题的绝佳设计。有了装饰器，我们就能够抽离出大量函数中与函数功能本身无关的雷同代码并继续重用。

概括的讲，装饰器的作用就是为已经存在的对象加入额外的功能。

假如我要计算一个函数的运行时间：

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

计算完一个之后，我又想计算其他几个函数的运行时间。为了避免反复写类似的函数，就能够让装饰器来帮你解决问题。

import time

def timeit(func):

def wrapper():

start = time.clock()

func()

end = time.clock()

print ‘used’,end - start

return wrapper

@timeit

def function():

print 'into function()'

function()

-----------------------------------

使用内嵌包装函数来确保每次新函数都被调用

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19

# -*- coding:gbk -*- '''演示样例4: 使用内嵌包装函数来确保每次新函数都被调用。 内嵌包装函数的形參和返回值与原函数同样，装饰函数返回内嵌包装函数对象'''   def deco(func):     def _deco():         print("before myfunc() called.")         func()         print("  after myfunc() called.")         # 不须要返回func。实际上应返回原函数的返回值     return _deco   @deco def myfunc():     print(" myfunc() called.")     return 'ok'   myfunc() myfunc()

对带參数的函数进行装饰

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19

# -*- 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) myfunc(3, 4)

对參数数量不确定的函数进行装饰

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26

# -*- coding:gbk -*- '''演示样例6: 对參数数量不确定的函数进行装饰， 參数用(*args, **kwargs)，自己主动适应变參和命名參数'''   def deco(func):     def _deco(*args, **kwargs):         print("before %s called." % func.__name__)         ret = func(*args, **kwargs)         print("  after %s called. result: %s" % (func.__name__, ret))         return ret     return _deco   @deco def myfunc(a, b):     print(" myfunc(%s,%s) called." % (a, b))     return a+b   @deco def myfunc2(a, b, c):     print(" myfunc2(%s,%s,%s) called." % (a, b, c))     return a+b+c   myfunc(1, 2) myfunc(3, 4) myfunc2(1, 2, 3) myfunc2(3, 4, 5)

让装饰器带參数

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24

# -*- coding:gbk -*- '''演示样例7: 在演示样例4的基础上，让装饰器带參数， 和上一演示样例相比在外层多了一层包装。 装饰函数名实际上应更有意义些'''   def deco(arg):     def _deco(func):         def __deco():             print("before %s called [%s]." % (func.__name__, arg))             func()             print("  after %s called [%s]." % (func.__name__, arg))         return __deco     return _deco   @deco("mymodule") def myfunc():     print(" myfunc() called.")   @deco("module2") def myfunc2():     print(" myfunc2() called.")   myfunc() myfunc2()