python如何回到指定行_Python修改文件往指定行插入内容的实例

weixin_39732506

墨衍会员 · AI 创作全网分发码龄9年

关注

最新推荐文章于 2024-01-05 14:46:16 发布 202 阅读 AI 写同款· GEO 优化 ›

需求：批量修改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修改文件往指定行插入内容的实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持我们。

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

本文地址: http://www.cppcns.com/jiaoben/python/251474.html

标签

#python如何回到指定行

解码运动想象：BCI Competition IV 2a数据集实战指南本文提供了BCI Competition IV 2a数据集的实战指南，详细解析了该经典运动想象脑电数据集的结构、文件格式与实验范式。通过MNE-Python等工具，逐步演示了从原始信号读取、预处理（通道选择、滤波、分段）到特征提取（CSP、频带功率）和模型训练（LDA、SVM）的完整流程，并强调了受试者独立评估的重要性，为BCI研究者提供了可复现的入门与进阶路径。阅读详情

python如何回到指定行_Python修改文件往指定行插入内容的实例

相关推荐