Python修改文件往指定行插入内容的实例
需求:批量修改py文件中的类属性,为类增加一个core = True新的属性
原py文件如下
a.py
class A(): description = "abc"
现在有一个1.txt文本,内容如下,如果有py文件中的description跟txt文本中的一样,则增加core属性
1.txt
description = "abc" description = "123"
实现思路:
1.需要遍历code目录下的所有py文件,然后读取所有行数内容保存到lines列表中
2.遍历每个文件的每一行,匹配1.txt中的description,如果匹配中,就返回行号
3.往lines列表中根据行号插入要增加的新属性
4.重新写回原文件,达到修改文件的目的
如果修改成功后,效果应该是这样的
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修改文件往指定行插入内容的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持亿速云。