python - 递归遍历目录并替换函数调用

Question

我想递归地遍历目录并查找至少具有以下一组函数调用之一的所有文件:

A(a)
B(a,b)
C(a,b,c)

现在，不管参数如何，我都可以获得此类文件的列表

grep -r -l '[A-C](' .

尽管我确信我也可以以某种方式匹配这些参数。在这些文件上，我想要执行以下操作:首先，我想要一个备份，即将原始文件保存到 filename.ext_bak 或其他文件中，而在 filename.ext 中，我想要替换每次出现的函数调用

X(a,...) 

由

#ifdef LOL
   X_new(f(a),...)
#else
   X(a,...)
#endif

其中 X 可以是 A、B、C，并注意 X_new 中的每个参数都包装在函数 f(...) 中。将不胜感激任何帮助!提前致谢!

Answer 1

这使用 os.walk递归遍历所有文件(从当前工作目录开始)。

backup='_bak' 参数告诉 fileinput.input备份每个文件。

import os
import sys
import re
import fileinput

def sub_callback(match):
    func,args=match.groups()
    fargs=','.join('f({a})'.format(a=a) for a in args.split(','))
    return ('''\
#if def LOL
    {func}_new({fa})        
#else
    {func}({a})
#endif
'''.format(func=func,a=args,fa=fargs))

for root, dirs, files in os.walk('.'):
    for line in fileinput.input(
        (os.path.join(root,name) for name in files),
        inplace=True,
        backup='_bak'
        ):
        line=re.sub(r'\b([A-C])\((.*?)\)',sub_callback,line)
        sys.stdout.write(line)