的命令行选项限制值(Restricting values of command line optio

如何限制的argparse选项的值？

在下面的代码sau选项应仅接受若干0或1和bg应该只允许一个整数。 我怎样才能实现呢？

import os
import sys, getopt
import argparse

def main ():
    parser = argparse.ArgumentParser(description='Test script')
    parser.add_argument('-sau','--set',action='store',dest='set',help='<Required> Set flag',required=True)
    parser.add_argument('-bg','--base_g',action='store',dest='base_g',help='<Required> Base g',required=True)
    results = parser.parse_args() # collect cmd line args
    set = results.set
    base_g = results.base_g

if __name__ == '__main__':
    main()

Answer 1:

您可以使用类型和选择的参数add_argument 。 只接受“0”和“1”，你会怎么做：

parser.add_argument(…, choices={"0", "1"})

而只接受整数，你会怎么做：

parser.add_argument(…, type=int)

请注意，在选择，你得给你指定为类型参数类型的选项。 因此，要检查整数，并仅允许0和1，你会怎么做：

parser.add_argument(…, type=int, choices={0, 1})

例：

>>> import argparse
>>> parser = argparse.ArgumentParser()
>>> _ = parser.add_argument("-p", type=int, choices={0, 1})
>>> parser.parse_args(["-p", "0"])
Namespace(p=0)

文章来源: Restricting values of command line options