目前我希望 Python 的 argparse 模块只打印出 '1 - 65535' 而不是 {1, 2, 3, ... 65535},但文档似乎没有为此提供任何方法。有什么建议么?
问问题
775 次
1 回答
5
formatter_class
您可以通过设置选项来更改默认设置的格式。
我会继承 HelpFormatter 类来改变它格式化你的choices
值的方式。这个类正式是一个“实现细节”,但我怀疑它会随着新的 python 版本发生很大变化。
该_metavar_formatter
方法格式化{1, 2, ..., 65535}
字符串,您的子类可以覆盖它:
class RangeChoiceHelpFormatter(HelpFormatter):
def _metavar_formatter(self, action, default_metavar):
if action.metavar is not None:
result = action.metavar
elif action.choices is not None:
result = '{%s .. %s}' % (min(action.choices), max(action.choices])
else:
result = default_metavar
def format(tuple_size):
if isinstance(result, tuple):
return result
else:
return (result, ) * tuple_size
return format
另一种选择是不对如此大的范围使用choices
参数,而是定义一个新的参数类型。
这只是一个可调用的,传递了一个字符串,引发了argparse.ArgumentTypeError
,TypeError
或者ValueError
如果字符串不能转换为目标类型,或者转换后的值:
class IntRange(object):
def __init__(self, start, stop=None):
if stop is None:
start, stop = 0, start
self.start, self.stop = start, stop
def __call__(self, value):
value = int(value)
if value < self.start or value >= self.stop:
raise argparse.ArgumentTypeError('value outside of range')
return value
您可以将其用作参数类型,如下所示:
parser.add_argument('foo', type=IntRange(1, 65536))
并调整您的帮助消息以指示可以接受的值。
于 2012-06-30T08:51:56.207 回答