可选参数装饰器:
解释:
既该装饰器可以作为简单装饰器一样去使用或者可以为装饰器传入一定的参数以达到对装饰器的控制。
示例:
from functools import wraps,partial
import logging
def logged(func=None,*,level=logging.DEBUG,name=None,message=None):
if func is None:
return partial(logged,level=level,name=name,message=message)
logname = name if name else func.__module__
log = logging.getLogger(logname)
logmsg = message if message else func.__name__
@wraps(func)
def wrapper(*args,**kwargs):
log.log(level,logmsg)
return func(*args,**kwargs)
return wrapper
@logged
def add(x,y):
return x+y
@logged(level=logging.CRITICAL,name='example')
def spam():
print('Spam!')
spam()
#结果:
'''
spam
Spam!
'''
这个知识点仅仅是告诉我们,大多数程序员更习惯用没有参数的装饰器(形如:@logged),但是本节课可以让编程更加具有一致性,如上例子既可以有参数(@logged(level,name,message)),也可以没参数(@logged),为了防止使用错误,这种可以接受可选参数的装饰器可以这样写‘@logged()’,虽然仅仅是多了个括号,却很好的提示了使用者这是一个可选参数的装饰器,以免和简单装饰器发生混淆。
利用装饰器对函数参数进行强制类型检查
我们想为函数参数添加强制性的类型检查功能,将其作为一种断言,或者与调用者之间的契约。
示例:
#我们需要实现的效果如下:
@typeassert(int,int)
def add(x,y):
return x+y
'''
>>>add(2,'hello')
>>>Traceback.......
>>>...
>>>TypeError:Argument y must be <class 'int'>
'''
要实现如上的案例效果,就得定义这个装饰器,示例如下,这个装饰器是一个高级装饰器,各位尽量理解里边的代码意思即可,编写这部分还是有难度的,里边的部分新知识会在代码中用注释来为各位解释:
inspect import signature
from functools import wraps
def typeassert(*args,**kwargs):#这里的参数传数据类型
def decorate(func):
if not __debug__:
return func
sig = signature(func)#该函数允许我们从一个可调用对象中抽取参数签名做检查;
print(sig,type(sig))## 得到的sig是一个包含了参数签名的signatrue对象,
# 形式有点像元组例如下方函数会得到(x, y=42) <class 'inspect.Signature'>
bounds_type = sig.bind_partial(*args,**kwargs).arguments
#返回的结果是一个Ordereddict有序字典对象(bounds_type),该顺序会依照参数签名顺序,键为参数名,值为数据类型
#bind_partial适合部分绑定提供的类型到参数名
@wraps(func)
def wrapper(*args,**kwargs):
bound_values = sig.bind(*args,**kwargs)#和bind_partial功能类似,只是不允许参数缺省
for name, value in bound_values.arguments.items():
#同样.arguments也会得到一个有序字典对象(Oredereddict)
#键为参数名,值为传入的实参值
if name in bounds_type:#name:键
if not isinstance(value,bounds_type[name]):#如果键对应的值不是这个名字所对应的数据类型
raise TypeError('Arguments {} must be {} '.format(name,bounds_type[name]))
return func
return wrapper
return decorate
@typeassert(int,int)
def add(x, y=42):
return x+y
add(23,'hello')
'''
Traceback (most recent call last):
File "F:/PycharmProjects/class_obj/class_one.py", line 853, in <module>
add(23,'hello')
File "F:/PycharmProjects/class_obj/class_one.py", line 843, in wrapper
raise TypeError('Arguments {} must be {} '.format(name,bounds_type[name]))
TypeError: Arguments y must be <class 'int'>
(x, y=42) <class 'inspect.Signature'>
Process finished with exit code 1
'''
如上装饰器就位更加通用的数据类型检测装饰器,其可以检测任意类型参数的数据类型。更为通用,后期如果需要可以直接使用。
想熟悉python?或者了解更多python基础知识,请关注:
小白piao: