From 8fcf885de3c0b98bc08570ee06e2f251806d0527 Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Sun, 5 May 2019 21:35:06 +0200 Subject: [PATCH] Small refactoring --- jedi/evaluate/gradual/stub_context.py | 18 ++++++------------ 1 file changed, 6 insertions(+), 12 deletions(-) diff --git a/jedi/evaluate/gradual/stub_context.py b/jedi/evaluate/gradual/stub_context.py index 4bc270c4..f0089648 100644 --- a/jedi/evaluate/gradual/stub_context.py +++ b/jedi/evaluate/gradual/stub_context.py @@ -14,13 +14,6 @@ class _StubContextMixin(object): def is_stub(self): return True - def _get_stub_filters(self, **filter_kwargs): - return [StubFilter( - self.evaluator, - context=self, - **filter_kwargs - )] - class StubModuleContext(_StubContextMixin, ModuleContext): def __init__(self, non_stub_context_set, *args, **kwargs): @@ -32,11 +25,12 @@ class StubModuleContext(_StubContextMixin, ModuleContext): yield next(context.get_filters(search_global=False)) def _get_stub_filters(self, search_global, **filter_kwargs): - stub_filters = super(StubModuleContext, self)._get_stub_filters( - search_global=search_global, **filter_kwargs - ) - stub_filters += self.iter_star_filters(search_global=search_global) - return stub_filters + return [StubFilter( + self.evaluator, + context=self, + search_global=search_global, + **filter_kwargs + )] + list(self.iter_star_filters(search_global=search_global)) def get_filters(self, search_global=False, until_position=None, origin_scope=None, **kwargs):