Можно ли проверить выбор аргументов без учета регистра?
import argparse
choices = ["win64", "win32"]
parser = argparse.ArgumentParser()
parser.add_argument("-p", choices=choices)
print(parser.parse_args(["-p", "Win32"]))
приводит к:
usage: choices.py [-h] [-p {win64,win32}]
choices.py: error: argument -p: invalid choice: 'Win32' (choose from 'win64','win32')
Преобразуйте аргумент в нижний регистр, используя
type = lambda s : s.lower()
для переключателя -p.
Как отмечено chepner в комментариях, поскольку str.lower уже является подходящей функцией, обертка лямбда не обязательно нужна, и вы могли бы просто просто использовать
type = str.lower
непосредственно.
Использование lower в type - хороший способ сделать это, если вы не возражаете потерять информацию о случае.
Если вы хотите сохранить этот случай, вы можете определить собственный choices класс. Для choices нужны два метода: __contains__ (для тестирования in) и итерация (для перечисления вариантов).
class mylist(list):
# list subclass that uses lower() when testing for 'in'
def __contains__(self, other):
return super(mylist,self).__contains__(other.lower())
choices=mylist(['win64','win32'])
parser = argparse.ArgumentParser()
parser.add_argument("-p", choices=choices)
print(parser.parse_args(["-p", "Win32"]))
# Namespace(p='Win32')
Справка:
usage: ipython [-h] [-p {win64,win32}]
optional arguments:
-h, --help show this help message and exit
-p {win64,win32}
Хранение информации о случае также возможно с одним вкладышем:
type = lambda arg: {x.lower(): x for x in choices}[arg.lower()],
Где варианты выбора будут такими же, как и в параметре выбора.