Ограничение значений параметров командной строки

Как ограничить значения параметров 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()

Ответ 1

Вы можете использовать аргументы типа и choices из add_argument. Чтобы принять только "0" и "1", вы должны сделать:

parser.add_argument(…, choices={"0", "1"})

А чтобы принимать только целые числа, вы должны сделать:

parser.add_argument(…, type=int)

Обратите внимание, что при выборе необходимо указать параметры в типе, который вы указали в качестве аргумента типа. Таким образом, чтобы проверить целые числа и разрешить только 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)