1
0
forked from VimPlug/jedi

Add new API: jedi.api.get_definitions

This commit is contained in:
Takafumi Arakaki
2013-02-13 19:52:38 +01:00
parent dac25ea342
commit 39feecee04
3 changed files with 63 additions and 0 deletions

View File

@@ -468,6 +468,25 @@ class Script(object):
api_classes._clear_caches()
def get_definitions(source, source_path=None, source_encoding='utf-8'):
"""
Get all definitions in `source` sorted by its position.
This functions can be used for listing functions, classes and
data defined in a file. This can be useful if you want to list
them in "sidebar". Each element in the returned list also has
`get_definition` method which can be used to get sub-definitions
(e.g., methods in class).
:rtype: list of api_classes.Definition
"""
parser = parsing.Parser(
modules.source_to_unicode(source, source_encoding),
module_path=source_path,
)
return api_classes.get_definitions(parser.scope)
def set_debug_function(func_cb=debug.print_to_stdout, warnings=True,
notices=True, speed=True):
"""

View File

@@ -317,6 +317,32 @@ class Definition(BaseDefinition):
position = ''
return "%s:%s%s" % (self.module_name, self.description, position)
def get_definitions(self):
"""
List sub-definitions (e.g., methods in class).
:rtype: list of Definition
"""
return get_definitions(self._parser.scope)
def get_definitions(scope):
"""
List sub-definitions (e.g., methods in class).
:type scope: Scope
:rtype: list of Definition
"""
def is_definition(s):
return isinstance(s, (pr.Import, pr.Function, pr.Class)) or \
isinstance(s, pr.Statement) and s.assignment_details
scopes = []
scopes.extend(scope.imports)
scopes.extend(scope.statements)
scopes.extend(scope.subscopes)
return map(Definition, sorted(filter(is_definition, scopes),
key=lambda s: s.start_pos))
class RelatedName(BaseDefinition):
"""TODO: document this"""