如何强制开发者调用get_tax时仅使用预定义税率属性,禁止传入字面数值参数
实现方法
你要实现的核心是校验传入的税率参数的身份而非数值,因为仅对比数值的话,传入字面量0.23刚好和预定义税率值相等时会误通过,不符合你的要求。
基础实现:自定义税率类型
通过封装数值为自定义类实例,在函数中校验参数类型即可实现需求:
# 自定义税率类型,可进一步私有化构造方法禁止外部随意构造实例 class TaxRate: def __init__(self, value): self.value = value # 重载乘法运算符,无需修改原有乘法逻辑 def __mul__(self, other): return self.value * other __rmul__ = __mul__ class TaxRates: vat = TaxRate(0.23) pit = TaxRate(0.15) def get_tax(price, tax): if not isinstance(tax, TaxRate): raise TypeError("税率参数仅允许使用TaxRates中预定义的税率属性,禁止传入字面量数值") return price * tax # 测试 get_tax(1000, TaxRates.vat) # 正常返回230.0 get_tax(1000, 0.23) # 抛出TypeError
更优方案:枚举类实现(推荐)
使用Python标准库的enum枚举类实现更符合语义,也更安全:枚举类的成员是全局唯一的单例,无法被外部随意构造新的实例,天然适配预定义常量的场景。
from enum import Enum # 定义税率枚举,所有预定义税率都是该枚举的成员 class TaxRates(Enum): vat = 0.23 pit = 0.15 # 重载乘法运算符 def __mul__(self, other): return self.value * other __rmul__ = __mul__ def get_tax(price, tax): if not isinstance(tax, TaxRates): raise TypeError("税率参数仅允许使用TaxRates枚举中预定义的选项") return price * tax # 测试 get_tax(1000, TaxRates.vat) # 正常返回230.0 get_tax(1000, 0.23) # 抛出TypeError
方案优势
- 语义清晰:枚举本身就是用来定义一组固定可选常量的语法,其他开发者一眼就能看懂TaxRates中是所有可用的税率选项
- 安全性更高:无法在外部构造新的TaxRates枚举成员,彻底避免了自定义税率值传入的可能
- 扩展性好:后续新增税率仅需要在枚举类中加一行代码即可,无需修改其他校验逻辑
内容的提问来源于stack exchange,提问作者PaszaVonPomiot
相关产品推荐
相关产品推荐

