python - 检查数字 Python 代码中的参数

Question

我发现自己一直在为数字运算编写相同的参数检查代码:

def myfun(a, b):
    if a < 0:
        raise ValueError('a cannot be < 0 (was a=%s)' % a)
    # more if.. raise exception stuff here ...
    return a + b

有没有更好的方法？我被告知不要对这些事情使用“断言”(尽管我没有看到问题，除了不知道导致错误的变量的值)。

编辑:需要澄清的是，参数通常只是数字，错误检查条件可能很复杂、非常重要，并且不一定会在以后导致异常，而只会导致错误的结果。 (不稳定的算法，无意义的解决方案等)

Answer 1

如果您使用 python -O 运行，

assert 会被优化掉(适度的优化，但有时很高兴拥有)。如果您有经常重复的模式，一个更好的选择可能是使用装饰器——消除重复的好方法。例如，假设您有无数个函数，必须按位置(而不是按关键字)使用参数调用，并且它们的第一个参数必须为正数；然后……:

def firstargpos(f):
  def wrapper(first, *args):
    if first < 0:
      raise ValueError(whateveryouwish)
    return f(first, *args)
  return wrapper

然后你说这样的话:

@firstargpos def myfun(a, b): ...

并且检查是在装饰器(或者更确切地说是它返回的包装闭包)中一劳永逸地执行的。所以，唯一棘手的部分是弄清楚你的函数需要什么检查以及如何最好地调用装饰器来表达这些(很难说，没有看到你定义的函数集和每个需要的检查集!-)。请记住，DRY(“Don't Repeat Yourself”)接近软件开发指导原则中的头把交椅，Python 有合理的支持让您实现 DRY 并避免样板、重复的代码!-)