Python ArgParse

Question

Я хочу использовать Python ArgParse для приема только определенных входных данных от пользователя.

Итак, в приведенном ниже примере предположим, что я хочу принять 'type1/type2/type3' только в качестве аргумента. Это возможно?

parser.add_argument('-t', '--type', type = str, help = 'type1/type2/type3') 

Answer 1

Используйте choices argument ограничить вход для ограниченного набора вариантов:

parser.add_argument('-t', '--type', choices=('type1', 'type2', 'type3'), 
        help='type1/type2/type3')