如何限制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()
答案 0 :(得分:31)
您可以使用add_argument
的{{3}}和type参数。要只接受'0'和'1',你可以:
parser.add_argument(…, choices={"0", "1"})
只接受整数,你可以这样做:
parser.add_argument(…, type=int)
请注意,在选择中,您必须以您指定的类型为type-argument提供选项。因此,要检查整数和仅允许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)