如何使用 python3 在特定的 position 中插入新的字符串行

Question

我的问题的难点在于“具体位置”。 我想编写一个脚本，首先测试代码（文件）中是否存在某些特定字符串，然后在所有其他包含的末尾包含一个 header 文件。 例如，如果我在这个文件中找到SomeString ，那么之后

#include <oldfile1.h>
#include "oldfile2.h"

插入#include "someheaderfile.h"

如果可能的话，我正在寻找一个优雅的解决方案。

Answer 1

with open(fname,'r') as f:
    s = f.read()
    if 'SomeString' in s:
        sl = s.splitlines()
        for i, x in enumerate(sl):
            if x[0] != '#':
                out = sl[:i] + ['#include "someheaderfile.h"'] + sl[i+1:]
                break

with open(fname,'w') as f:
    f.write('\n'.join(out))

您阅读了文件，检查您的字符串是否在其中。 然后逐行阅读以定位 header 区域的末尾，然后重新构建行列表，添加额外的 header。

然后通过加入新列表重写您的文件。