背景
闲来无事想熟悉一下Python的装饰器写法,就写了一个简单的类型检查装饰器。
思路
利用带参数的装饰器直接传入变量的预期类型,然后使用isinstance检查类型。
代码
def typeLimits(*types, **typesdict):
def desc(func):
def f(*args, **kwargs):
for i, v in enumerate(types):
if not isinstance(args[i], v):
raise Exception("Need a {} where {} is not.".format(v.__name__, args[i]))
for parm in typesdict:
if kwargs.get(parm, None) is None:
raise Exception("Need parameter {}.".format(parm))
if not isinstance(kwargs[parm],typesdict[parm]):
raise Exception('Need a {} for {} where {} is not.'.format(typesdict[parm].__name__, parm, kwargs[parm]))
return func(*args, **kwargs)
return f
return desc
@typeLimits(int, str)
def printOneIntAndOneString(n, s):
print(n ** 2, s.upper())
@typeLimits(int, name = str)
def printOneIntAndName(n, name):
print(n + n, name.upper())
printOneIntAndOneString(1,'abc') # 正确执行
# printOneIntAndOneString(1 + 2,3 + 4) 报错
printOneIntAndName(1, name = "aaa") 正确执行
# printOneIntAndName(1) 报错
# printOneIntAndName(1, name = 2.0) 报错
感想
这只是一个游戏之作,检查类型会耗费一些性能,真正部署运行的项目是不需要类型检查的,但可以在测试的时候使用类型检查来避免一些无谓的错误。
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。