Python装饰器学习（九步入门）

原文链接： http://www.cnblogs.com/rhcad/archive/2011/12/21/2295507.html
本文对原文略有改动。。。增加了自己的理解。。。
装饰器其实也就是一个函数，一个用来包装函数的函数，返回一个修改之后的函数对象，将其重新赋值原来的标识符。

1》在函数执行前和执行后分别附加其他的功能

def demo(func):#装饰函数的参数是被装饰的函数对象，返回原函数对象
    print("before myfunc() is called.")
    func()
    print("after myfunc() is called.")
    return func

def myfunc():
    print 'myfunc() is called!'

myfunc()#调用myfunc函数本身
print '------------------' 
myfunc=demo(myfunc)#该语句就是装饰的实质语句！！
print '------------------'
myfunc()#由于demo函数返回的是原函数本身，此处调用的依然是myfunc函数本身

运行结果：
myfunc() is called!
------------------
before myfunc() is called.
myfunc() is called!
after myfunc() is called.
------------------
myfunc() is called!

2》使用语法糖@来装饰函数

def demo(func):#装饰函数的参数是被装饰的函数对象，返回原函数对象
    print("before myfunc() is called.")
    func()
    print("after myfunc() is called.")
    return func
@demo  #程序运行到此处时，以myfunc为实参转去调用demo函数，将demo的返回值赋给myfunc
def myfunc():
    print 'myfunc() is called!'
print '-----------'
myfunc()#由于demo函数返回的是原函数本身，此处调用的依然是myfunc函数本身

运行结果：
before myfunc() is called.
myfunc() is called!
after myfunc() is called.
-----------
myfunc() is called!

3》使用内嵌包装函数来确保新函数被调用

def deco(func):
    def _deco():#内嵌包装函数的形参和返回值与原函数相同
        print("before myfunc() called.")
        result=func()
        print("after myfunc() called.")
        return result
    return _deco #装饰函数返回 内嵌包装函数对象
 
@deco
def myfunc():
    print("myfunc() called.")
    return 'ok'
 
print myfunc()#实际上，调用的是_demo函数

运行结果：
before myfunc() called.
myfunc() called.
after myfunc() called.
ok

4》对带参数的函数进行装饰

def deco(func):
    def _deco(a, b):#内嵌包装函数的形参和返回值与原函数相同
        print("before myfunc() called.")
        result = func(a, b)
        print("after myfunc() called.")
        return result
    return _deco#装饰函数返回内嵌包装函数对象
 
@deco
def myfunc(a,b):
    print("myfunc(%s,%s) called." % (a, b))
    return a + b
 
print myfunc(1, 2)#实际上，调用的是_demo函数

运行结果：
before myfunc() called.
myfunc(1,2) called.
after myfunc() called.
3

5》对参数数量不确定的函数进行装饰

def deco(func):
    def _deco(*args, **kwargs):#参数用(*args,**kwargs),自动适应变参和命名参数
        print("before %s called." % func.__name__)
        result=func(*args, **kwargs)
        print args
        print kwargs
        print("after %s called." % func.__name__)
        return result
    return _deco
@deco
def myfunc(a,b):
    print("myfunc() called.")
    return a+b
@deco
def myfunc1(*args, **kwargs):
    print("myfunc1() called.")

print myfunc(1, 2)
print '-----------'
print myfunc1(3, 4, 5,a=1,b=2)

运行结果：
before myfunc called.
myfunc() called.
(1, 2)
{}
after myfunc called.
3
-----------
before myfunc1 called.
myfunc1() called.
(3, 4, 5)
{'a': 1, 'b': 2}
after myfunc1 called.
None
6》让装饰器带参数

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("mymodule1")
def myfunc1():
    print("myfunc2() called.")
 
myfunc()
myfunc1()

运行结果：
before myfunc called [mymodule].
myfunc() called.
after myfunc called [mymodule].
before myfunc1 called [mymodule1].
myfunc2() called.
after myfunc1 called [mymodule1].

7》让装饰器带 类 参数

class locker:
    def __init__(self):
        print("locker.__init__() should be not called.")    
    @staticmethod
    def acquire():
        print("locker.acquire() called.（这是静态方法）") 
    @staticmethod
    def release():
        print("locker.release() called.（不需要对象实例）")

