带有编号参数名称的 Python argparse

如何解决带有编号参数名称的 Python argparse

GNU grep 有一个参数可以在匹配行周围打印一定数量的额外行。来自手册页：

-C NUM,-NUM,--context=NUM 打印输出上下文的 NUM 行。在连续的组之间放置包含组分隔符 (--) 的行 火柴。使用 -o 或 --only-matching 选项，这不起作用并且 发出警告。

所以我可以做类似的事情

grep -5 pattern file.txt

相当于

grep -C 5 pattern file.txt

如何使用 argparse 模拟 -NUMBER 行为到用户可以传入 -NUM 的位置，并且我可以轻松地从 argparse 获取该数字。

一个愚蠢的做法是做这样的事情：

parser = argparse.ArgumentParser()
for i in range(1,10000):
    parser.add_argument(f'-{i}',action='store_true',dest=f'border_lines_{i}',help=argparse.SUPPRESS)

args = parser.parse_args()

border_line_count = 0
for i in range(1,10000):
    if getattr(args,f'border_lines_{i}'):
        border_line_count = i
        break

基本上，我为不同的 -NUM 添加了一堆隐藏参数。然后找到设置好的那个。有没有更好的办法？

解决方法

这是一种实现方式，但我个人不喜欢它。

parser.add_argument('-',dest='border_lines')

会给你你想要的基本行为。请注意，这是一个必需参数，如果您希望这是一个可选参数，您可以：

parser.add_argument('-',dest='border_lines',default=0)

现在有一个（奇怪的）问题是它只接受个位数。要解决此问题，您需要定义并覆盖解析操作。

 class ConcatAction(argparse.Action):
     def __call__(self,parser,namespace,values,option_string):
         value = int(''.join(values))
         setattr(namespace,'border_lines',value)

现在您可以将其用作：

parser = argparse.ArgumentParser()
parser.add_argument('-',default=0,nargs='*',action=ConcatAction)
parser.parse_args('-555')  # Gives Namespace(border_lines=555)

我强烈认为这会与其他参数混淆，我还没有测试过。我再说一遍，我不喜欢这种方式，遵循 python zen，我们应该明确（使用 arg 名称）并尽可能清楚。