python深入学习笔记5-装饰器

作者：Vamei 出处：http://www.cnblogs.com/vamei

05、装饰器

装饰器用于对一个函数、方法或者类进行加工。

• 装饰函数和方法

我们先定义两个简单的数学函数，一个用来计算平方和，一个用来计算平方差：

# get square sum
def square_sum(a, b):
    return a**2 + b**2

# get square diff
def square_diff(a, b):
    return a**2 - b**2

print(square_sum(3, 4))
print(square_diff(3, 4))

在拥有了基本的数学功能之后，我们可能想为函数增加其它的功能，比如打印输入。我们可以改写函数来实现这一点：

# modify: print input

# get square sum
def square_sum(a, b):
    print("intput:", a, b)
    return a**2 + b**2

# get square diff
def square_diff(a, b):
    print("input", a, b)
    return a**2 - b**2

print(square_sum(3, 4))
print(square_diff(3, 4))

现在，我们使用装饰器来实现上述修改：

def decorator(F):
	def new_F(a,b):
		print("input",a,b)
		return F(a,b)

	return new_F

@decorator
def square_sum(a,b):
	return a**2+b**2;

@decorator
def square_diff(a,b):
	return a**2-b**2

print(square_sum(3,4))
print(square_diff(3,4))

相当于在每个函数前加上了print("input",a,b)。

我们传入一个F函数，装饰器返回一个带有打印功能的new_F函数！！

只要在每个原函数（square_sum和square_diff）前调用@decorator就可实现对函数的装饰功能！

• 含参的装饰器

装饰器的语法允许我们调用decorator时提供其他参数，如@decorator(a)

def pre_str(pre=''):#a new wrapper layer
	#old decorator
	def decorator(F):
		def new_F(a,b):
			print(pre+" input",a,b)
			return F(a,b)

		return new_F
	return decorator

@pre_str('^_^')
def square_sum(a,b):
	return a**2+b**2;

@pre_str('T_T')
def square_diff(a,b):
	return a**2-b**2

print(square_sum(3,4))
print(square_diff(3,4))

输出：

^_^ input 3 4
25
T_T input 3 4
-7

上面的pre_str是允许参数的装饰器。它实际上是对原有装饰器的一个 函数封装，并返回一个装饰器。我们可以将它理解为一个含有 环境参量的闭包。当我们使用@pre_str('^_^')调用的时候，Python能够发现这一层的封装，并把参数传递到装饰器的环境中。

相当于：

square_sum = pre_str('^_^') (square_sum)

• 装饰类

装饰器可以装饰一个函数，也可以装饰一个类。

def decorator1(aClass):
	class newClass:
		def __init__(self,age):
			self.total_display=0
			self.wrapped=aClass(age)
		def display(self):
			self.total_display+=1
			print("total_display",self.total_display)
			self.wrapped.display()

	return newClass 

@decorator1
class Birdd:
	def __init__(self,age):
		self.age=age
	def display(self):
		print("My age is",self.age)

eagle=Birdd(5)
for i in range(1,4):
	eagle.display()

print(eagle.total_display)

输出：

total_display 1
My age is 5
total_display 2
My age is 5
total_display 3
My age is 5
3

装饰器decorator1增加了self.total_display用于记录display()的调用次数，self.wrapped用于接收原有的类（本例中为Birdd())