解析Python中的嵌套函数

Bay*_*ram 2 python parsing nested-function

line = "add(multiply(add(2,3),add(4,5)),1)"

def readLine(line):
    countLeftBracket=0
    string = ""
    for char in line:
        if char !=")":
            string += char
        else:
            string +=char
            break

    for i in string:
        if i=="(":
            countLeftBracket+=1

    if countLeftBracket>1:
        cutString(string)
    else:
        return execute(string)

def cutString(string):
    countLeftBracket=0

    for char in string:
        if char!="(":
            string.replace(char,'')
        elif char=="(":
            string.replace(char,'')
            break

    for char in string:
        if char=="(":
            countLeftBracket+=1

    if countLeftBracket>1:
        cutString(string)
    elif countLeftBracket==1:
        return execute(string)

def add(num1,num2):
    return print(num1+num2)

def multiply(num1,num2):
    return print(num1*num2)

readLines(line)
Run Code Online (Sandbox Code Playgroud)

我需要执行整个行字符串。我尝试将括号内的每个函数一一剪切并用结果替换它们,但我有点迷失了。不知道如何继续,我的代码收到错误:

  File "main.py", line 26, in cutString                                                                                 
    if char!="(":                                                                                                       
RuntimeError: maximum recursion depth exceeded in comparison 
Run Code Online (Sandbox Code Playgroud)

请告诉我要移动到哪里、使用哪种方法?

the*_*orn 5

这是一个使用 pyparsing 的解决方案,因此更容易扩展:

from pyparsing import *
Run Code Online (Sandbox Code Playgroud)

第一个方便函数(使用第二个标签函数并打印解析树以了解原因)

def tag(name):
    """This version converts ["expr", 4] => 4
       comment in the version below to see the original parse tree
    """
    def tagfn(tokens):
        tklist = tokens.asList()
        if name == 'expr' and len(tklist) == 1:
            # LL1 artifact removal
            return tklist
        return tuple([name] + tklist)
    return tagfn

# def tag(name):
#     return lambda tokens: tuple([name] + tokens.asList())
Run Code Online (Sandbox Code Playgroud)

我们的词法分析器不需要识别左右括号、整数和名称。这是使用 pyparsing 定义它们的方式:

LPAR = Suppress("(")
RPAR = Suppress(")")
integer = Word(nums).setParseAction(lambda s,l,t: [int(t[0])])
name = Word(alphas)
Run Code Online (Sandbox Code Playgroud)

我们的解析器有函数调用,它接受零个或多个表达式作为参数。函数调用也是一个表达式,因此为了处理循环性,我们必须向前声明 expr 和 fncall:

expr = Forward()
fncall = Forward()

expr << (integer | fncall).setParseAction(tag('expr'))
fnparams = delimitedList(expr)

fncall << (name + Group(LPAR + Optional(fnparams, default=[]) + RPAR)).setParseAction(tag('fncall'))
Run Code Online (Sandbox Code Playgroud)

现在我们可以解析字符串(我们也可以向函数添加空格以及多于或少于两个参数):

line = "add(multiply(add(2,3),add(4,5)),1)"
res = fncall.parseString(line)
Run Code Online (Sandbox Code Playgroud)

要查看返回的内容,您可以打印它,这称为解析树(或者,因为我们的标记函数已简化它,所以称为抽象语法树):

import pprint
pprint.pprint(list(res))
Run Code Online (Sandbox Code Playgroud)

其输出:

[('fncall',
  'add',
  [('fncall',
    'multiply',
    [('fncall', 'add', [2, 3]), ('fncall', 'add', [4, 5])]),
   1])]
Run Code Online (Sandbox Code Playgroud)

使用注释掉的标签函数将会是(这只是更多的工作需要处理,没有任何额外的好处):

[('fncall',
  'add',
  [('expr',
    ('fncall',
     'multiply',
     [('expr', ('fncall', 'add', [('expr', 2), ('expr', 3)])),
      ('expr', ('fncall', 'add', [('expr', 4), ('expr', 5)]))])),
   ('expr', 1)])]
Run Code Online (Sandbox Code Playgroud)

现在定义我们的程序可用的函数:

FUNCTIONS = {
    'add': lambda *args: sum(args, 0),
    'multiply': lambda *args: reduce(lambda a, b: a*b, args, 1),
}

# print FUNCTIONS['multiply'](1,2,3,4)   # test that it works ;-)
Run Code Online (Sandbox Code Playgroud)

我们的解析器现在编写起来非常简单:

def parse(ast):
    if not ast:  # will not happen in our program, but it's good practice to exit early on no input
        return

    if isinstance(ast, tuple) and ast[0] == 'fncall':
        # ast is here ('fncall', <name-of-function>, [list-of-arguments])
        fn_name = ast[1]          # get the function name
        fn_args = parse(ast[2])   # parse each parameter (see elif below)
        return FUNCTIONS[fn_name](*fn_args)  # find and apply the function to its arguments
    elif isinstance(ast, list):
        # this is called when we hit a parameter list
        return [parse(item) for item in ast]
    elif isinstance(ast, int):
        return ast
Run Code Online (Sandbox Code Playgroud)

现在根据词法分析阶段的结果调用解析器:

>>> print parse(res[0])  # the outermost item is an expression
46
Run Code Online (Sandbox Code Playgroud)