Consentire valori specifici per un argomento Argparse

Question

È possibile richiedere che un argomento argparse sia uno dei pochi valori preimpostati?

Il mio approccio attuale sarebbe esaminare l'argomento manualmente e se non è uno dei valori consentiti chiamare print_help() e uscire.

Ecco l'implementazione attuale:

... 
parser.add_argument('--val', dest='val', action='store', 
         help='Special testing value') 

args = parser.parse_args(sys.argv[1:]) 
if args.val not in ['a','b','c']: 
    parser.print_help() 
    sys.exit(1) 

Non è che questo è particolarmente difficile, ma che sembra essere disordinato.

Answer 1

Un argomento argparse può essere limitata ai valori specifici con il parametro choices:

... 
parser.add_argument('--val', dest='val', action='store', 
         choices=['a','b','c'], 
         help='Special testing value') 

args = parser.parse_args(sys.argv[1:]) 

Vedere la docs per maggiori dettagli.