python - 重写 python 脚本，在脚本的每个类中注入(inject)一个方法

Question

假设我有一个 python 模块 foo.py，其中包含:

class Foo(object):
    def __init__(self):
        pass

接下来我想解析这个脚本并在它的每个类中注入(inject)一个方法，将其重写为如下所示:

class Foo(object):
    def __init__(self):
        pass
    def my_method(self):
        pass # do some stuff here

我注意到 python 2.6 有一个可以用于此目的的 ast 模块，但不幸的是我需要在 python 2.5 中执行此操作。欢迎任何建议。

Answer 1

我明白你想要做什么，但是如果你的源代码相当简单，也许你可以跳过整个解析和写回，只运行正则表达式？

类似于:

re.sub('^class\s+(\w+):\s*\n', 'class \\1:\n%s' % method_source, file_source, flags=re.M)

只要缩进正确..