A first try to load foo-stub directories

This commit is contained in:
Dave Halter
2019-05-15 08:19:46 +02:00
parent dcbc60e1f0
commit 0e95aaeaad

View File

@@ -135,7 +135,16 @@ def _try_to_load_stub(evaluator, actual_context_set, parent_module_context, sys_
except KeyError:
pass
# 2. try to load pyi files next to py files.
# 1. Try to load foo-stubs folders on path for import name foo.
if not parent_module_context:
# foo-stubs
for p in sys_path:
init = os.path.join(p, *import_names) + '-stubs' + os.path.sep + '__init__.pyi'
m = _try_to_load_stub_from_file(evaluator, actual_context_set, init, import_names)
if m is not None:
return m
# 2. Try to load pyi files next to py files.
for c in actual_context_set:
try:
method = c.py__file__
@@ -218,6 +227,7 @@ def _try_to_load_stub_from_file(evaluator, actual_context_set, path, import_name
import_names
)
def create_stub_module(evaluator, actual_context_set, stub_module_node, path, import_names):
if import_names == ('typing',):
module_cls = TypingModuleWrapper