no_toplevel_code

包装一段Python代码的包，以将其封装在函数和函数调用中，以避免在脚本的顶级运行代码。

示例

例如，如果您有一个看起来像这样的脚本

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 2*np.pi, 100)
y = np.sin(x)

plt.plot(x, y)
plt.show()

no_toplevel_code将其转换为这个

import numpy as np
import matplotlib.pyplot as plt

def _main():
    x = np.linspace(0, 2*np.pi, 100)
    y = np.sin(x)

    plt.plot(x, y)
    plt.show()

_main()

接口

此包提供了两个函数：wrap_code和unwrap_code，以及它们的等效函数wrap_ast和unwrap_ast。前者函数以字符串为输入，而后者函数以AST（抽象语法树）为输入。