python - 访问传递给 argparse 参数的选项？

Question

是否可以访问传递给参数的选择元组？如果是这样，我该怎么做

例如如果我有

parser = argparse.ArgumentParser(description='choose location')
parser.add_argument(
    "--location",
    choices=('here', 'there', 'anywhere')
)
args = parser.parse_args()

我可以访问元组('here', 'there', 'anywhere')吗？

Answer 1

事实证明，parser.add_argument 实际上返回了关联的Action。您可以从中选择:

>>> import argparse
>>> parser = argparse.ArgumentParser(description='choose location')
>>> action = parser.add_argument(
...     "--location",
...     choices=('here', 'there', 'anywhere')
... )
>>> action.choices
('here', 'there', 'anywhere')

请注意，(据我所知)这在任何地方都没有记录，可能被视为“实现细节”，因此如有更改，恕不另行通知等。

也没有任何公开可访问的方法来获取添加后存储在 ArgumentParser 上的操作。我相信它们可以作为 parser._actions 使用，如果您愿意研究实现细节(并承担与此相关的任何风险)...

---

最好的办法是为位置选择创建一个常量，然后在您的代码中使用它:

LOCATION_CHOICES = ('here', 'there', 'anywhere')

parser = argparse.ArgumentParser(description='choose location')
parser.add_argument(
    "--location",
    choices=LOCATION_CHOICES
)
args = parser.parse_args()

# Use LOCATION_CHOICES down here...