如何在python中结合交互式提示和argparse?

时间:2012-06-11 04:25:14

标签: python argparse

我有一个包含大量潜在args的程序。 它基本上允许用户get/set/delete/list在一个非常小的数据库中的任何表。每列都有一个可以传递的arg 这不是主界面,只是偶尔有用的管理界面。

无论如何,如果用户想要在表中插入一行,他必须知道所有列的名称。虽然从帮助中很容易看出 - 但这也是一个痛苦的屁股。

所以,我已经更改了代码,因此用户可以传递--prompt选项,程序将提示他们输入与他们指定的表关联的每个列名。但我的问题是我现在需要在两个地方复制我的输入约束(选择,类型等) - argparse和这个提示功能。

  • 问题1:有更好的方法吗?
  • 问题2:为此目的,是否有一种干净的方法来引用argparse约束?
  • 问题3:是否有任何库推荐作为组合命令行参数,选项和提示的替代方案?

由于

1 个答案:

答案 0 :(得分:3)

执行此操作的典型方法是使用自定义操作。您可以使用argparse选项和使用相同函数验证的输入。

#!/usr/bin/env python3

import argparse


class ColumnsAction(argparse.Action):
    def __call__(self, parser, namespace, value, option_string=None):
        self.validate(parser, value)
        setattr(namespace, self.dest, value)

    @staticmethod
    def validate(parser, value):
        if value not in ('foo', 'bar'):
            parser.error('{} not valid column'.format(value))


parser = argparse.ArgumentParser()
parser.add_argument('--columns', action=ColumnsAction)
args = parser.parse_args()
if args.columns is None:
    args.columns = input('Enter columns: ')
    ColumnsAction.validate(parser, args.columns)
print(args.columns)