python脚本替换指定行实现步骤

 python脚本替换指定行实现步骤

       本文主要介绍了Python的脚本替换，由于工作的需要，必须对日志系统进行更新，这里在网上搜索到一篇文章比较不错，这里记录下，大家可以参考下，

工作中需要迁移代码，并把原来的日志系统更新到现在的格式，原来获取log的格式是

AuctionPoolLoggerUtil.getLogger() 

现在获取log的格式是：

LoggerFactory.getLogger(XXXXX.class) 

这里的XXXXX需要替换为当前的类名。如果这样的java文件不多还好，可以一个个人肉替换。一旦这样的文件很多，特别是迁移过来大量的文件时，你就会发现简直是一场灾难。其实我们发现上面的工作很多是机械单调的。ide中的替换功能不能做到的是把XXXXX替换成当前的类名。而python很容易处理文本，利用正则表达式可以比较方便的拿到类名，然后替换掉xxxxx就可以了。

实现代码：

import fileinput
import os
import re

__author__ = 'ykdsg'

packDir='/Users/ykdsg/svn_workspace/auctionplatform/misc_refactory/auctionplatform/ap-biz/src/main/java/com/yk/misccenter'
#查找class name
findClassNameP=re.compile(r"(?<=class\s)\w*")
findXP=re.compile(r"XXXXX")


def processDirectory(args,dirname,filenames):
  # print 'Directory',dirname
  for filename in filenames:

    if os.path.splitext(filename)[1]=='.java':
      # print 'file',filename
      fullFileUrl=dirname+ "/"+filename
      fileObj=open(fullFileUrl)
      className=''
      # Optional in-place filtering: if the keyword argument inplace=1 is passed to fileinput.input() or to
      # the FileInput constructor, the file is moved to a backup file and standard output is directed to the
      # input file (if a file of the same name as the backup file already exists, it will be replaced silently)
      # . This makes it possible to write a filter that rewrites its input file in place. If the backup
      # parameter is given (typically as backup='.<some extension>'), it specifies the extension for the
      # backup file, and the backup file remains around; by default, the extension is '.bak' and it is deleted
      # when the output file is closed. In-place filtering is disabled when standard input is read.
      for line in fileinput.input(fullFileUrl, inplace=1):
        matchClass = findClassNameP.search(line)
        if matchClass:
          className = matchClass.group()
        matchX=findXP.search(line)
        if matchX:
          #print 后面需要有, 否则会出现多余的空行
          print line.replace('XXXXX',className),
        else:
          print line,


def search():
  os.path.walk(packDir,processDirectory,None)

if __name__ == '__main__':
  search()

上面的脚本中大部分是fileinput.input的注释，就是说了inplace=1其实就是把源文件的内容放到缓存区，然后直接将内容写入源文件

findClassNameP 是查找class name的正则表达式，上面的逻辑就是对文件逐行分析，拿到class name。然后再分析当前行是否有xxxxx，有的话就用class name 替换，没有的话就原行输出。

       以上使用对python脚本替换指定行的简单实例，如果大家有疑问或者更好的方法可以留言讨论，感谢阅读，希望能帮助到大家，谢谢大家对本站的支持！