From 65c02a2332ecfd9e476189dfae570241151bd5d1 Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Sun, 3 Sep 2017 01:05:53 +0200 Subject: [PATCH] A bit of shuffling code around get_definition around. --- jedi/evaluate/filters.py | 6 ++---- jedi/evaluate/finder.py | 2 +- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/jedi/evaluate/filters.py b/jedi/evaluate/filters.py index 597613ed..7e803b3f 100644 --- a/jedi/evaluate/filters.py +++ b/jedi/evaluate/filters.py @@ -89,11 +89,9 @@ class TreeNameDefinition(AbstractTreeName): @property def api_type(self): - definition = self.tree_name.get_definition() + definition = self.tree_name.get_definition(import_name_always=True) if definition is None: - definition = self.tree_name.parent - if definition.type == 'dotted_as_name': - definition = definition.parent + return 'statement' return self._API_TYPES.get(definition.type, 'statement') diff --git a/jedi/evaluate/finder.py b/jedi/evaluate/finder.py index 1bbb093c..9c11ae5e 100644 --- a/jedi/evaluate/finder.py +++ b/jedi/evaluate/finder.py @@ -173,7 +173,7 @@ class NameFinder(object): def _name_to_types(evaluator, context, tree_name): types = [] - node = tree_name.get_definition() + node = tree_name.get_definition(import_name_always=True) if node is None: node = tree_name.parent if node.type == 'global_stmt':