python 装饰器

一、装饰器定义。

装饰器(Decorators)是 Python 的一个重要部分。简单地说：他们是修改其他函数的功能的函数。他们有助于让我们的代码更简短。

应用场景：

　　假如：此时写了很多函数，有一个新的需求需要在多个函数中添加或者修改一个功能，如果每个函数都修改一次显得太过繁重，此时装饰器就出现了。

　　首先定义一个装饰器，写好要修改或者添加的功能，每个函数前用@装饰器名称，即可调用装饰器函数。

定义一个装饰器：

def outer(func):
    def inner():
        print "start"
        r = func() # r = None
        print  "end"
        return r
    return inner # 注意：返回的只是函数名，不是inner()

定义一个函数，并引用装饰器：

@outer  # 引用装饰器
def faa():
    print ("faa")

a = faa()
print a

查看执行结果

start
faa
end

从执行结果我们可以看出装饰器的执行顺序，先执行装饰器的部分功能，再执行普通方法中，然后执行装饰器其它代码，顺序可以根据需求调整。

“@”作用：

•  执行outer函数，并将@下面的函数名传给outer函数，当做参数。
•  将outer函数的返回值重新赋值给faa faa=outter的返回值
•  本质上：相当于在原来的函数faa的基础上，前后增加了一些操作。将原函数封装到另外一个函数上，并将返回值重新赋值给原函数、

二、装饰器加参数。

装饰器内部调用函数时，假如函数需要传参，那么以上代码无法满足传参的需求。

def outer(func):
    def inner(a1,a2):
        print "==== start ====="
        r = func(a1,a2)
        print "==== end ===="
        return r
    return inner

@outer
def faa(a1,a2):
    print "普通函数中的功能"

# ============= 执行结果 ================
ret = faa("a","b")
print ret

以上代码即可实现传参功能。

但是我们会发现这样传参很不灵活，既假如还有其它函数需要传参，而传参的个数或者参数的类型不一样，那么以上代码则无法满足其它函数传参。

三、装饰器传多个参数

 普通函数传多个参数：

def fun(*args,**kwargs):
    print args
    print kwargs

fun("a1","a2",k1="steven",k2="18")

# ============= 返回结果 ==============
('a1', 'a2')
{'k2': '18', 'k1': 'steven'}

装饰器传参

def outer(func):
    def inner(*args,**kwargs):
        print  "=" * 20
        print "装饰器 start"
        r = func(*args,**kwargs)
        print "装饰器，end"
        return r
    return inner

@outer
def fun1(a1):
    print "fun1 一个参数"

@outer
def fun2(a1,a2):
    print "fun2 两个参数"

@outer
def fun3(*args,**kwargs):
    print args
    print kwargs

ret1 = fun1("f1_a1")
ret2 = fun2("f2_a1","f2_a2")
ret3 = fun3("f3_a1","f3_a3",k1="k1",k2="k2")



============ 返回结果 ==============

装饰器 start
fun1 一个参数
装饰器，end
====================
装饰器 start
fun2 两个参数
装饰器，end
====================
装饰器 start
('f3_a1', 'f3_a3')
{'k2': 'k2', 'k1': 'k1'}
装饰器，end

 

 四、添加多个装饰器

def outer(func):
    def inner(*args,**kwargs):
        print "="*20
        print "装饰器outer start"
        r = func(*args,**kwargs)
        print "装饰器outer end"
        return r
    return inner

def outer_1(func):
    def inner(*args,**kwargs):
        print '='*20
        print "装饰器outer_1 start"
        r = func(*args,**kwargs)
        print "装饰器outer_1 end"
        return r
    return inner

@outer
@outer_1
def fun1(a1):
    print "fun1"

f1 = fun1("多个装饰器")


# =========== 返回值 =====================


====================
装饰器outer start
====================
装饰器outer_1 start
fun1
装饰器outer_1 end
装饰器outer end