Get a function/class code from a file knowing the line and the column of it's definition

Question

Basically I want to use jedi to retrieve a function's or a class' code from the details of it's definition(s) (path, line, column). To be more explicit, what I really wish is to get the code from a file, that is not executed, static.

@DanielRoseman I greatly appreciate you are willing to spare the little time you have to help me. — khael, Mar 02 '16 at 14:22

score 0 · Answer 1 · answered Mar 02 '16 at 13:56

0

I use the function find_pyfunc_above_row defined in this file https://github.com/Erotemic/utool/blob/next/utool/util_inspect.py to accomplish a similar task.

answered Mar 02 '16 at 13:56

Erotemic

4,806
4
39
80

`find_pyfunc_above_row` gets me the function name, and for `get_func_sourcecode` I woud need a live function, I wanted a static code function retrieval, no live code whatsoever. Does `utool` provide such a functionality, or am I missing something? – khael Mar 02 '16 at 14:17
1

No, I don't think you're missing anything. I thought that function did a bit more than it did. Most of utool's inspection capability uses live code objects. You could however, ust ast's parse tree to just parse the file after the line that you know the function is on and then grab the code from visit_FunctionDef. Take a look at find_child_kwarg_funcs in the same file, which does static analysis. You can get source code from a parse tree. – Erotemic Mar 02 '16 at 14:30

score 0 · Accepted Answer · answered Mar 29 '16 at 13:24

It seems that you can use ast and codegen to accomplish this task.

I will post a sample of code to illustrate this:

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)

Thanks to @Erotemic for suggesting `ast`, he practically gave me the solution. — khael, Mar 29 '16 at 13:29

score 0 · Answer 3 · answered Apr 29 '16 at 13:39

This is currently not something that is supported in Jedi. You could certainly do it, but not with the public API. There are two things that Jedi's API is currently missing:

Getting the class/function by position (You can get this by playing with jedi's Parser).
Getting the code once you have your class. This is very easy: node.get_code()

Try to play with jedi.parser.Parser. It's quite a powerful tool, but not yet publicly documented.

Get a function/class code from a file knowing the line and the column of it's definition

3 Answers3