能否用Sed匹配嵌套括号？实现idof()代码替换需求

问题描述 #

我的代码里大量存在idof(some_object)这样的结构，想要替换成some_object["id"]。一开始用的命令是：
sed -i 's/idof(\([^)]\+\))/\1["id"]/g' source.py
但当some_object包含嵌套括号（比如idof(get_some_object())或idof(my_class().get_some_object())）时，替换结果不对，会错误截断内容。能不能让Sed跟踪括号的开闭匹配，忽略内部嵌套括号，实现把idof(ANYTHING)替换成ANYTHING["id"]的需求？

解决方案 #

Sed本身是基于行处理和正则表达式的工具，原生正则不支持递归匹配嵌套结构，单条sed命令搞不定这种场景。推荐用更适合处理复杂语法结构的工具，比如awk或者Python脚本，下面是两个可行方案：

方案1：用Awk实现括号匹配 #

Awk支持自定义变量跟踪括号层级，能精准匹配成对的括号：

BEGIN {
    RS = ""
    ORS = "\n"
}
{
    while (match($0, /idof\(/)) {
        start = RSTART + 5  # 跳过"idof("
        end = start
        depth = 1
        while (end <= length($0) && depth > 0) {
            char = substr($0, end, 1)
            if (char == "(") depth++
            else if (char == ")") depth--
            end++
        }
        if (depth == 0) {
            content = substr($0, start, end - start - 1)
            $0 = substr($0, 1, RSTART - 1) content "[\"id\"]" substr($0, end)
        } else {
            break  # 处理不完整的括号，避免死循环
        }
    }
    print $0
}

使用方式：把上面的代码保存为replace_idof.awk，然后执行：

awk -f replace_idof.awk source.py > temp.py && mv temp.py source.py

方案2：用Python脚本实现 #

Python的字符串处理更灵活，能轻松处理嵌套括号：

import re

def replace_idof(match):
    content = match.group(1)
    return f'{content}["id"]'

# 递归匹配嵌套括号的正则（需要Python 3.10+支持正则递归）
pattern = re.compile(r'idof\(([^()]|(?R))*\)')

with open('source.py', 'r') as f:
    content = f.read()

new_content = pattern.sub(replace_idof, content)

with open('source.py', 'w') as f:
    f.write(new_content)

使用方式：保存为replace_idof.py，直接运行即可修改文件。

说明 #

• Sed的正则属于POSIX基本正则，不支持递归匹配，没法处理嵌套括号这种需要层级跟踪的场景，换用Awk或Python是更靠谱的选择。
• 上面的方案都会完整匹配idof(到对应闭合)之间的内容，不管里面有多少层嵌套括号，替换后都会得到正确的内容["id"]格式。

内容的提问来源于stack exchange，提问作者John