def deco(cls):
    '''cls 必须实现acquire和release静态方法
       因为装饰器中要调用acquire和release方法，所以类中要定义这2个方法
       之所以定义成静态方法，是因为静态方法可以通过类名来调用，不需要生成实例
    '''
    def _deco(func):
        def __deco():
            print("before %s called [%s]." % (func.__name__, cls))
            cls.acquire()
            try:
                return func()
            finally:
                cls.release()
        return __deco
    return _deco

@deco(locker)
def myfunc():
    print("myfunc() called.")

myfunc()

运行结果：
before myfunc called [__main__.locker].
locker.acquire() called.（这是静态方法）
myfunc() called.
locker.release() called.（不需要对象实例）

8》让装饰器带函数参数

def Filter(before_func,after_func):  
    print before_func  
    print after_func  
    def outer(main_func):  
        print main_func  
        def wrapper(request,kargs):  
            before_result=before_func(request,kargs)  
            if(before_result!=None):  
                return before_result;  
  
            main_result=main_func(request,kargs)  
            if(main_result!=None):  
                return main_result;  
  
            after_result=after_func(request,kargs)  
            if(after_result!=None):  
                return after_result;  
        return wrapper  
    return outer  
  
def before(request,kargs):  
    print request,kargs,'之前！'  
  
def after(request,kargs):  
    print request,kargs,'之后！'  
 
@Filter(before,after)  
def main(request,kargs):  
    print request,kargs  
  
main('hello','python')  
print main  

运行结果:

<function before at 0x02AC7BF0>
<function after at 0x02AC7C30>
<function main at 0x02AC7CF0>
hello python 之前！
hello python
hello python 之后！
<function wrapper at 0x02AC7D30>

9》装饰器带类参数，并分拆公共类到mylocker.py文件中，同时演示了对一个函数应用多个装饰器。。。

mylocker.py文件：

#!/usr/bin/env python
#coding:utf-8
class mylocker:
    def __init__(self):
        print("mylocker.__init__() called.")   
    @staticmethod
    def acquire():
        print("mylocker.acquire() called.")    
    @staticmethod
    def unlock():
        print("mylocker.unlock() called.")

class lockerex(mylocker):
    @staticmethod
    def acquire():
        print("lockerex.acquire() called.")   
    @staticmethod
    def unlock():
        print("lockerex.unlock() called.")

def lockhelper(cls):#参数cls是 类
    '''cls必须实现acquire和unlock静态方法，静态方法可以通过类名来调用，不需要实例'''
    def _deco(func):
        def __deco(*args, **kwargs):
            print("before %s called." % func.__name__)
            cls.acquire()
            try:
                return func(*args, **kwargs)
            finally:
                cls.unlock()
        return __deco
    return _deco

decorator.py文件：

#!/usr/bin/env python
#coding:utf-8
from mylocker import *
class example:
    def __init__(self):
        print '生成一个example类的实例！！'
    @lockhelper(mylocker)
    def myfunc(self):
        print("myfunc() called.")
    @lockhelper(mylocker)
    @lockhelper(lockerex)
    def myfunc1(self, a, b):
        print("myfunc1() called.")
        return a + b
if __name__=="__main__":
    a = example()
    print a
    print '-------------'
    print a.myfunc()
    print '-------------'
    print a.myfunc1(1, 2)

运行decorator.py文件，结果：

生成一个example类的实例！！
<__main__.example instance at 0x02AA2B48>
-------------
before myfunc called.
mylocker.acquire() called.
myfunc() called.
mylocker.unlock() called.
None
-------------
before __deco called.
mylocker.acquire() called.
before myfunc1 called.
lockerex.acquire() called.
myfunc1() called.
lockerex.unlock() called.
mylocker.unlock() called.
3

如果你感觉 “对一个函数应用多个装饰器” 不容易理解，没关系，上边的例子确实有点复杂，

这儿有一个简单的例子可以帮助你理解：python 对一个函数应用多个装饰器

虽然一个函数上面可以有多个装饰器，这种情况一般不推荐使用，因为这种情况下，程序的维护比较困难。。。

（完）