python批量替換多文件字符串問題詳解

更新時間：2018年04月22日 15:06:25 作者：Mike_Zhang

批量替換是我們在日常工作中經(jīng)常會遇到的一個問題，下面這篇文章主要給大家介紹了關于python批量替換多文件字符串問題的相關資料，文中通過示例代碼介紹的非常詳細，需要的朋友們下面隨著小編來一起學習學習吧。

系統(tǒng)如下：

操作系統(tǒng) ： CentOS7.3.1611_x64

Python 版本 : 2.7.5

問題描述

編碼過程中有時候會遇到在多個源文件中存在同一個變量名（比如： writeBuffer），需要替換為新的變量名（比如： write_buffer）的問題。怎么能方便快捷的解決該問題呢？

解決方案

使用sed

sed和grep結(jié)合使用可以替換當前文件夾多個文件的內(nèi)容。

格式：

sed -i 's/原字符串/新字符串/g' `grep -rl 原字符串所在目錄`

示例代碼：

sed -i 's/writeBuffer/write_buffer/g' `grep -rl writeBuffer ./*`

使用Python

使用Python腳本可以實現(xiàn)替換當前文件夾多個文件的內(nèi)容。

替換單個文件的代碼如下：

def doReplace(fpath,src,dst):
 newConent,bFlag = "",False
 with open(fpath,"rb") as fin:
 for line in fin :
  if line.find(src) == -1 :
  newLine = line
  else:
  bFlag = True
  newLine = line.replace(src,dst)
  newConent += newLine
 if not bFlag : return None
 print fpath
 with open(fpath,"wb") as fout:
 fout.write(newConent)
 return None

替換多個文件僅需添加目錄遍歷代碼。

完整示例代碼如下：

#! /usr/bin/python
#-*- coding: utf-8 -*- 

import os,sys

def doReplace(fpath,src,dst):
 newConent,bFlag = "",False
 with open(fpath,"rb") as fin:
 for line in fin :
  #if len(line.strip()) == 0 : continue
  if line.find(src) == -1 :
  newLine = line
  else:
  bFlag = True
  newLine = line.replace(src,dst)
  newConent += newLine
 if not bFlag : return None
 print fpath
 with open(fpath,"wb") as fout:
 fout.write(newConent)
 return None

def replaceMain(dirName,src,dst):
 for root, dirs, files in os.walk(dirName):
 for name in files:
  fpath = os.path.join(root, name)
  doReplace(fpath,src,dst)
 return None

if __name__ == "__main__":
 if len(sys.argv) < 3 :
 print "usage : replaceMulti srcStr dstStr"
 print "replace current dir files"
 sys.exit(1)
 srcStr = sys.argv[1]
 dstStr = sys.argv[2]
 dirName = "."
 dirName = os.path.realpath(dirName)
 print "working dir :",dirName
 replaceMain(dirName,srcStr,dstStr)

添加可執(zhí)行權(quán)限：

chmod a+x replaceMulti.py

使用示例：

./replaceMulti.py writeBuffer write_buffer

將當前文件夾中所有 writeBuffer 替換為 write_buffer

也可以將 replaceMulti.py 放入 /usr/local/bin/ 目錄：

[root@local ~]# mv replaceMulti.py /usr/local/bin/
[root@local ~]# replaceMulti.py
usage : replaceMulti srcStr dstStr
replace current dir files
[root@local ~]#

該腳本在windows下也可以使用，將 replaceMulti.py 所在目錄加入環(huán)境變量即可。

好，就這些了，希望對你有幫助。

總結(jié)

以上就是這篇文章的全部內(nèi)容了，希望本文的內(nèi)容對大家的學習或者工作具有一定的參考學習價值，如果有疑問大家可以留言交流，謝謝大家對腳本之家的支持。

您可能感興趣的文章: