Python：使用 argparse 解析多种数据类型

Question

我尝试使用 argparse 来了解它是如何解析给定列表的：

parser = argparse.ArgumentParser()

parser.add_argument('--ls', nargs='*', type=str, default = [])
Out[92]: _StoreAction(option_strings=['--ls'], dest='ls', nargs='*', const=None, default=[], type=<type 'str'>, choices=None, help=None, metavar=None)

args = parser.parse_args("--ls 'tomato' 'jug' 'andes'".split())

args
Out[94]: Namespace(ls=["'tomato'", "'jug'", "'andes'"])

args.ls
Out[96]: ["'tomato'", "'jug'", "'ande'"]

args.ls[0]
Out[97]: "'tomato'"

eval(args.ls[0])
Out[98]: 'tomato'

Q1：上述方法可行，但有没有更好的方法来访问列表中的值？

然后我尝试用字典来解析给定的字典：

dict_parser = argparse.ArgumentParser()
dict_parser.add_argument('--dict', nargs='*',type=dict,default={})

Out[104]: _StoreAction(option_strings=['--dict'], dest='dict', nargs='*', const=None, default={}, type=<type 'dict'>, choices=None, help=None, metavar=None)

arg2 = dict_parser.parse_args("--dict {'name':'man', 'address': 'kac', 'tags':'don'}")
usage: -c [-h] [--dict [DICT [DICT ...]]]
-c: error: unrecognized arguments: - - d i c t   { ' n a m e ' : ' m a n' ,   ' a d d r e s s ' :   ' k a c' ,   ' t a g s ' : ' d o n ' }
To exit: use 'exit', 'quit', or Ctrl-D.
An exception has occurred, use %tb to see the full traceback.

SystemExit: 2

那是行不通的。 Q2：以上对字典有什么作用？

Q3：现在我想要

python my.py --ls tomato jug andes  --dict {'name':'man', 'address': 'kac', 'tags':'don'}

待解析

我该怎么做？

我提到了http://parezcoydigo.wordpress.com/2012/08/04/from-argparse-to-dictionary-in-python-2-7/

...并发现在字典下分配所有内容非常有用。有人可以简化此任务以解析参数中的多种数据类型吗？

Answer 1

import ast
dict_parser.add_argument('--dict', nargs='*',type=ast.literal_eval,default={})
args = dict_parser.parse_args(["--dict", "{'name':'man', 'address': 'kac', 'tags':'don'}"])

Answer 2

parser.add_argument('--ls', nargs='*', type=str, default = []) Q1：上述方法可行，但有没有更好的方法来访问列表中的值？

我经常认为“越简单越好”，这是一种非常简单的做事方式，我想说没有更好的方法。但是，还有其他方法。

dict_parser.add_argument('--dict', nargs='*',type=dict,default={}) arg2 = dict_parser.parse_args("--dict {'name':'man', 'address': 'kac', 'tags':'don'}") Q2：以上对字典有什么作用？

我建议你使用 json 解析你的字符串：

>>> class FromJSON():
...     def __init__(self, string):
...         self.string = string
...     def decode(self):
...         return json.loads(self.string)
...    
>>> dict_parser.add_argument('--dict',type=FromJSON)
>>> arg2 = dict_parser.parse_args(['--dict', '{"name":"man", "address": "kac", "tags":"don"}'])

虽然 JSON 很像 python，但它不是 python。它对引号（值仅由 双引号 包围）和尾随逗号（没有）很挑剔。但至少它对代码注入是安全的！当然你应该用引号括起你的参数

您可能有另一个解决方案来将 dict 作为参数，那就是去掉括号：

>>> parser.add_argument('--dict', nargs='*', type=str, default = [])
>>> args = parser.parse_args(['--dict', 'name:man', 'address:kac', 'tags:don'])
>>> args.dict = dict([arg.split(':') for arg in args.dict])
>>> print args.dict
{'tags': 'don', 'name': 'man', 'address': 'kac'}

Q3：现在我想要

python my.py --ls tomato jug andes  --dict {'name':'man', 'address': 'kac', 'tags':'don'}

待解析

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--dict', type=FromJSON, default=FromJSON("{}"))
>>> parser.add_argument('--ls', nargs='*', type=str, default = [])
>>> args = parser.parse_args(['--ls', 'tomato', 'jug', 'andes', '--dict', '{"name":"man", "address": "kac", "tags":"don"}'])
>>> args.ls
['tomato', 'jug', 'andes']
>>> args.dict
<__main__.FromJSON instance at 0x7f932dd20c20>
>>> args.dict.decode()
{u'tags': u'don', u'name': u'man', u'address': u'kac'}

关于FromJSON()，可以改进一下是这样的：

class JsonToDict():
    def __call__(self, string):
        return json.loads(string)

您可以按如下方式使用：

dict_parser.add_argument('--dict',type=JsonToDict())

HTH