从文件中操作现有 python 代码的有效方法

Question

我有一个由外部程序生成的（损坏的）python 代码。问题是，生成的代码被破坏了（或者只是导出得很糟糕）。我想创建一个脚本，根据预定义的规则更改损坏的代码。
例如，考虑以下代码：

def test_1TestName(self):
    time.sleep(1000)
    assert "a" == "self.vars["value"]"

我想通过添加更改来修复它，例如：

@pytest.mark.order(1)
def Test_TestName(self):
    time.sleep(1)
    print("hello")
    assert "a" == self.vars["value"]

1. 添加装饰器
2. 更改函数名称。
3. 更改睡眠时间。
4. 添加行。
5. 更改比较。 等

这可以通过逐行遍历文件来完成。
我想知道是否有更好的方式，更pythonic的方式来实现这一点，也许将函数加载到对象，修改对象并将其写回文件。

Answer 1

我最终使用more_itertools 中的peekable，它允许您在不移动光标的情况下查看下一行。