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

我有一个（坏的）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"]

添加装饰器

更改函数名

改变睡眠时间

添加行

更改比较。 等等

这可以通过逐行检查文件来完成。

---

我想知道是否有更好的方法，更具Python风格的方法来实现这一点，也许是将函数加载到对象，修改对象并将其写回文件。

我最终使用了

more\u itertools

中的

peekable

，它允许您在不推进光标的情况下查看下一行。

我最终使用

more\u itertools

中的

peekable

，它允许您在不推进光标的情况下查看下查看下一行光标