主要内容:
- 小目标:理解装饰器
- 主要内容:装饰器基本语法,装饰器应用
如果看完这篇文章,你还是弄不明白装饰器;
你来找我,我保证不打你,我给你发100的大红包。
1. 一个案例
1.1 需求:两个函数
mySum:计算列表的和,参数必须为列表
myMax:统计列表的最大值,参数必须为列表
代码实现:
def mySum(args):
#判断args是否是列表
if isinstance(args, list):
return sum(args)
return -1
def myMax(args):
#判断args是否是列表
if isinstance(args, list):
return max(args)
return -1
print(mySum([1,2,3]))
print(mySum((1,2,3)))
print(myMax(123))
结果:6, -1, -1
1.2 新需求:参数可以使元组
修改代码:
def mySum(args):
#判断args是否是列表
if isinstance(args, (list, tuple)):
return sum(args)
return -1
def myMax(args):
#判断args是否是列表
if isinstance(args, (list, tuple)):
return max(args)
return -1
print(mySum([1,2,3]))
print(mySum((1,2,3)))
print(myMax(123))
结果:6,6,-1
1.3 问题:
- 如果增加其他数据结构如何处理?
- 如果有N个这样函数如何处理?
从上一步修改中可以看到,判断条件时相同的,唯一不同时处理方式。
2. 闭包的应用
对上面进行分析,解决方式:
每个函数的核心功能不同,但是判断条件相同
将相同部分拆分成函数,使用闭包处理下:
def decoFunc(func):
#checkArgs为内置函数,func对于checkArgs是外部变量
def checkArgs(args):
#统一处理参数
if isinstance(args, (list, tuple)):
return func(args)
return -1
return checkArgs
#自定义的sun函数
mySum = decoFunc(sum)
#自定义max函数
myMax = decoFunc(max)
print(mySum([1,2,3]))
print(mySum((1,2,3)))
print(myMax(123))
结果:6,6,-1
- 分析:
decoFunc(func)中的func对于checkArgs为外部变量
decoFunc返回值为checkArgs函数
mySum, myMax指向checkArgs函数,但是每个函数内部的func不同
mySum, myMax调用之后,检查args,然后使用不同func处理ars
大家可以吧这个过程理解下,然后我们来看装饰器
3.装饰器
如何理装饰器:
装饰器:修改函数并返回新的函数;
装饰器与闭包原理类似,但是只用来处理函数;
基本语法:
@deco
def func(args):
pass
来看例子
def decoFunc(func):
#checkArgs为内置函数,func对于checkArgs是外部变量
def checkArgs(args):
#统一处理参数
if isinstance(args, (list, tuple)):
return func(args)
return -1
print('checkArgs:', checkArgs)
return checkArgs
@decoFunc
def mySum(args):
return sum(args)
print('mySum:', mySum)
结果:
checkArgs: .checkArgs at 0x000001C4A21580D8>
mySum: .checkArgs at 0x000001C4A21580D8>
分析:
mySum变量名指向decoFunc中的返回值:checkArgs函数br/>@decoFunc就是装饰器的语法糖
从结果中看到,我们没有调用mySum函数,但是装饰器函数会默认执行
装饰器实质:对比两个例子,装饰器实质:mySum = decoFunc(mySum)
3.带参数的装饰器
新需求:在不修改checkArgs参数前提下,调用mySum,myMax前打印相应的函数名
直接上代码:
def decoFunc(fname):
print("call decoFunc")
#在_decoFunc在加一层函数,返回_decoFunc
def _decoFunc(func):
#checkArgs为内置函数,func对于checkArgs是外部变量
def checkArgs(args):
#统一处理参数
if isinstance(args, (list, tuple)):
print("call func:", fname)
return func(args)
return -1
print('checkArgs:', checkArgs)
return checkArgs
return _decoFunc
@decoFunc('mySum')
def mySum(args):
return sum(args)
print('mySum:', mySum)
mySum([1,2,3,4,5])
结果:
call decoFunc
checkArgs: ._decoFunc..checkArgs at 0x000001C4A1D00B88>
mySum: ._decoFunc..checkArgs at 0x000001C4A1D00B88>
call func: mySum
分析:
函数包含关系:decoFunc->_decoFunc->checkArgs
@decoFunc('mySum')执行过程:
1.执行decoFunc('mySum'),设置fname,返回_decoFuncbr/>2.执行@decoFunc('mySum')实质是@返回_decoFunc,
3>mysum = _decoFunc(mySum)
大家看到装饰器,尤其带参数的装饰器,感觉晕晕的,
我们要做的,就是一步一步分析,搞清楚原理;
以上就是装饰相关内容;
总结:
- 理解装饰器,必须先理解闭包;
- 理解闭包,必须弄清楚作用域及函数返回值;
- 装饰器在实际工作,尤其适用第三方框架时候(Djano,Scrapy等),应用十分广泛