VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2026-05-19 23:09:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Better py__getattribute__ for ContextSet

Browse Source
This commit is contained in:
Dave Halter
2019-04-11 22:59:54 +02:00
parent 0f037d0e6c
commit 44639ee50e
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
jedi/evaluate/base_context.py
+7
View File
@@ -6,6 +6,8 @@ A ContextSet is typically used to specify the return of a function or any other
static analysis operation. In jedi there are always multiple returns and not static analysis operation. In jedi there are always multiple returns and not
just one. just one.
""" """
from functools import reduce
from operator import add
from parso.python.tree import ExprStmt, CompFor from parso.python.tree import ExprStmt, CompFor
from jedi import debug from jedi import debug
@@ -356,6 +358,11 @@ class ContextSet(BaseContextSet):
def execute_evaluated(self, *args, **kwargs): def execute_evaluated(self, *args, **kwargs):
return ContextSet.from_sets(execute_evaluated(c, *args, **kwargs) for c in self._set) return ContextSet.from_sets(execute_evaluated(c, *args, **kwargs) for c in self._set)
def py__getattribute__(self, *args, **kwargs):
if kwargs.get('is_goto'):
return reduce(add, [c.py__getattribute__(*args, **kwargs) for c in self._set], [])
return ContextSet.from_sets(c.py__getattribute__(*args, **kwargs) for c in self._set)
def get_item(self, *args, **kwargs): def get_item(self, *args, **kwargs):
return ContextSet.from_sets(_getitem(c, *args, **kwargs) for c in self._set) return ContextSet.from_sets(_getitem(c, *args, **kwargs) for c in self._set)