python – 用于健全性检查的循环函数参数

2019年8月4日 247次阅读

我有一个
Python函数,我正在对输入参数进行一些消毒：

def func(param1, param2, param3):
    param1 = param1 or ''
    param2 = param2 or ''
    param3 = param3 or ''

这适用于作为None而不是空字符串传递的参数.是否有更简单/更简洁的方法来循环函数参数以将这样的表达式应用于所有这些.我的实际功能有九个参数.

最佳答案 对于装饰者来说这看起来不错.这个怎么样：

def sanitized(func):
    def sfunc(*args, **kwds):
        return func(*[arg or '' for arg in args],
                    **dict((k, v or '') for k,v in kwds.iteritems()))
    sfunc.func_name = func.func_name
    sfunc.func_doc = func.func_doc
    return sfunc

你会在你的函数上使用它,如下所示：

@sanitized
def func(param1, param2, param3):
    print param1, param2, param3

如果参数将为空,则将替换为空字符串：

>>> func('foo', None, 'spam')
foo  spam

(请注意,这仍然会破坏功能签名,因为Ned Batchelder在他的回答中指出.为了解决这个问题,你可以使用Michele Simionato’s decorator module–我认为你只需要在清理定义之前添加一个@decorator)

点赞