Python修改文件往指定行插入內(nèi)容的實例
需求：批量修改py文件中的類屬性，為類增加一個core = True新的屬性
原py文件如下
a.py
class A(): description = "abc"
現(xiàn)在有一個1.txt文本，內(nèi)容如下，如果有py文件中的description跟txt文本中的一樣，則增加core屬性
1.txt
description = "abc" description = "123"
實現(xiàn)思路：
1.需要遍歷code目錄下的所有py文件，然后讀取所有行數(shù)內(nèi)容保存到lines列表中
2.遍歷每個文件的每一行，匹配1.txt中的description，如果匹配中，就返回行號
3.往lines列表中根據(jù)行號插入要增加的新屬性
4.重新寫回原文件，達(dá)到修改文件的目的
如果修改成功后，效果應(yīng)該是這樣的
a.py
class A(): description = "abc" core = True
實現(xiàn)代碼：
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代表目標(biāo)py文件的description屬性的行號，我之前用的是hit+1，但是后面發(fā)現(xiàn)有些文件出現(xiàn)了語法錯誤，原因是py文件中有些description的值太長，導(dǎo)致原文件使用了代碼換行符\，如下：
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修改文件往指定行插入內(nèi)容的實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
雖然Python的標(biāo)準(zhǔn)庫中urllib2模塊已經(jīng)包含了平常我們使用的大多數(shù)功能，但是它的API使用起來讓人實在感覺不好。它已經(jīng)不適合現(xiàn)在的時代，不適合現(xiàn)代的互聯(lián)網(wǎng)了。而Requests的誕生讓我們有了更好的選擇。本文就介紹了Python中第三方庫Requests庫的高級用法。2017-03-03
python數(shù)學(xué)建模是加深Numpy和Pandas學(xué)習(xí)
這篇文章主要介紹了python數(shù)學(xué)建模是加深Numpy和Pandas學(xué)習(xí)，緊接上一篇學(xué)習(xí)內(nèi)容展開Numpy更多相關(guān)內(nèi)容，需要的小伙伴可以參考一下2022-07-07
Python使用random模塊生成隨機(jī)數(shù)操作實例詳解
這篇文章主要介紹了Python使用random模塊生成隨機(jī)數(shù)操作,結(jié)合具體實例形式詳細(xì)分析了random模塊生成隨機(jī)數(shù)的各種常用技巧與相關(guān)操作注意事項,需要的朋友可以參考下2019-09-09
使用Python對Dicom文件進(jìn)行讀取與寫入的實現(xiàn)
這篇文章主要介紹了使用Python對Dicom文件進(jìn)行讀取與寫入的實現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-04-04
這篇文章主要介紹了python的第三方庫pandas的相關(guān)知識，本文給大家介紹的非常詳細(xì)，具有一定的參考借鑒價值,需要的朋友可以參考下2020-02-02
今天小編就為大家分享一篇pytorch 常用線性函數(shù)詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-01-01
python PIL和CV對 圖片的讀取,顯示,裁剪,保存實現(xiàn)方法
今天小編就為大家分享一篇python PIL和CV對 圖片的讀取,顯示,裁剪,保存實現(xiàn)方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08
這篇文章主要介紹了解決Pymongo insert時會自動添加_id的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-12-12
python openpyxl方法 zip函數(shù)用法及說明
這篇文章主要介紹了python openpyxl方法 zip函數(shù)用法及說明，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-05-05
Python實現(xiàn)PDF和TIFF格式之間的相互轉(zhuǎn)換
PDF是數(shù)據(jù)文檔管理領(lǐng)域常用格式之一,主要用于存儲和共享包含文本、圖像、表格、鏈接等的復(fù)雜文檔,而TIFF常見于圖像處理領(lǐng)域, 在實際應(yīng)用中,我們可能有時需要將PDF文件轉(zhuǎn)換為TIFF圖像,本文將介紹如何使用Python實現(xiàn)PDF和TIFF格式之間的相互轉(zhuǎn)換,需要的朋友可以參考下2024-07-07