如何在 argparse 中排序

如何解决如何在 argparse 中排序

在 python argparse 中，我想对用户的输入进行排序。

例如在以下代码中，我希望用户输入 -x。如果用户输入任何其他内容，例如 -b，则应显示错误。

  <cellXfs count="2">
   <xf numFmtId="0" fontId="0" fillId="0" borderId="0" pivotButton="0" quotePrefix="0" xfId="0"/>
   <xf numFmtId="0" fontId="0" fillId="2" borderId="0" pivotButton="0" quotePrefix="0" xfId="0" applyFill="1"/>

解决方法

带输入的选项

如果你想使用输入功能，那么下面的解决方案是可能的（奇怪）：

import argparse


class StopAction(argparse.Action):
    def __call__(self,parser,namespace,values,option_string=None):
        parser.exit('Exit')

parser = argparse.ArgumentParser(description='SImple Example')    
parser.add_argument('-x','--height',type=int,help='Height of the box')
parser.add_argument('-l','--length',help='Length of the box')
parser.add_argument('-b','--breadth',help='Breadth of the box')
parser.add_argument('--stop',nargs=0,required=False,action=StopAction)


values = []
for i in ['-x','-l','-b']:
    i_value = input(f'enter {i} value (integer): ').strip()
    if i_value == '--stop':
        parser.parse_args([i_value])
    elif i_value in ['-h','--help']:
        parser.parse_args([i_value])
    else:
        # check for int
        parser.parse_args([i,i_value])
        values.append(i)
        values.append(i_value)

args = parser.parse_args(values)
# do something
print(args)

输出：

>test.py
enter -x value (integer): --stop
Exit

>test.py
enter -x value (integer): 1
enter -l value (integer): 2
enter -b value (integer): 3
Namespace(breadth=3,height=1,length=2,stop=None)

>test.py
enter -x value (integer): 1
enter -l value (integer): q
usage: test.py [-h] [-x HEIGHT] [-l LENGTH] [-b BREADTH] [--stop]
test.py: error: argument -l/--length: invalid int value: 'q'

---

带有 sys.argv 的选项

通常使用命令行界面如下（无输入）：

test.py -x 1 -l 2 -b 3

您可以使用 sys.argv 获取命令行参数列表。

sys.argv

传递给 Python 脚本的命令行参数列表。 argv[0] 是脚本名称（取决于操作系统是否为 完整路径名与否）。如果命令是使用 -c 执行的 解释器的命令行选项，argv[0] 设置为字符串 '-C'。如果没有脚本名称传递给 Python 解释器，则 argv[0] 是空字符串。

import argparse
import sys

# parser code here

# enable -h
args = parser.parse_args()

print(sys.argv)


if len(sys.argv) == 1:
    parser.error('First,set the height of the box.')
else:
    # throws an error if the arguments are out of order
    if sys.argv[1] not in ['-x','--height']:
        parser.error('First,set the height of the box.')
    else:
        # do something
        print(args)

输出：

>test.py -l 7 -b 4 -x 3
['test.py','7','-b','4','-x','3']
usage: test.py [-h] [-x HEIGHT] [-l LENGTH] [-b BREADTH] [--stop]
test.py: error: First,set the height of the box.

---

重要

参数的顺序无关紧要，它们的值可以通过 args.argumentname

获得

height = args.height

length = args.length

breadth = args.breadth

import argparse


# parser code here

# Namespace object
args = parser.parse_args()

if args.height is None: # not set
    parser.error('Set the height of the box.')
else:
    # do something
    print(args.height + 1)