使用Python在文件中插入文本

Question

我有几个java代码文件,我必须在每个文件中添加一个公共类级变量.有这么多文件我需要编写一个python 2.2脚本来做同样的事情.

package xx.xx.xx;

import java.util.Properties;

public class MyClass extends YourClass{

    public myMethod() throws MyException {

    }
}

预期产出

package xx.xx.xx;

import java.util.Properties;

public class MyClass extends YourClass{

    public static final String CO_ID = "XXXXX" 

    public myMethod() throws MyMException {

    }
}

我知道file.find('{')给我第一次出现的索引{但我需要按下代码的其余部分并在每个文件中插入我的公共成员.

Answer 1

Python有一个方便的工具:fileinput模块:

import fileinput
import sys
import re

newline = '''\
{l}
    public static final String CO_ID = "XXXXX"
'''

filename = '/path/to/file.java'
for line in fileinput.input([filename], inplace=True, backup='.bak'):
    if re.match(r'public class', line):
        sys.stdout.write(newline.format(l=line))
    else:
        sys.stdout.write(line)

• inplace=True,改变文件"就地".实际上创建了一个临时文件,然后移动到原始位置.
• backup='.bak'告诉fileinput.input您创建原始文件的备份.
• sys.stdout.write使用而不是print因为print添加额外的新行而sys.stdout.write不是.