python装饰器

#coding=utf-8
import sys
'''
Created on 2016年8月7日

@author: yongliu
'''

def deco(func):
    print("before myfunc() called.")
    func()
    print("after myfunc() called.")
    return func

@deco
def myfunc():
    print("myfunc() called." )
    return 'ok'

    
def deco2(func):
    def _deco():
        print("before myfunc() called.")
        ret = func()
        print("after myfunc() called.")
        return ret
    return _deco

@deco2
def myfunc2():
    print("myfunc2() called.")
    return 'ok'



def deco3(func):
    def _deco(a, b):
        print("before %s called." % func.__name__)
        ret = func(a, b)
        print("after %s called. result: %s" % (func.__name__, ret))
        return ret
    return _deco
 
@deco3
def myfunc3(a, b):
    print(" myfunc3(%s,%s) called." % (a, b))
    return a + b

def deco4(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
 
@deco4
def myfunc41(a, b):
    print(" myfunc41(%s,%s) called." % (a, b))
    return a+b
 
@deco4
def myfunc42(a, b, c):
    print(" myfunc42(%s,%s,%s) called." % (a, b, c))
    return a+b+c
 
def deco5(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
 
@deco5("mymodule")
def myfunc51():
    print(" myfunc51() called.")
 
@deco5("module2")
def myfunc52():
    print(" myfunc52() called.")
    
    
    
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 deco6(cls):
    '''cls 必须实现acquire和release静态方法'''
    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
 
@deco6(locker)
def myfunc6():
    print("myfunc6() called.")
 
 
 
 
 
class mylocker7:
    def __init__(self):
        print("mylocker7.__init__() called.")
         
    @staticmethod
    def acquire():
        print("mylocker7.acquire() called.")
         
    @staticmethod
    def unlock():
        print("mylocker7.unlock() called.")
 
class lockerex7(mylocker7):
    @staticmethod
    def acquire():
        print("lockerex7.acquire() called.")
         
    @staticmethod
    def unlock():
        print("  lockerex7.unlock() called.")
 
def lockhelper(cls):
    '''cls 必须实现acquire和release静态方法'''
    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
 
 
#from mylocker import *
 
class example:
    @lockhelper(mylocker7)
    def myfunc(self):
        print(" myfunc() called.")
 
    @lockhelper(mylocker7)
    @lockhelper(lockerex7)
    def myfunc2(self, a, b):
        print(" myfunc2() called.")
        return a + b



if __name__ == '__main__':
    print "---------------------"
    #myfunc = deco(myfunc);
    print myfunc()
    print "---------------------"
    print myfunc2()
    print "---------------------"
    print myfunc3(3, 4)
    print "---------------------"
    myfunc41(1, 2)
    myfunc41(3, 4)
    myfunc42(1, 2, 3)
    myfunc42(3, 4, 5)
    print "---------------------"
    myfunc51()
    myfunc52()
    print "---------------------"
    myfunc6()
    myfunc6()
    print "---------------------"
    
    a = example()
    a.myfunc()
    print(a.myfunc())
    print(a.myfunc2(1, 2))
    print(a.myfunc2(3, 4))
    
    pass