Python的decorator学习笔记

今天将python的decorator系统地学习了一下，自己也试着写了下，共有三种类型，第一种是普通装饰器，不带参数也不带返回值的。只需要将func（*args, **kwargs）在闭包函数块儿里执行就行，这是一定要执行的，不然就会返回与原来函数没有任何关系的函数了，甚至报错。同时如果是用decorator，那么原来函数的函数名字就变成了wrapper

的名字，这时，需要我们将原来的和wrapper的函数名字对调过来才行，如用@wrap(func)或者是return functools.update_wrapper(wrapper, func)这样的语句将其调换过来，这样函数的名字就和原来的一模一样了，不会再以后的使用中出错。有参数的decorator，就是多了一层嵌套，最外面的是传入参数的函数，里面嵌套原来的不带参数的decorator，最后将那个decorator返回就行。最后是被装饰的函数有返回值的情况，将func的结果赋值给result， 最后将result返回出去即可。下面是自己讲各种情况都试了下的代码，自己当笔记。

# __author__ = 'lulongfei'
from functools import wraps
import functools

'''
common decorator no args no return result
'''

def my_decortor(func):
@wraps(func)
def wrapper(*args, **kwargs):
print ('Calling decorated function')
return func(*args, **kwargs)
return wrapper

@my_decortor
def example():
'''Docstring'''
print ('Called example function')

example()
example.__name__
'''
decorator have args
'''
def log(text):
def decorator(func):
def wrapper(*args, **kwargs):
print '%s%s():'%(text, func.__name__)
return func(*args, **kwargs)
return wrapper
return decorator

@log('my_decorator:')
def example():
'''Docstring'''
print ('Called example function')

example()

def spamdef(fn):
def wrapper(*args):
print "spam, spam, spam"
return fn(*args)
return wrapper

@spamdef
def useful(a, b):
print a**2 + b**2

useful(3, 4)

def arg_sayer(what):
def what_sayer(meth):
def new(self, *args, **kwargs):
print what
return meth(self, *args, **kwargs)
return new
return what_sayer

def FooMaker(word):
class Foo(object):
@arg_sayer
def say(self):pass
return Foo()

foo1 = FooMaker('this')
foo2 = FooMaker('that')

print type(foo1)
foo1.say()

def flaz(self):return 'flaz'
def flam(self):return 'flam'

def change_method(new):
'''waring: only decorate the __new__() method with this decorator'''
if new.__new__ != '__new__':
return new

def __new__(cls, *args, **kwargs):
cls.flaz = flaz
cla.flam = flam
if hasattr(cls, 'say'):del cls.say
return super(cls.__class__, cls).__new__(cls, *args, **kwargs)
return __new__

class Foo(object):
@change_method
def __new__():pass
def say(self):print 'Hi me:', self

foo = Foo()
print foo.flaz()
foo.say()

'''decortor include the return result'''
def printdebug(func):
def decorator(user):
print ('enter the login')
result = func(user)
print ('exit the login')
return result
return decorator

@printdebug
def login(user):
print ('in login:' + user)
msg = "success" if user == "chuck" else "bruno"
return msg

result1 = login("chuck")
print result1

python的装饰器在编程里使用时，可以高效地减少一些重复的工作。

参考文章：
http://www.ibm.com/developerworks/cn/linux/l-cpdecor.html http://www.cnblogs.com/jerry-Chou/archive/2012/05/23/python-decorator-explain.html
感谢机器猫的文章