pyparsing的嵌套字典输出

时间:2014-08-11 08:21:47

标签: python pyparsing s-expression

我正在使用pyparsing来解析表单的表达式:

"and(or(eq(x,1), eq(x,2)), eq(y,3))"

我的测试代码如下:

from pyparsing import Word, alphanums, Literal, Forward, Suppress, ZeroOrMore, CaselessLiteral, Group

field = Word(alphanums)
value = Word(alphanums)
eq_ = CaselessLiteral('eq') + Group(Suppress('(') + field + Literal(',').suppress() + value + Suppress(')'))
ne_ = CaselessLiteral('ne') + Group(Suppress('(') + field + Literal(',').suppress() + value + Suppress(')'))
function = ( eq_ | ne_ )

arg = Forward()
and_ = Forward()
or_ = Forward()

arg << (and_ | or_ |  function) + Suppress(",") + (and_ | or_ | function) + ZeroOrMore(Suppress(",") + (and_ | function))

and_ << Literal("and") + Suppress("(") + Group(arg) + Suppress(")")
or_ << Literal("or") + Suppress("(") + Group(arg) + Suppress(")")

exp = (and_ | or_ | function)

print(exp.parseString("and(or(eq(x,1), eq(x,2)), eq(y,3))"))

我的输出格式为:

['and', ['or', ['eq', ['x', '1'], 'eq', ['x', '2']], 'eq', ['y', '3']]]

列表输出看起来没问题。但是对于后续处理,我希望以嵌套字典的形式输出:

{
    name: 'and',
    args: [
        {
            name: 'or',
            args: [
                {
                    name: 'eq',
                    args: ['x','1']
                },
                {
                    name: 'eq',
                    args: ['x','2']
                }
            ]
        },
        {
            name: 'eq',
            args: ['y','3']
        }
    ]
}

我尝试了Dict课程,但没有成功。

在pyparsing中可以做到吗?或者我应该手动格式化列表输出?

2 个答案:

答案 0 :(得分:7)

您正在寻找的功能是pyparsing中的重要功能,即设置结果名称的功能。对于大多数pyparsing应用程序,建议使用结果名称。此功能自0.9版以来一直存在,如

expr.setResultsName("abc")

这允许我以res["abc"]res.abc(其中res是从parser.parseString返回的值)访问整个已解析结果的特定字段。您也可以致电res.dump()查看结果的嵌套视图。

但是仍然注意保持解析器一目了然,我在1.4.6中添加了对这种形式的setResultsName的支持:

expr("abc")

这是你的解析器,有一点清理,并添加了结果名称:

COMMA,LPAR,RPAR = map(Suppress,",()")
field = Word(alphanums)
value = Word(alphanums)
eq_ = CaselessLiteral('eq')("name") + Group(LPAR + field + COMMA + value + RPAR)("args")
ne_ = CaselessLiteral('ne')("name") + Group(LPAR + field + COMMA + value + RPAR)("args")
function = ( eq_ | ne_ )

arg = Forward()
and_ = Forward()
or_ = Forward()
exp = Group(and_ | or_ | function)

arg << delimitedList(exp)

and_ << Literal("and")("name") + LPAR + Group(arg)("args") + RPAR
or_ << Literal("or")("name") + LPAR + Group(arg)("args") + RPAR

不幸的是,dump()只处理结果的嵌套,而不是值列表,因此它不如json.dumps那么好(可能这对转储是一个很好的增强?)。所以这是一个自定义方法来转储你的嵌套name-args结果:

ob = exp.parseString("and(or(eq(x,1), eq(x,2)), eq(y,3))")[0]

INDENT_SPACES = '    '
def dumpExpr(ob, level=0):
    indent = level * INDENT_SPACES
    print (indent + '{')
    print ("%s%s: %r," % (indent+INDENT_SPACES, 'name', ob['name']))
    if ob.name in ('eq','ne'):
        print ("%s%s: %s"   % (indent+INDENT_SPACES, 'args', ob.args.asList()))
    else:
        print ("%s%s: ["   % (indent+INDENT_SPACES, 'args'))
        for arg in ob.args:
            dumpExpr(arg, level+2)
        print ("%s]"   % (indent+INDENT_SPACES))
    print (indent + '}' + (',' if level > 0 else ''))
dumpExpr(ob)

,并提供:

{
    name: 'and',
    args: [
        {
            name: 'or',
            args: [
                {
                    name: 'eq',
                    args: ['x', '1']
                },
                {
                    name: 'eq',
                    args: ['x', '2']
                },
            ]
        },
        {
            name: 'eq',
            args: ['y', '3']
        },
    ]
}

答案 1 :(得分:2)

我不认为pyparsing有类似的东西,但你可以递归地创建数据结构:

def toDict(lst):
    if not isinstance(lst[1], list):
        return lst
    return [{'name': name, 'args': toDict(args)}
            for name, args in zip(lst[::2], lst[1::2])]

您的示例在args个孩子的数量上表现不同。如果它只是一个你只使用dict,否则它是一个dicts列表。这将导致复杂的使用。即使只有一个孩子,最好使用一个dicts列表。这样,您总是知道如何在没有类型检查的情况下迭代子项。

实施例

我们可以使用json.dumps来打印输出(请注意,我们在这里打印parsedict[0],因为我们知道root有一个子节点,但我们总是返回前面指定的列表):

import json
parsed = ['and', ['or', ['eq', ['x', '1'], 'eq', ['x', '2']], 'eq', ['y', '3']]]
parsedict = toDict(parsed)
print json.dumps(parsedict[0], indent=4, separators=(',', ': '))

输出

{
    "name": "and",
    "args": [
        {
            "name": "or",
            "args": [
                {
                    "name": "eq",
                    "args": [
                        "x",
                        "1"
                    ]
                },
                {
                    "name": "eq",
                    "args": [
                        "x",
                        "2"
                    ]
                }
            ]
        },
        {
            "name": "eq",
            "args": [
                "y",
                "3"
            ]
        }
    ]
}

要获得该输出,我将dict替换为toDict功能区中的collections.OrderedDict,只是为了name args之前的{{1}}。