Python批量修改文件：在匹配行前插入指定属性
本文介绍如何使用Python脚本批量修改目录下的所有py文件。通过读取目标txt文件中的description列表，遍历py文件匹配对应行，并在匹配行的前一行插入core = True属性。文章详细解析了处理长字符串换行导致的语法错误问题，提供了完整的代码实现及注意事项，帮助开发者高效完成批量代码修改任务。
需求场景：需要批量修改Python文件中的类属性，具体目标是为特定的类增加一个名为 core = True 的新属性。
原始py文件结构如下所示：
a.py
class A(): description = "abc"
现有一个名为 1.txt 的文本文件，其内容包含若干 description 值。逻辑要求是：如果某个py文件中的 description 值与 1.txt 中的某一行内容一致，则在该py文件中增加 core 属性。
1.txt
description = "abc" description = "123"
实现思路如下：
首先，需要遍历 code 目录下的所有py文件，读取每个文件的所有行内容，并将其保存到 lines 列表中。
接着，遍历每个文件的每一行，将当前行内容与 1.txt 中的 description 进行匹配。如果匹配成功，则记录并返回该行的行号。
然后，根据获取到的行号，在 lines 列表的指定位置插入需要增加的新属性代码。
最后，将修改后的 lines 列表重新写回原文件，从而完成对文件的修改操作。
如果修改成功，最终的效果应如下所示：
a.py
class A(): description = "abc" core = True
具体实现代码如下：
import os
original_folder = 'E:\\code\\'
core_list = []
count = 0
# if the description is in the current line
def isMatchDescription(line_buffer):
global core_list
# if not catch the core_list in global, reload it.
if not core_list:
with open("./core.txt","r") as f:
core_list = f.readlines()
# if match the core description
for des in core_list:
if line_buffer.strip() == des.strip():
return True
return False
def modifySignatures():
for dirpath, dirnames, filenames in os.walk(original_folder):
for filename in filenames:
modifyFile(os.path.join(dirpath,filename))
def modifyFile(filename):
global count
#print "Current file: %s"% filename
lines = []
with open(filename,"r") as f:
lines = f.readlines()
hit = 0
# Enume every single line for match the description
for index, line in enumerate(lines):
if isMatchDescription(line):
hit = index
print hit
print "Matched file:%s" % filename
count+=1
if hit > 0:
lines.insert(hit-1,' core = True\n')
f.close()
# Write back to file
with open(filename,"w") as f:
for line in lines:
f.write(line)
f.close()
if __name__ == '__main__':
modifySignatures()
print "Modified:%d"%count
在代码中，关键行 lines.insert(hit-1,' core = True\n') 中的 hit 代表目标py文件中 description 属性所在的行号。起初尝试使用 hit+1 进行插入，但发现部分文件出现了语法错误。原因是原py文件中某些 description 的值过长，导致使用了代码换行符 \ 进行续行。
a.py
class A(): description = "abc\ aaaaabbbbb"
若按原逻辑修改，结果可能变为：
class A(): description = "abc\ core = True aaaaabbbbb"
为了避免此类bug，后续将插入逻辑调整为 hit-1。
lines.insert(hit-1,' core = True\n')
经过上述调整后，修改后的py文件效果如下：
class A(): core = True description = "abc\ aaaaabbbbb"
以上即为Python批量修改文件并在指定行插入内容的完整示例。希望本文内容能对大家有所帮助，欢迎参考与支持。
免责声明：本文内容仅供参考，不代表本站观点。转载或引用请注明出处。
