python - 不区分大小写的 argparse 选择不会丢失选择列表中的大小写信息

Question

我有一个 very similar question但是提供的解决方案不适合，因为我的用例略有不同。为了简单起见，我将根据已经回答的问题调整我的问题。

是否可以检查argparse choices以不区分大小写的方式，不丢失区分大小写的信息？

import argparse
choices = ["ADASYN", "BorderlineSMOTE", "KMeansSMOTE"]
parser = argparse.ArgumentParser()
parser.add_argument("-p", choices=choices)
print(parser.parse_args(["-p", "adasyn"]))

这导致:

choices.py: error: argument -p: invalid choice: 'adasyn' (choose from 'ADASYN', 'BorderlineSMOTE', 'KMeansSMOTE')

我显然可以使用 type.lower，但是我的选择列表需要小写，并且我失去了很多 -h(帮助输出)的可读性。另一方面，我希望我的用户能够使用不同的大小写，只要两者的字符串 .lower() 相同即可。

solution provided @hpaulj(类似问题)在我的情况下不起作用。它只允许参数输入具有任意大小写。因此，如果我的选择列表是小写的，AdAsyN 会起作用，但如果我将选择列表保留为 CamelCase，则对于参数，选择列表中的大写字母也必须是大写的。

有没有人能为我提供解决方案，让我可以保留选择列表的大小写，同时完全忽略用户输入的大小写？

Answer 1

您可以自定义类型函数以在选择 列表中执行搜索而不是简单的类型转换。这看起来有点老套，但工作正常。

def make_type(choices):
    def find_choice(choice):
        for key, item in enumerate([choice.lower() for choice in choices]):
            if choice.lower() == item:
                return choices[key]
        else:
            return choice
    return find_choice

choices = ["ADASYN", "BorderlineSMOTE", "KMeansSMOTE"]    
parser = argparse.ArgumentParser()
parser.add_argument("-p", choices=choices, type=make_type(choices))


print(parser.parse_args("-p adasyn".split()))
# Namespace(p='ADASYN')

print(parser.parse_args("-p ADASYN".split()))
# Namespace(p='ADASYN')

print(parser.parse_args("-p aDaSyN".split()))
# Namespace(p='ADASYN')

print(parser.parse_args("-p adasyn1".split()))
# usage: argsaction.py [-h] [-p {ADASYN,BorderlineSMOTE,KMeansSMOTE}]
# argsaction.py: error: argument -p: invalid choice: 'adasyn1' (choose from 'ADASYN', 'BorderlineSMOTE', 'KMeansSMOTE')

取自 answer 的闭包函数.
答案基于@hpaulj 评论。