限制命令行選項的值

Question

如何限制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的type和choices參數。只接受「0」和「1」，你會怎麼做：

parser.add_argument(…, choices=set("01")) 

而且只接受整數，你會怎麼做：

parser.add_argument(…, type=int) 

注意，在選擇，你會必須提供您指定的類型作爲類型參數的選項。因此，要檢查整數和只允許0和1，你會怎麼做：

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

例子：

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