Zezwalanie na określone wartości argumentu Argparse

Question

Czy można wymagać, aby argument argparse był jedną z kilku ustawionych wartości?

Moje bieżące podejście polegałoby na ręcznym sprawdzeniu argumentu i jeśli nie jest to jedna z dozwolonych wartości, wywołaj print_help() i zakończ.

Oto obecna implementacja:

... 
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) 

To nie tak, że jest to szczególnie trudne, ale raczej, że wydaje się być brudny.

Answer 1

argparse argument może być ograniczone do określonych wartości z parametrem choices:

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

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

Zobacz docs więcej szczegółów.