我发现自己一直在为数字运算编写相同的参数检查代码:
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
有没有更好的办法?有人告诉我不要对这些事情使用“断言”(尽管我不知道问题,除了不知道引起错误的变量的值).
编辑:为澄清起见,参数通常只是数字,并且错误检查条件可以是复杂的,不平凡的,并且不一定会在以后导致异常,而只会导致错误的结果. (不稳定的算法,无意义的解决方案等)
解决方法:
如果您使用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(“不要重复自己”)在软件开发的指导原则中排在首位.Python提供了合理的支持,可让您实现DRY并避免样板的,重复的代码!-)