替换文本文件中的字符串

Question

我写了一个脚本，它将打开我的文本文件以搜索某个单词，然后选择包含该单词的行并将其分成三部分，然后选择数字部分并将其加1，因此每次运行脚本一个被添加到该编号。 这是脚本：

#!/usr/bin/env python
inputFile = open('CMakeLists.txt', 'r')

version = None
saved = ""

for line in inputFile:
    if "_PATCH " in line:
        print "inside: ", line
        version = line

    else:
        saved += line

inputFile.close()

inputFile = open('CMakeLists.txt', 'w')

x = version.split('"')
print "x: ", x

a = x[0]
b = int(x[1]) + 1
c = x[2]

new_version = str(a) + '"' + str(b) + '"' + str(c)
print "new_version: ", new_version

inputFile.write(str(saved))
inputFile.write(str(new_version))

inputFile.close()

但是我的问题是新编号正在文件的末尾写入，我希望它保留在其原始位置。 有任何想法吗 ？

谢谢

Answer 1

问题是您将新版本号写在原始文件之后（没有版本行）：

inputFile.write(str(saved))
inputFile.write(str(new_version))

您可以通过以下方式解决此问题：分别保存包含版本的行之前和之后的行，然后以正确的顺序保存它们：

#!/usr/bin/env python
inputFile = open('CMakeLists.txt', 'r')

version = None
savedBefore = ""
savedAfter = ""

for line in inputFile:
    if "_PATCH " in line:
        print "inside: ", line
        version = line
    elif version is None:
        savedBefore += line
    else:
        savedAfter += line

inputFile.close()

inputFile = open('CMakeLists.txt', 'w')

x = version.split('"')
print "x: ", x

a = x[0]
b = int(x[1]) + 1
c = x[2]

new_version = str(a) + '"' + str(b) + '"' + str(c)
print "new_version: ", new_version

inputFile.write(savedBefore)
inputFile.write(str(new_version))
inputFile.write(savedAfter)

inputFile.close()

注意：您可能需要在版本行中添加一些额外的文本，以使其具有与原始格式相同的格式（例如添加“ _PATCH”）。

Answer 2

您的代码有很多要说的。

您的错误是您正在编写“已保存”行，并且在编写修改后的版本之后。 因此，此修改的行将被写入文件的末尾。

此外，我建议您使用with语句。

lines = []    
with open('CmakeLists.txt', 'r') as _fd:
    while True:
        line = _fd.readline()
        if not line:
            break
        if '_PATCH ' in line:
            a, b, c = line.split('"')
            b = int(b) + 1
            line = '{} "{}" {}'.format(a, b, c)
        lines.append(line)
with open('CmakeLists.txt', 'w') as _fd:
    for line in lines:
        _fd.write(line)

此代码未经测试，可能包含一些错误...此外，如果您的输入文件很大，则将每一行都放在列表中可能是个坏主意。

Answer 3

#!/usr/bin/env python
inputFile = open('CMakeLists.txt', 'r')

version = None
saved = ""

for line in inputFile:
    if "_PATCH " in line:
        print "inside: ", line
        version = line
        x = version.split('"')
        print "x: ", x

        a = x[0]
        b = int(x[1]) + 1
        c = x[2]

        new_version = str(a) + '"' + str(b) + '"' + str(c)
        saved += new_version
    else:
        saved += line

inputFile.close()

inputFile = open('CMakeLists.txt', 'w')
inputFile.write(str(saved))

inputFile.close()

如果找到某行，则更新其内容并添加到saved ，一旦for循环结束，只需将saved写入文件