实现解析python文件,生成AST,然后修改python文件内的内容并输出。

RedBaron

使用RedBaron库可以实现这个功能,官网是https://redbaron.readthedocs.io/en/latest/。 安装:pip install redbaron

Demo

样例文件

a = 1


def hello():
    return "world"


b = 2
c = 3
d = [1, 2, 3]

修改样例文件

def lab001():
    # 读取sample.py文件
    red = RedBaron(open("sample.py").read())

    # find ,直接查找node的type
    node = red.find("assignment", lambda x: x.name.value == 'a')
    node.value = '9'

    print(red.dumps())  # get code back

image.png|150