装饰器

装饰器

一、概念

　　装饰器作用是为了给某一函数在不改变该函数代码的情况下增加额外的功能，实现原理是通过闭包实现，改进后可以通过语法糖实现，具体有函数装饰器，类装饰器等可以传参。

二、原理

　　1.1 闭包装饰器

def debug(func):
    def wrapper():
        print "[DEBUG]: enter {}()".format(func.__name__)
        return func()
    return wrapper

def say_hello():
    print "hello!"

say_hello = debug(say_hello)  # say_hello函数添加功能，并保持say_hello函数名不变

　　2.1 语法糖装饰器

def debug(func):
    def wrapper():
        print "[DEBUG]: enter {}()".format(func.__name__)
        return func()
    return wrapper

@debug
def say_hello():
    print "hello!"

 

三、被装饰函数传参

　　如果被装饰的函数需要传入参数，那么这个装饰器就坏了。因为返回的函数并不能接受参数，你可以指定装饰器函数wrapper接受和原函数一样的参数，比如：

def debug(func):
    def wrapper(something):  # 指定一毛一样的参数
        print "[DEBUG]: enter {}()".format(func.__name__)
        return func(something)
    return wrapper  # 返回包装过函数

@debug
def say(something):
    print "hello {}!".format(something)

四、带参数的装饰器

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


myfunc()
myfunc2()
"""
before myfunc called [mymodule].
 myfunc() called.
after myfunc called [mymodule].
before myfunc2 called [module2].
 myfunc2() called.
after myfunc2 called [module2].
"""

 

五、基于类装饰器

六、内置装饰器

　　1.1 @property

　　1.2 @staticmethod 

　　1.3 @classmethod

七、装饰器顺序