在Python中操作嵌套的布尔查询字符串

Question

我有这样的字符串布尔查询

   queryString= """And(
                      OR(abc,xyz,wxy),
                      AND(AND(xyz,wxy),xzy),
                      XOR(x1,y1, AND(xy,zz))  
                      )"""

目前，我很难像我想的那样修改上面的查询字符串

1. 在最后一个XOR中添加另一个OR(x3,y3)
2. 删除整个OR(abc,xyz,wxy)

具有所需输出

   resultQueryString= """And(                        
                            AND(AND(xyz,wxy),xzy),
                            XOR(x1,y1, AND(xy,zz),OR(x3,y3))  
                            )"""

我想除非我为每个不同的查询提供一个复杂的正则表达式，否则我很难做到这一点。你知道吗

我正在尝试编写一个python函数，它将上面的字符串布尔查询作为输入并输出一个树数据结构。你知道吗

这样我就可以遍历树并计算或更改查询的任何部分。你知道吗

在上面的例子中，如果我把它作为一个树，我可以很容易地看到根是AND，并遍历/修改其他分支等等。你知道吗

Answer 1

ast.parse函数似乎几乎完全满足您的要求：

ast.dump(ast.parse("""And(                        
                        AND(AND(xyz,wxy),xzy),
                        XOR(x1,y1, AND(xy,zz),OR(x3,y3))  
                        )""").body[0].value)

Call(func=Name(id='And', ctx=Load()), args=[Call(func=Name(id='AND', ctx=Load()), args=[Call(func=Name(id='AND', ctx=Load()), args=[Name(id='xyz', ctx=Load()), Name(id='wxy', ctx=Load())], keywords=[], starargs=None, kwargs=None), Name(id='xzy', ctx=Load())], keywords=[], starargs=None, kwargs=None), Call(func=Name(id='XOR', ctx=Load()), args=[Name(id='x1', ctx=Load()), Name(id='y1', ctx=Load()), Call(func=Name(id='AND', ctx=Load()), args=[Name(id='xy', ctx=Load()), Name(id='zz', ctx=Load())], keywords=[], starargs=None, kwargs=None), Call(func=Name(id='OR', ctx=Load()), args=[Name(id='x3', ctx=Load()), Name(id='y3', ctx=Load())], keywords=[], starargs=None, kwargs=None)], keywords=[], starargs=None, kwargs=None)], keywords=[], starargs=None, kwargs=None)

（.body[0].value删除了两个毫无意义的抽象层，.dump只用于输出。你知道吗

下面是执行您对输出请求的转换的代码：

class Filterer(ast.NodeTransformer):
    def visit_Call(self, node):
            name=node.func.id
            if name == "OR" and len(node.args) == 3:
                    return None
            elif name == "XOR":
                    args = [ast.Name("x3",ast.Load()),
                            ast.Name("y3",ast.Load())]
                    func = ast.Name("OR",ast.Load())
                    node.args.append(ast.Call(func, args, [], None, None))
            return self.generic_visit(node)

下面是以您的格式打印结果的代码，除了空格：（Python在它的ast模块中没有用于此目的的内置代码）：

class Printer(ast.NodeVisitor):
    def visit_Call(self, node):
            self.visit(node.func)
            print("(",end="")
            comma = False
            for arg in node.args:
                    if comma:
                            print(",",end="")
                    comma=True
                    self.visit(arg)
            print(")",end="")
    def visit_Name(self, node):
            print(node.id,end="")

因此，最终代码为：

Printer().visit(Filterer().visit(ast.parse(queryString)))