我有一个Python程序,它只能是某些字符串,标志rock,paper或scissors.Python argparse有一个很好的方法来实现它,使用choices容器的参数允许值.
以下是文档中的示例:
import argparse
...
parser = argparse.ArgumentParser(prog='game.py')
parser.add_argument('move', choices=['rock', 'paper', 'scissors'])
parser.parse_args(['rock'])
### Namespace(move='rock')
parser.parse_args(['fire'])
### usage: game.py [-h] {rock,paper,scissors}
### game.py: error: argument move: invalid choice: 'fire' (choose from 'rock','paper', 'scissors')
Run Code Online (Sandbox Code Playgroud)
我想实现choices这样的选择不区分大小写,即用户可以输入RoCK并且它仍然有效.
这样做的标准方法是什么?
你可以设置type=str.lower.
import argparse
parser = argparse.ArgumentParser(prog='game.py')
parser.add_argument('move', choices=['rock', 'paper', 'scissors'], type=str.lower)
parser.parse_args(['rOCk'])
# Namespace(move='rock')
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
2599 次 |
| 最近记录: |