goto_definition does not handle decorators

goto_definition does not handle decorated functions correctly. When executed on a wrapped function, you are taken to the wrapper function instead of the original function.

Here’s a simple example that will reproduce the bug:

from functools import wraps

def logged(f):
    # goto_definition directs here instead of the wrapped function
    @wraps(f)
    def wrapper(*args, **kwargs):
        print('Calling function {}'.format(f.__name__))
        f(*args, **kwargs)
    return wrapper


@logged
def add(x, y):
    return x + y

# When executing goto_definition with the cursor over "add" 
# I am taken to "wrapper"
print(add(2, 2))

It doesn’t make a difference whether or not functools.wraps is used.

[sloria]

Thanks @davidhalter for the clarification.

I think the common case is that the user wants to go the wrapped function if it is decorated, and the definition if it is not decorated. If I wanted to go to the wrapper, I would just put my cursor on the decorator and run goto_definition. Is there a way to define a single keymapping to do this (i.e. always go to the definition of a function, sans wrappers)?

[davidhalter]

By default it’s <leader>d. You can also read the docs/README. Please make sure you have an up-to-date jedi-vim (and submodules).