第一个参数类型决定执行特定的方法体
一、单分派泛函数
只对函数的第一个参数做类型检查,发现局限性太大,没什么特别的应用场景,也可能是我没用到
1 from functools import singledispatch 2 3 4 @singledispatch 5 def func(a, *args): 6 raise TypeError("暂不支持{}类型的执行".format(type(a))) 7 8 9 @func.register(int) 10 def _(a, *args): 11 print(">>【{}】是整数".format(a)) 12 return a + args[0] 13 14 15 @func.register(str) 16 def _(a, *args): 17 print(">>【{}】是字符串".format(a)) 18 return a + args[0] 19 20 21 if __name__ == '__main__': 22 print(func(4, 5)) 23 print(func("只有", "永")) 24 print(func([1, 2], 3))
二、多分派泛函数
对第一个参数以外的参数也做类型检查,保证参数类型一致性
1 from functools import singledispatch 2 3 4 @singledispatch 5 def func(a, *args): 6 raise TypeError("暂不支持{}类型的执行".format(type(a))) 7 8 9 @func.register(int) 10 @func.register(str) 11 def _(a, *args): 12 print(">>【{}】类型检查通过".format(a)) 13 if not isinstance(args[0], type(a)): 14 raise TypeError(">>【{}】类型应为【{}】".format(args[0], type(a))) 15 print(">>【{}】类型检查通过".format(args[0])) 16 return a + args[0] 17 18 19 if __name__ == '__main__': 20 print(func(4, 5)) 21 print(func("只有", "永")) 22 print(func(4, "测试"))
output:
>>【4】类型检查通过
>>【5】类型检查通过
9
>>【只有】类型检查通过
>>【永】类型检查通过
只有永
>>【4】类型检查通过
TypeError: >>【测试】类型应为【<class 'int'>】
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:泛型 - Python技术站