docstring_utils.py 759 B

123456789101112131415161718192021
  1. from jedi.inference.value import ModuleValue
  2. from jedi.inference.context import ModuleContext
  3. class DocstringModule(ModuleValue):
  4. def __init__(self, in_module_context, **kwargs):
  5. super().__init__(**kwargs)
  6. self._in_module_context = in_module_context
  7. def _as_context(self):
  8. return DocstringModuleContext(self, self._in_module_context)
  9. class DocstringModuleContext(ModuleContext):
  10. def __init__(self, module_value, in_module_context):
  11. super().__init__(module_value)
  12. self._in_module_context = in_module_context
  13. def get_filters(self, origin_scope=None, until_position=None):
  14. yield from super().get_filters(until_position=until_position)
  15. yield from self._in_module_context.get_filters()