在 argprse 中使用循环而不是想用 stop 命令退出

如何解决在 argprse 中使用循环而不是想用 stop 命令退出

我在 python 中使用带有 argparse 的 while 循环，如下所示。

parser.add_argument('-x','--height',help='Height of the Box')
parser.add_argument('-l','--length',type=int,help='Length of the Box')
parser.add_argument('-b','--breadth',help='Breadth of the Box')

exitcode = "stop"
args = ""
while args != exitcode:
     args = parser.parse_args(input("enter text: ").split())
     print (args.height)

当用户输入停止时，我希望程序退出。 但在这里它显示如下错误

错误：无法识别的参数：停止

我怎么能用stop退出这个

解决方法

在 argparse 中，您可以定义 Action 类，这些类可用于在传递参数后执行特定逻辑。 下面是一段代码，一旦 --stop 被传递，它就会执行退出。

from argparse import ArgumentParser,Action


class StopAction(Action):
    def __call__(self,parser,namespace,values,option_string=None):
        # Do whatever actions you want
        if values:
            print("Exiting")
            exit(0)


arg_parser = ArgumentParser()
arg_parser.add_argument('-x','--height',help='Height of the box')
arg_parser.add_argument('-l','--length',type=int,help='Length of the box')
arg_parser.add_argument('-b','--breadth',help='Breadth of the box')
arg_parser.add_argument('stop',nargs='?',action=StopAction,default=False)

while True:
    arg_parser.parse_args(input("enter text: ").split())

执行时如下所示：

± % python test.py    
enter text: -x 1
enter text: -x 2
enter text: stop
Exiting

,

在解析之前检查输入可能是一个解决方案：

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('-x',help='Height of the box')
parser.add_argument('-l',help='Length of the box')
parser.add_argument('-b',help='Breadth of the box')

while True:
    text = input("enter text: ")
    if text != "stop":
        args = parser.parse_args(text.split())
        print(args.height)
    else:
        print("Stop")
        break

试运行：

enter text: -x 1
1
enter text: -x 2
2
enter text: -l 1
None
enter text: -b 1
None
enter text: -x 1 -l 1 -b 1
1
enter text: stop
Stop