溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

Python修改文件往指定行插入內(nèi)容的實(shí)例

發(fā)布時(shí)間：2020-08-29 16:34:30 來源：腳本之家 閱讀：263 作者：FFE4 欄目：開發(fā)技術(shù)

需求：批量修改py文件中的類屬性，為類增加一個(gè)core = True新的屬性

原py文件如下

a.py

class A():
  description = "abc"

現(xiàn)在有一個(gè)1.txt文本，內(nèi)容如下，如果有py文件中的description跟txt文本中的一樣，則增加core屬性

1.txt

description = "abc"
description = "123"

實(shí)現(xiàn)思路：

1.需要遍歷code目錄下的所有py文件，然后讀取所有行數(shù)內(nèi)容保存到lines列表中

2.遍歷每個(gè)文件的每一行，匹配1.txt中的description，如果匹配中，就返回行號(hào)

3.往lines列表中根據(jù)行號(hào)插入要增加的新屬性

4.重新寫回原文件，達(dá)到修改文件的目的

如果修改成功后，效果應(yīng)該是這樣的

a.py

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

實(shí)現(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屬性的行號(hào)，我之前用的是hit+1，但是后面發(fā)現(xiàn)有些文件出現(xiàn)了語法錯(cuò)誤，原因是py文件中有些description的值太長(zhǎng)，導(dǎo)致原文件使用了代碼換行符\，如下：

a.py

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

這樣的如果修改后就變成了

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

為了避免這個(gè)bug，后面我才改成了hit-1

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

這樣修改的py文件后就是這樣的效果

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

以上這篇Python修改文件往指定行插入內(nèi)容的實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持億速云。

向AI問一下細(xì)節(jié)

免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱：is@yisu.com進(jìn)行舉報(bào)，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手