python - 从文件中获取函数/类代码，知道它的定义的行和列

Question

基本上我想使用绝地从它的定义（路径，行，列）的细节中检索一个函数或一个类的代码。更明确地说，我真正希望的是从文件中获取代码，即未执行的静态代码。

Answer 1

看来您可以使用astandcodegen来完成这项任务。

我将发布一个代码示例来说明这一点：

import ast,codegen

def find_by_line(root, line):
    found = None
    if hasattr(root, "lineno"):
        if root.lineno == line:
            return root
    if hasattr(root, "body"):
        for node in root.body:
            found = find_by_line(node, line)
            if found:
                break
    return found

def get_func_code(path, line):
    with open(path) as file:
        code_tree = ast.parse(file.read())
        unit = find_by_line(code_tree, line)
        return codegen.to_source(unit)

Answer 2

Jedi 目前不支持此功能。您当然可以这样做，但不能使用公共 API。Jedi 的 API 目前缺少两件事：

1. 按位置获取类/函数（您可以通过玩绝地解析器来获取）。
2. 上课后获取代码。这很容易：node.get_code()

尝试玩jedi.parser.Parser。这是一个非常强大的工具，但尚未公开记录。

Answer 3

我使用此文件https://github.com/Erotemic/utool/blob/next/utool/util_inspect.py中定义的函数 find_pyfunc_above_row来完成类似的任务。