Преглед изворни кода

Merge pull request #56766 from touilleMan/issue-56698-2

Rémi Verschelde пре 3 година
родитељ
комит
9b3535a33a
1 измењених фајлова са 16 додато и 0 уклоњено
  1. 16 0
      SConstruct

+ 16 - 0
SConstruct

@@ -10,6 +10,7 @@ import os
 import pickle
 import sys
 import time
+from types import ModuleType
 from collections import OrderedDict
 from importlib.util import spec_from_file_location, module_from_spec
 
@@ -24,6 +25,21 @@ def _helper_module(name, path):
     module = module_from_spec(spec)
     spec.loader.exec_module(module)
     sys.modules[name] = module
+    # Ensure the module's parents are in loaded to avoid loading the wrong parent
+    # when doing "import foo.bar" while only "foo.bar" as declared as helper module
+    child_module = module
+    parent_name = name
+    while True:
+        try:
+            parent_name, child_name = parent_name.rsplit(".", 1)
+        except ValueError:
+            break
+        try:
+            parent_module = sys.modules[parent_name]
+        except KeyError:
+            parent_module = ModuleType(parent_name)
+            sys.modules[parent_name] = parent_module
+        setattr(parent_module, child_name, child_module)
 
 
 _helper_module("gles3_builders", "gles3_builders.py")