python - 在 Python 中拆分可能嵌套的函数表达式列表

Question

PostgreSQL 允许在表达式上创建索引，例如，CREATE INDEX ON films ((lower(title)))。它还具有一个 pg_get_expr() 信息函数，可将表达式的内部格式转换为文本，即前一个示例中的 lower(title)。表情有时会变得很毛茸茸。以下是一些示例(使用 Python):

sample_exprs = [
    'lower(c2)',
    'lower(c2), lower(c3)',
    "btrim(c3, 'x'::text), lower(c2)",
    "date_part('month'::text, dt), date_part('day'::text, dt)",
    '"substring"(c2, "position"(c2, \'_begin\'::text)), "substring"(c2, "position"(c2, \'_end\'::text))',
    "(((c2)::text || ', '::text) || c3), ((c3 || ' '::text) || (c2)::text)",
    'f1(f2(arga, f3()), arg1), f4(arg2, f5(argb, argc)), f6(arg3)']

最后一项并非真正来自 Postgres，而只是我的代码应该处理的一个极端示例。

我编写了一个 Python 函数来将文本列表拆分为组件表达式。例如，最后一项被分解为:

 f1(f2(arga, f3()), arg1)
 f4(arg2, f5(argb, argc))
 f6(arg3)

我尝试了 str 方法，比如 find() 和 count() 也考虑了正则表达式，但最后我写了一个函数这就是我用 C 编写的内容(基本上计算打开和关闭括号以找到中断文本的位置)。这是函数:

def split_exprs(idx_exprs):
    keyexprs = []
    nopen = nclose = beg = curr = 0
    for c in idx_exprs:
        curr += 1
        if c == '(':
            nopen += 1
        elif c == ')':
            nclose += 1
            if nopen > 0 and nopen == nclose:
                if idx_exprs[beg] == ',':
                    beg += 1
                if idx_exprs[beg] == ' ':
                    beg += 1
                keyexprs.append(idx_exprs[beg:curr])
                beg = curr
                nopen = nclose = 0
    return keyexprs

问题是是否有更 Pythonic 或更优雅的方法来执行此操作或使用正则表达式来解决此问题。

Answer 1

这是我的版本，我认为它更像 pythonic，更简洁，并且适用于 chars 流，尽管我看不出有任何优势:)

def split_fns(fns):
    level = 0
    stack = [[]]
    for ch in fns:
        if level == 0 and ch in [' ',',']:
            continue        
        stack[-1].append(ch)

        if ch == "(":
            level += 1
        elif ch == ")":
            level -= 1
            if level == 0:
                stack.append([])

    return ["".join(t) for t in stack if t]