Question

我正在尝试使用argparse来创建一个我在Unix控制台中输入的实例：

python getFood.py --food <(echo Bread) --calories yes

我已经实现了食物选项，并希望使用argparse添加卡路里是或否选项（二进制输入），这将决定是否从我导入的类调用卡路里方法。

我目前的代码主程序是：

parser = argparse.ArgumentParser(description='Get food details.')
parser.add_argument('--food', help='name of food to lookup', required=True, type=file)
args = parser.parse_args()

这成功地允许我使用上面显示的第一个食物选项返回食物详细信息。

基本上我想添加第二个二进制选项，如果用户指示为true，将调用另一个方法。有关如何编辑我的主例程argparse参数的任何帮助？我对argparse还很新。

Answer 1

如果action='store_true'未包含在内，您只需添加一个args.calories参数，默认--calories为False。为了进一步说明，如果用户添加--calories，则args.calories将设置为True。

parser = argparse.ArgumentParser(description='Get food details.')
# adding the `--food` argument

parser.add_argument('--food', help='name of food to lookup', required=True, type=file)
# adding the `--calories` argument
parser.add_argument('--calories', action='store_true', dest='calories', help='...')
# note: `dest` is where the result of the argument will go.
# as in, if `dest=foo`, then `--calories` would set `args.foo = True`.
# in this case, it's redundant, but it's worth mentioning.

args = parser.parse_args()

if args.calories:
    # if the user specified `--calories`, 
    # call the `calories()` method
    calories()
else:
    do_whatever()

但是，如果您要专门检查yes或no，请替换<{p>}中的store_true

parser.add_argument('--calories', action='store_true', dest='calories', help='...')

store，如下所示

parser.add_argument('--calories', action='store', dest='calories', type='str', help='...')

这将允许您稍后检查

if args.calories == 'yes':
    calories()
else:
    do_whatever()

请注意，在这种情况下，我添加了type=str，它将参数解析为字符串。由于您指定选项为yes或no，argparse实际上允许我们使用choices进一步指定可能输入的域：

parser.add_argument('--calories', action='store', dest='calories', type='str', 
                    choices=['yes', 'no'], help='...')

现在，如果用户在['yes', 'no']中输入任何而非，则会引发错误。

最后一种可能性是添加default，这样用户就不必一直指定某些标记：

parser.add_argument('--calories', action='store', dest='calories', type='str', 
                    choices=['yes', 'no'], default='no', help='...')

编辑：正如@ShadowRanger在评论中指出的那样，在这种情况下，dest='calories'，action='store'和type='str'是默认值，因此您可以省略它们：

parser.add_argument('--calories', choices=['yes', 'no'], default='no', help='...')

Python 2.7 Argparse是或否输入

1 个答案: