用argparse和python接受字典作为自变量

Question

我正在尝试使用argparse接受type = dict的参数，但是无论输入如何，都会给出无效dict值的错误。

#!/usr/bin/env python

import argparse

MYDICT = {'key': 'value'}

parser = argparse.ArgumentParser()
parser.add_argument("-m", "--mydict", action="store",
                    required=False, type=dict,
                    default=MYDICT)

args = parser.parse_args()

print args.mydict

这是我尝试将字典传递给脚本时发生的情况

./argp.py -m "{'key1': 'value1'}"
usage: argp.py [-h] [-m MYDICT]
argp.py: error: argument -m/--mydict: invalid dict value: "{'key1': 'value1'}"

查看这些文件，我认为这是可能的。

http://docs.python.org/dev/library/argparse.html

“任何支持in运算符的对象都可以作为options值传递，因此dict对象，set对象，自定义容器等都受支持。”

Answer 1

我不认为这是可能通过字典作为命令行参数，因为不从字符串存在一个转换功能的dict（ 编辑 ：黑客攻击 是
可能的，见下文）。您实质上要告诉python要做的是：

dict("{'key1': 'value1'}")

如果您在python控制台中尝试了该方法，将无法正常工作。

什么短语：

“任何支持in运算符的对象都可以作为options值传递，因此dict对象，set对象，自定义容器等都受支持。”

指的是可以通过 add_argument* 函数传递的 choices 参数，而不是 类型 参数。 ***

最好的选择是接受您的参数作为字符串，然后使用python的json功能将其转换：

parser.add_argument('-m', '--my-dict', type=str)
args = parser.parse_args()

import json
my_dictionary = json.loads(args.my_dict)

然后，您可以以字符串形式传递字典。您可以在python控制台中自行尝试json编码器/解码器，以了解其工作原理：

>>>json.loads('{"value1":"key1"}')
{u'value1': u'key1'}

编辑 ：hpaulj向我指出，您可以通过传递 json.loads 来“破解” 类型 参数。 __

import json
parser.add_argument('-d', '--my-dict', type=json.loads)
args = parse.parse_args()

mydict = args.my_dict  # Will return a dictionary

注意 ：您传递的输入格式是 不 一样的Python字典，但可能是为您的使用情况类似的就够了。

这项工作的原因实际上非常有趣，因为内部argparse只会将参数值用作函数来转换参数。也就是说，如果type =
int，则它将使用int（arg）；如果type = json.loads，则将使用json.loads（arg）

这也意味着您可以传递任何将单个参数作为参数的函数，以键入和执行自定义转换（如果需要）：