使用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运算符的任何对象都可以作为选项值传递,因此支持dict对象,设置对象,自定义容器等."

Answer 1

我不认为在命令行中传递字典作为参数是不可能的,因为不存在从字符串到字典的转换函数(编辑:可能是黑客攻击,见下文).你基本上告诉python要做的是:

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

如果你在python控制台中试用它,哪个不起作用.

这句话是什么:

"支持in运算符的任何对象都可以作为选项值传递,因此支持dict对象,设置对象,自定义容器等."

是指可以使用add_argument函数传递的choices参数- 而不是type参数.

你最好的选择是接受你的参数作为字符串然后使用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

这个工作的原因实际上非常有趣,因为内部argparse只会使用参数值作为转换参数的函数.即如果type = int则它将使用int(arg)或者如果type = json.loads则使用json.loads(arg)

这也意味着您可以传递任何带有单个参数的函数作为参数来键入并执行自定义转换,如果您需要:)