将元组作为命令行参数传递

Question

我的要求是将元组作为命令行参数传递，例如

--data (1,2,3,4)

我尝试使用argparse模块，但如果我像这样传递，它会作为字符串'(1,2,3,4)' 。 我尝试为argparse.add_argument提供type=tuple ，但在这里没有用。

我是否必须添加一个新类型类并将其传递给add_argument类型参数？

更新

我根据答案尝试了ast.literal_eval 。 感谢那。 但它在结果中给出了空格，如下所示。

(1,2,3,4)
<type 'str'>
(1, 2, 3, 4)
<type 'tuple'>

Answer 1

集nargs所述的data参数nargs="+" （意思是一个或多个）和类型int ，可以再这样设置的参数在命令行上：

--data 1 2 3 4

args.data现在将是[1, 2, 3, 4] 。

如果你必须有一个元组，你可以这样做：

my_tuple = tuple(args.data)

把它们放在一起：

parser = argparse.ArgumentParser()
parser.add_argument('--data', nargs='+', type=int)
args = parser.parse_args()
my_tuple = tuple(args.data)

Answer 2

我有同样的要求。 在标准包中找不到。 所以，推出了这个。 蟒蛇 3 倍。 放入简单的 argv 解析器，一次调用一个 arg：

• var是默认传递的，类型正确。
• argval是找到命令行字符串。 逗号分隔，没有空格。 即 1、2、3 或一、二、三。

与 bash shell 一起工作，可能需要引号，其他 shell 的其他调整，操作系统的。 相同的技术适用于非列表类型。 此版本基于 var[0] 创建同构列表或元组。

例子：

./foo.py   --mytuple 1,2,3,4,5

代码：

def chk_arg( var, argstr, dbg=False): 
    ### loop on sys.argv here .....
    # if list/tuple type, use list comprehension
    # converts csv list to list or tuple 'type(var)' of 'type(var[0])' items. 
    if type(var) is list or type(var) is tuple:  
        val = type(var)([type(var[0])(i) for i in argval.split(',')]) 

Answer 3

如果你想扩展它，而不是处理语法树，并在参数中提供元组转换，你需要一个自定义的 argparse 类型。 （我需要这个，因为我的解析器在一个包中，我不想要求我的用户都写my_tuple = tuple(args.data) 。使用参数类型，您还可以像原始一样接收带括号的字符串问题。另外，我不会包括nargs ，因为这会将您的自定义类型输出包装到一个列表中，这基本上违背了您的自定义类型的目的。我映射了一个浮点数，因为我的输入类似于“0.9, 0.99”，但是你的 int 工作原理相同：

def tuple_type(strings):
    strings = strings.replace("(", "").replace(")", "")
    mapped_int = map(int, strings.split(","))
    return tuple(mapped_int)

parser = argparse.ArgumentParser()
parser.add_argument('--data', type=tuple_type)
args = parser.parse_args()

这样你就可以发送--data "(1,2,3,4)"或--data "1,2,3,4"