Format: text/x-python
⬇ Download File (759 B)
   1  from jedi.inference.value import ModuleValue
   2  from jedi.inference.context import ModuleContext
   3  
   4  
   5  class DocstringModule(ModuleValue):
   6      def __init__(self, in_module_context, **kwargs):
   7          super().__init__(**kwargs)
   8          self._in_module_context = in_module_context
   9  
  10      def _as_context(self):
  11          return DocstringModuleContext(self, self._in_module_context)
  12  
  13  
  14  class DocstringModuleContext(ModuleContext):
  15      def __init__(self, module_value, in_module_context):
  16          super().__init__(module_value)
  17          self._in_module_context = in_module_context
  18  
  19      def get_filters(self, origin_scope=None, until_position=None):
  20          yield from super().get_filters(until_position=until_position)
  21          yield from self._in_module_context.get_filters()
  22