Python学习笔记 装饰器详解

Python装饰器详解：

装饰器的作用无疑是在函数执行的前后添加额外的内容，以满足我们需要的需求，例如日志处理，性能测试，事物处理等。

1：最简单的实现前后添加类容的需求：

#!/usr/bin/python

#-*-coding:utf-8-*-

#deco1.py

 

def deco(func):

print 'before'

func()

print 'after'

 

def myfunc():

print 'myfunc'

 

myfunc=deco(myfunc)

before

myfunc

after

无疑这样的方式不易于进行扩展以及维护 因为它为了需求改动了我原来的代码 原来应该只是  func() 这一个调用，于是装饰器模式就出现了

2：极少改动的源代码的装饰器：

#!/usr/bin/python

#-*-coding:utf-8-*-

#deco2.py

 

def deco(func):

def wrapper():

print 'before'

func()

print 'after'

return wrapper

 

 

def myfunc():

print 'myfunc'

 

myfunc=deco(myfunc)

myfunc()

print myfunc.__name__

before

myfunc

after

wrapper

3：使用语法糖@来装饰函数 .

#!/usr/bin/python

#-*-coding:utf-8-*-

#deco3.py

 

def deco(func):

def wrapper():

print 'before'

func()

print 'after'

return wrapper

 

@deco

def myfunc():

print 'myfunc'

 

myfunc()

 

print myfunc.__name__

before

myfunc

after

wrapper

无疑可以看出来 @deco 相当于执行了命令 myfunc=deco(myfunc) 这一个命令，同时它也就只是这样而已

4：带有参数的高阶装饰器：

#!/usr/bin/python

#-*-coding:utf-8-*-

#deco4.py

 

def outer(text):

def deco(func):

def wrapper(*args,**kw):

print 'before'

print text

func(*args,**kw)

print 'after'

return wrapper

return deco

 

@outer('hello outer')

def myfunc(name):

print 'myfunc',name

 

myfunc('licheng')

 

print myfunc.__name__

before

hello outer

myfunc licheng

after

wrapper

无疑语法糖加上参数也只是在外围包上了一层函数和返回值而已。 @outer('hello outer')相当于 myfunc=outer('hello outer')(myfunc)

5：解决函数名的问题

上述装饰器基本已经可以说完整了 但是依旧存在一个问题，那就是我们可以看到经过装饰后，myfunc的函数名变成了wrapper.显然这对于一些依赖于函数名的反射机制会发生异常错误等。因此我们需要将 myfunc.__name__ =myfunc 实现

而这只需要一句话  @functools.wraps(func)就可以实现。

#!/usr/bin/python

#-*-coding:utf-8-*-

#deco4.py

import functools

def outer(text):

def deco(func):

@functools.wraps(func)

def wrapper(*args,**kw):

print 'before'

print text

func(*args,**kw)

print 'after'

return wrapper

return deco

 

@outer('hello outer')

def myfunc(name):

print 'myfunc',name

 

myfunc('licheng')

 

print myfunc.__name__

before

hello outer

myfunc licheng

after

myfunc

这个装饰器 @functools.wraps(func) 是functools的一个特殊的装饰器，相当于 wrapper=functools.wraps(func)(wrapper)