装饰器内核
python装饰器的内核其实是将函数作为参数传递到另一个函数体内部,也就是说在该函数执行前优先执行装饰器函数,应用场景包括用户验证,日志撰写,数据验证等等,可以避免多段的重复代码,提升代码的可复用性。接下来将由浅入深的介绍三种以函数作为装饰器的基本形式。
不带参数的装饰器
先上代码
#不带参数的装饰器
def func_name(func):
def inner():
print('decorator called')
func()
return inner
@func_name
def s1():
print('s1 called')
s1() #开始执行
#---------------执行结果-------------------
#decorator called
#s1 called
讲解一下这段代码,代码执行入口为s1()
,此时解释器接触到了func_name
装饰器,于是此时s1()
函数暂停执行,将s1
作为一个参数带入到装饰器函数,解释器开始执行func_name(s1)()
,根据func_name
函数定义,执行函数inner()
,因此首先打印decorator called,继续执行会打印s1 called。从而实现了在函数执行前先做一部分工作。
带参数的装饰器
接下来做一个用户验证的示例来讲解,想象这样一个场景,系统登录时,如果用户名密码正确那么可以进入系统,否则就拒绝进入,因此用户验证是系统登陆这样一个功能之前必须经历的一个步骤。代码如下
#带参数的装饰器
def dec(func):
def inner(*args,**kwargs): #s2参数传递
print('开始装饰')
if kwargs.get('user')=='root' and kwargs.get('password')=='dlut':
print('验证通过')
func(*args,**kwargs) #执行s2()
else:
print('验证不通过') #不执行s2()
return inner
@dec
def s2(*args,**kwargs):
print('开始求和,打印关键词~')
print(sum(args))
print(kwargs)
s2(3,6,8,user='root',password='dlut') #函数入口
#---------------执行结果-------------------
#开始装饰
#验证通过
#开始求和,打印关键词~
#17
#{'user': 'root', 'password': 'dlut'}
同理,执行带参数的s2()
函数时,遇到dec
装饰器,将参数传递到装饰器内部定义的inner()
函数中,从而可以获取相关参数进行用户验证。
装饰带返回值的函数
装饰具有返回值的函数应该怎么做呢,话不多说,代码如下
# 装饰带返回值的函数
def dec2(func):
def inner(*args):
print('开始装饰~')
ret=func(*args)
return ret #此处返回func_run函数的返回值即可
return inner
@dec2
def func_run(*args):
return sum(args)
_sum=func_run(1,4,6) #函数入口
print('求和结果',_sum)
#---------------执行结果-------------------
#开始装饰~
#求和结果 11
类用作修饰器
接下来给大家分享一段代码,关于类用作修饰器,它的功能是,通过修饰器完成用户验证,若验证成功则继续执行函数,否则直接返回False
#类用作修饰器
print('-------------------------')
class Dec_obj(object):
def __init__(self,func):
self._func=func
pass
def __call__(self, *args, **kwargs):
print('dec called')
if kwargs.get('user')=='mark' and kwargs.get('password')=='wo4xuhao':
return self._func(*args,**kwargs)
else:
return False
@Dec_obj
def func(*args, **kwargs):
print('func entered')
return sum(args)
print(func(2,6,9,user='mark',password='wo4xuhao'))
关于python函数装饰器的分享就到这里啦,还请务必亲自写代码实践!