get_ast_from_globals.py

#-------------------------------------------------------------------------------
# get an AST of a function defined in globals()
#-------------------------------------------------------------------------------
import ast
import inspect
def getFunctionAst(name):
    if name not in globals():
        raise NameError("name '%s' is not defined" % name)
    func = globals()[name]
    source = inspect.getsource(func)
    tree = ast.parse(source)
    return tree