AST 模块：用 Python 修改 Python 代码_python ast修改函数的参数(1)

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化学习资料的朋友，可以戳这里获取

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

ast.dump(expr_ast)
“Module(
body=[
FunctionDef(
name=‘add’, args=arguments(
args=[
Name(id=‘arg1’, ctx=Param()),
Name(id=‘arg2’, ctx=Param())
],
vararg=None,
kwarg=None,
defaults=[]),
body=[
Return(
value=BinOp(
left=Name(id=‘arg1’, ctx=Load()),
op=Add(),
right=Name(id=‘arg2’, ctx=Load())))
],
decorator_list=[])
])”

正如我们所见， Module 是父节点，它的 body 中包含了一个函数定义的元素，这个函数定义包含了函数名、参数列表和函数体。函数体又包含了一个单独的 Return 节点，节点中含有一个  Add 运算。




### 修改 AST


我们如何修改这棵树以改变代码的作用呢？为了说明这个问题，我们来做点也许你永远也不会在你自己代码中做的疯狂的事情吧。我们将遍历这棵树，并且将  Add 运算修改为 Mult 运算。看，我说过这很疯狂吧！


我们要先建立一个 NodeTransformer 变换器的子类，并且定义 visit\_BinOp 方法。每当这个变换器访问到一个二元运算符节点时，就会调用这个方法。

class CrazyTransformer(ast.NodeTransformer):

def visit\_BinOp(self, node):
    print node.\_\_dict\_\_
    node.op = ast.Mult()
    print node.\_\_dict\_\_
    return node

现在我们已经定义好了我们这个奇怪的变换器，让我们看看将它应用于我们开始时写的那些代码会怎么样：

transformer = CrazyTransformer()
transformer.visit(expr_ast)
{
‘op’: <_ast.Add object at 0x10a8321d0>,
‘right’: <_ast.Name object at 0x10a839390>,
‘lineno’: 3, ‘col_offset’: 8,
‘left’: <_ast.Name object at 0x10a839350>}
{
‘op’: <_ast.Mult object at 0x10a839510>,
‘right’: <_ast.Name object at 0x10a839390>,
‘lineno’: 3, ‘col_offset’: 8,
‘left’: <_ast.Name object at 0x10a839350>}

你可以从输出的结果对比发现， Add 节点已经被替换成了一个 Mult 。我们有许多方法没有提到，比如访问子节点，不过这个例子已经足以刻画出它的基本原理。




### 


### 编译和执行修改后的 AST


我们在最初的代码后面加上一个调用，比如：

print add(4, 5)

让我们看看这些代码是如何运行的：

unmodified = ast.parse(expr)
exec compile(unmodified, ‘’, ‘exec’)
9
transformer = CrazyTransformer()
modified = transformer.visit(unmodified)
exec compile(modified, ‘’, ‘exec’)
20

我们可以看到，未修改的和修改后的 AST 所编译出的代码，一个输出了9，一个输出了20。




### 重新翻译回源代码


最后，我们可以用 unparse 模块将修改后的代码转换回对应的源代码， unparse 模块可以在  [这里](https://bbs.csdn.net/topics/618317507) 找到。

unparse.Unparser(modified, sys.stdout)

def add(arg1, arg2):
return (arg1 * arg2)
print add(4, 5)

正如我们所看到的， \* 运算符取代了  + 。在这个反解析工具对于理解你的 AST 变换器如何修改代码很有帮助。


**（1）Python所有方向的学习路线（新版）**  

这是我花了几天的时间去把Python所有方向的技术点做的整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照上面的知识点去找对应的学习资源，保证自己学得较为全面。

最近我才对这些路线做了一下新的更新，知识体系更全面了。



![在这里插入图片描述](https://img-blog.csdnimg.cn/1f807758e039481fa866130abf71d796.png#pic_center)



**（2）Python学习视频**



包含了Python入门、爬虫、数据分析和web开发的学习视频，总共100多个，虽然没有那么全面，但是对于入门来说是没问题的，学完这些之后，你可以按照我上面的学习路线去网上找其他的知识资源进行进阶。

![在这里插入图片描述](https://img-blog.csdnimg.cn/d66e3ad5592f4cdcb197de0dc0438ec5.png#pic_center)



**（3）100多个练手项目**

我们在看视频学习的时候，不能光动眼动脑不动手，比较科学的学习方法是在理解之后运用它们，这时候练手项目就很适合了，只是里面的项目比较多，水平也是参差不齐，大家可以挑自己能做的项目去练练。

![在这里插入图片描述](https://img-blog.csdnimg.cn/f5aeb4050ab547cf90b1a028d1aacb1d.png#pic_center)




**网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。**

**[需要这份系统化学习资料的朋友，可以戳这里获取](https://bbs.csdn.net/topics/618317507)**

**一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！**