1
0
forked from VimPlug/jedi

Make it possible for users to infer annotations/defaults

Fixes #1039
This commit is contained in:
Dave Halter
2019-08-01 18:27:37 +02:00
parent 7988c1d11b
commit c2e50e1d0d
2 changed files with 22 additions and 2 deletions

View File

@@ -3,7 +3,7 @@ from abc import abstractmethod
from parso.tree import search_ancestor
from jedi._compatibility import Parameter
from jedi.evaluate.base_context import ContextSet
from jedi.evaluate.base_context import ContextSet, NO_CONTEXTS
from jedi.cache import memoize_method
@@ -220,6 +220,18 @@ class ParamName(BaseTreeParamName):
def annotation_node(self):
return self._get_param_node().annotation
def infer_annotation(self):
node = self.annotation_node
if node is None:
return NO_CONTEXTS
return self.parent_context.parent_context.eval_node(node)
def infer_default(self):
node = self.default_node
if node is None:
return NO_CONTEXTS
return self.parent_context.parent_context.eval_node(node)
@property
def default_node(self):
return self._get_param_node().default