python 装饰器

装饰器研究

装饰器种类

装饰器分为两类，一种是不需要参数的装饰器，另一种需要带参数。

下面展示一个不带参数的装饰器，实现简单的类似AOP功能，记录函数运行时间，或者记录日志等。

def running_time(func):
@functools.wraps(func)  #备注
def wrapper(*args, **kw):
start_time = time.time()
ret = func(*args, **kw)
end_time = time.time()
print '[{0}()] done, run time : {1} sec'.format(func.__name__, end_time - start_time)
return ret
return wrapper

@running_time
def nothing():
for x in xrange(10):
print "Maybe YJ ,have a try"

对于带参数的装饰器，需要进行二级封装，在上面装饰器的基础上，在封装一层。示例代码如下：

def mydecorator(arg1, arg2):
def _mydecorator(function):
def __mydecorator(*args, **kw):
# 借助参数做些什么事情
res = function(*args, **kw)
return res
return __mydecorator()
return _mydecorator

注意functools.wraps()函数的作用：调用经过装饰的函数，相当于调用一个新函数，那查看函数参数，注释，甚至函数名的时候，就只能看到装饰器的相关信息，被包装函数的信息被丢掉了。而wraps则可以帮你转移这些信息。

f = running_time(nothing)
print f.__name_

输出结果会应为wraps包装而不同。

装饰器应用

除了前面介绍可以用于日志，运行时间等记录外。还有一些应用。

1 . 用来实现单例模式：

def singleton(cls, *args, **kw):
instances = {}
def _singleton():
if cls not in instances:
instances[cls] = cls(*args, **kw)
return instances[cls]
return _singleton

@singleton
class MyClass4(object):
a = 1
def __init__(self, x=0):
self.x = x

one = MyClass4()
two = MyClass4()
print id(one)
print id(two)

2 . 实现函数静态属性的注入

def static_var(**kwargs):
def decorate(func):
for varname, value in kwargs.iteritems():
setattr(func, varname, value)
return func
return decorate

@static_var(age=12)
def struggle2(number):
print number
print struggle2.age
struggle2.age += 1
struggle2(222)

这个静态属性，与全局变量不同，是函数专享的。

3 . 还有比如 参数检查，缓存， 代理，上下文提供者等应用。暂时没有使用经验。大概使用的情况都差不多。还是很实用的技术。