python装饰器

1.装饰器

1.1通过代码初识装饰器

#!/usr/bin/python
# -*- coding: utf-8 -*-
def w1(func):
    def inner():
      func()
    return inner
@w1
def f1():
    print('f1')

if __name__ == '__main__':
    a = f1()
    print a

 首先，python解释器会执行以下两个步骤：

                 （1）将w1函数加载到内存

                 （2）@w1

其中第一步都很容易理解，但是第二步却有很大的内容，怎么去理解@w1的作用呢？

其实@w1等同于 f1 = w1(f1)，可以看出f1此时等同于 ，把f1传入w1函数，返回的结果inner，所以有以下运行结果：

1.2通过代码深入理解装饰器执行原理

#!/usr/bin/python
# -*- coding: utf-8 -*-
def makeBold(fn):
    def wrapped():
        return "<b>" + fn() + "</b>"
    return wrapped

def makeItalic(fn):
    def wrapped():
        return "<i>" + fn() + "</i>"
    return wrapped
    
@makeBold
def test1():
    return "hello world-1"

@makeItalic
def test2():
    return "hello world-2"

@makeBold
@makeItalic
def test3():
    return "hello world-3"

print test1()
print test2()
print test3()

该代码执行结果：

1. 第一个print 结果分析：

             首先test1()函数被@makeBold装饰，等同于：test1 = makeBold(test1)，执行后，test1 = wrapped,故程序运行的结果为函数wrapped()执行的结果。

2. 第二个print结果分析：

             同第一个print结果分析

3. 第三个print结果分析：

              首先test3()函数被@makeBold@makeItalic装饰，test3 = makeItalic(test3)，执行后，test3 = makeItalic()函数的返回结果，然后再被@makeBold装饰，等同于test3 = makeBold(test3)=makeBold(makeItalic()函数的返回结果), 此时我们可以发现装饰器的装饰是从里到外进行装饰，从打印的结果可以看出执行是从外到里进行执行。

1.3 装饰的函数无参数

#!/usr/bin/python
# -*- coding: utf-8 -*-
def makeBold(fn):
    def wrapped():
        return "<b>" + fn() + "</b>"
    return wrapped

#test1()为装饰的函数
@makeBold
def test1():
    return "hello world-1"
print test1()

1.4 装饰的函数带参数

#!/usr/bin/python
# -*- coding: utf-8 -*-
def timefun(func):
	def wrappedfunc(a, b):
		func(a, b)
	return wrappedfunc

@timefun
def foo(a, b):
	print(a+b)
print foo(3,5)

 

我们可以看出当我们装饰的函数foo()有两个参数a和b，timefun()函数中参数func也必须传两个参数

1.5 装饰的函数有不定⻓参数

#!/usr/bin/python
# -*- coding: utf-8 -*-
def timefun(func):
	def wrappedfunc(*args, **kwargs):
		func(*args, **kwargs)
	return wrappedfunc
@timefun
def foo(a, b, c):
	print(a+b+c)

print foo(3,5,7)

1.6 装饰器中的return

#!/usr/bin/python
# -*- coding: utf-8 -*-
def timefun(func):
	def wrappedfunc():
		return func()
	return wrappedfunc

@timefun
def getInfo():
	return '----hahah---'

print foo()
print "==============="
print getInfo()

我们可以看出被装饰的函数有return时，timefun()函数中func参数，也要有return的使用

1.7 当装饰器带参数,在原有装饰器的基础上， 设置外部变量

#!/usr/bin/python
# -*- coding: utf-8 -*-
# 原有的装饰器
def timefun_arg(pre="hello"):
	def timefun(func):
		def wrappedfunc():
			return func()
		return wrappedfunc
	return timefun

@timefun_arg("python")
def too():
	print("I am too")
#此时可理解为too()==timefun_arg("python")(foo)()
print too()

此时我们可知如果装饰器带参数，则需要在原装饰器上重新定义一个函数，且该函数需要定义一个参数接收装饰器参数。