我有一个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的装饰模块 - 我认为你只需要@decorator
在定义之前添加一个sanitized
)