如何在Vim或Linux工具中批量修改Python函数调用签名?
当然可以!不管是用Vim还是sed这类Linux工具,都能轻松搞定这个批量修改函数调用的需求,下面分别给你两种工具的具体实现方案:
用Vim实现批量替换
如果你习惯在Vim里操作,全局正则替换就能搞定这个需求,步骤很简单:
- 打开目标文件,要是需要批量处理多个文件,可以用
:args path/to/your/python/files/*.py来加载所有要修改的文件。 - 执行全局替换命令:
:%s/my_function(\s*\([^,]*\)\s*,\s*\([^,]*\)\s*,\s*\([^)]*\)\s*)/my_function(\2, \3, outFile=\1)/g
- 要是想逐个确认每一处替换(避免误改),可以在命令末尾加
c参数:
:%s/my_function(\s*\([^,]*\)\s*,\s*\([^,]*\)\s*,\s*\([^)]*\)\s*)/my_function(\2, \3, outFile=\1)/gc
- 批量处理多文件的话,执行完替换后加上
| update自动保存每个文件:
:argdo %s/my_function(\s*\([^,]*\)\s*,\s*\([^,]*\)\s*,\s*\([^)]*\)\s*)/my_function(\2, \3, outFile=\1)/g | update
小提示:这个正则会匹配参数间任意数量的空格(包括没有空格的情况),能覆盖大部分常规的函数调用写法。但如果你的参数里包含逗号(比如传递元组(a,b)),这个简单正则会失效,这种复杂场景可以搭配Vim的Python语法插件(比如vim-textobj-python)来精准选中参数,再进行替换。
用sed实现批量替换
要是想直接在终端里批量修改文件,sed是个不错的选择:
对于GNU sed(大部分Linux发行版默认),执行以下命令:
sed -i 's/my_function(\s*\([^,]*\)\s*,\s*\([^,]*\)\s*,\s*\([^)]*\)\s*)/my_function(\2, \3, outFile=\1)/g' your_python_files*.py
如果是macOS的BSD sed,需要调整-i参数的写法:
sed -i '' 's/my_function(\s*\([^,]*\)\s*,\s*\([^,]*\)\s*,\s*\([^)]*\)\s*)/my_function(\2, \3, outFile=\1)/g' your_python_files*.py
注意:替换前建议先去掉-i参数,把结果输出到终端确认匹配是否正确,没问题再执行原地修改。和Vim的情况一样,参数里带逗号的复杂场景这个方法不太适用,这种时候可以写个简单的Python脚本用ast模块解析语法,精准修改函数调用。
内容的提问来源于stack exchange,提问作者Samizdis
相关产品推荐
相关产品推荐

