python在txt指定行添加文本（Python修改文件往指定行插入内容的实例）

python在txt指定行添加文本

Python修改文件往指定行插入内容的实例

需求：批量修改py文件中的类属性，为类增加一个core = True新的属性

原py文件如下

a.py

?

1 2	class A():   description = "abc"

现在有一个1.txt文本，内容如下，如果有py文件中的description跟txt文本中的一样，则增加core属性

1.txt

?

1 2	description = "abc" description = "123"

实现思路：

1.需要遍历code目录下的所有py文件，然后读取所有行数内容保存到lines列表中

2.遍历每个文件的每一行，匹配1.txt中的description，如果匹配中，就返回行号

3.往lines列表中根据行号插入要增加的新属性

4.重新写回原文件，达到修改文件的目的

如果修改成功后，效果应该是这样的

a.py

?

1 2 3

class A():   description = "abc"   core = True

实现代码：

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60

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

?

1 2 3

class A():   description = "abc\   aaaaabbbbb"

这样的如果修改后就变成了

?

1 2 3 4

class A():   description = "abc\   core = True   aaaaabbbbb"

为了避免这个bug，后面我才改成了hit-1

?

1	lines.insert(hit-1,' core = True\n')

这样修改的py文件后就是这样的效果

?

1 2 3 4

class A():   core = True   description = "abc\   aaaaabbbbb"

以上这篇Python修改文件往指定行插入内容的实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持开心学习网。

原文链接：https://blog.csdn.net/cssxn/article/details/81003146