第二篇,来说说装饰器,在介绍装饰器之前先要理解闭包的概念。
闭包
内部函数对外部环境变量(但不是全局作用域变量)进行引用,内部函数就是一个闭包。
- 条件1:内部函数
- 条件2:外部环境变量
闭包就是满足这两个条件的函数
注意:闭包中不能修改外部作用域的局部变量
程序示例:
def addbase(x):
def addp(y):
return x+y
return addp
f=addbase(10)
print(f(8))
print(addbase(10)(9))
运行结果:
18
19
说完闭包,理解装饰器就比较简单了。
装饰器
装饰器(decorator)可以用来增强函数的功能,同时不改变原函数的定义,使用起来非常方便。
程序示例1:
#在不改变sum_1函数的基础上我们希望加入附加功能,
#检验参数,当参数为负数时,返回零值
def dec_1(func):
def jianyan(num1,num2):
#参数检验
if num1<0 or num2<0:
print("参数最好不要小于0")
return 0
return func(num1,num2)
return jianyan
#定义普通函数
def sum_1(num1,num2):
return (num1+num2)
sum_1=dec_1(sum_1)
print(sum_1(-2,0))
print(sum_1(5,4))
运行结果:
参数最好不要小于0
0
9
上面的定义过程略显复杂,Python提供了更加简洁方便的方式,使用@,上面的程序可以改写为下面这样,功能不变:
def dec_1(func):
def jianyan(num1,num2):
#参数检验
if num1<0 or num2<0:
print("参数最好不要小于0")
return 0
return func(num1,num2)
return jianyan
#使用@语法糖的方式
@dec_1
#定义普通函数
def sum_1(num1,num2):
return (num1+num2)
#sum_1=dec_1(sum_1)
print(sum_1(-2,0))
print(sum_1(5,4))
程序运行结果和上面保持一致:
参数最好不要小于0
0
9
装饰器还可以使用更加通用的定义方式,去装饰各种类型的函数,在这里仅仅用于演示我们写了一个比较简单的例子,用装饰器去输出每个函数的参数。
def dec_2(func):
def printer(*args, **kwargs):
print("参数是:%s,%s"%(args, kwargs))
return(func(*args, **kwargs))
return printer
@dec_2
def sum(num1,num2):
#求和
return num1+num2
@dec_2
def fool(num1,num2,num3):
#取最后一个参数返回
return num3
print(sum(1,2))
print(fool(6,7,8))
运行结果:
参数是:(1, 2),{}
3
参数是:(6, 7, 8),{}
8