相关疑难解决方法(0)

可argparse用于验证文件名cmd行参数的文件扩展名？

例如,如果我有一个python脚本,我从cmd行运行:

$ script.py file.csv
$ script.py file.tab
$ script.py file.txt

我希望argparse接受前两个文件名cmd行选项,但拒绝第三个

我知道你可以这样做:

parser = argparse.ArgumentParser()
parser.add_argument("fn", choices=["csv","tab"])
args = parser.parse_args()

为cmd行选项指定两个有效选项

我想要的是这个:

parser.add_argument("fn", choices=["*.csv","*.tab"])

为cmd行选项指定两个有效的文件扩展名.不幸的是,这不起作用 - 是否有办法实现这一目的argparse？