| from jedi.inference.value import ModuleValue |
| from jedi.inference.context import ModuleContext |
|
|
|
|
| class DocstringModule(ModuleValue): |
| def __init__(self, in_module_context, **kwargs): |
| super().__init__(**kwargs) |
| self._in_module_context = in_module_context |
|
|
| def _as_context(self): |
| return DocstringModuleContext(self, self._in_module_context) |
|
|
|
|
| class DocstringModuleContext(ModuleContext): |
| def __init__(self, module_value, in_module_context): |
| super().__init__(module_value) |
| self._in_module_context = in_module_context |
|
|
| def get_filters(self, origin_scope=None, until_position=None): |
| yield from super().get_filters(until_position=until_position) |
| yield from self._in_module_context.get_filters() |
|
|
