[GR-23542] add __graalpython__.context_config builtin

PullRequest: graalpython/958

Author: cosminbasca

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/GraalPythonModuleBuiltins.java

@@ -40,6 +40,7 @@
  */
 package com.oracle.graal.python.builtins.modules;
 
+import static com.oracle.graal.python.nodes.BuiltinNames.__GRAALPYTHON__;
 import static com.oracle.graal.python.nodes.SpecialAttributeNames.__NAME__;
 import static com.oracle.graal.python.runtime.exception.PythonErrorType.ImportError;
 import static com.oracle.graal.python.runtime.exception.PythonErrorType.NotImplementedError;
@@ -64,7 +65,6 @@
 import com.oracle.graal.python.builtins.objects.method.PMethod;
 import com.oracle.graal.python.builtins.objects.module.PythonModule;
 import com.oracle.graal.python.builtins.objects.object.PythonObject;
-import com.oracle.graal.python.nodes.BuiltinNames;
 import com.oracle.graal.python.nodes.argument.ReadIndexedArgumentNode;
 import com.oracle.graal.python.nodes.argument.ReadVarArgsNode;
 import com.oracle.graal.python.nodes.call.PythonCallNode;
@@ -105,7 +105,7 @@
 import com.oracle.truffle.api.nodes.NodeVisitor;
 import com.oracle.truffle.llvm.api.Toolchain;
 
-@CoreFunctions(defineModule = BuiltinNames.__GRAALPYTHON__)
+@CoreFunctions(defineModule = __GRAALPYTHON__)
 public class GraalPythonModuleBuiltins extends PythonBuiltins {
     public static final String LLVM_LANGUAGE = "llvm";
 
@@ -126,7 +126,7 @@ public void initialize(PythonCore core) {
     public void postInitialize(PythonCore core) {
         super.postInitialize(core);
         PythonContext context = core.getContext();
-        PythonModule mod = core.lookupBuiltinModule("__graalpython__");
+        PythonModule mod = core.lookupBuiltinModule(__GRAALPYTHON__);
         if (!ImageInfo.inImageBuildtimeCode()) {
             mod.setAttribute("home", context.getLanguage().getHome());
         }

graalpython/lib-graalpython/__graalpython__.py

@@ -92,9 +92,44 @@ def __getattr__(attr):
     owner_module.__dict__['__getattr__'] = __getattr__
 
 
+@builtin
+def auto_wrap_methods(delegate_name, delegate_attributes, owner_globals):
+    func_type = type(import_current_as_named_module)
+
+    new_globals = dict(**owner_globals)
+
+    for attr in owner_globals:
+        if attr.startswith("__"):
+            continue
+        elif not isinstance(owner_globals[attr], func_type):
+            continue
+        elif attr not in delegate_attributes:
+            raise AttributeError("attribute '{}' not allowed in module '{}', permitted values are: '{}'".format(
+                attr, __name__, delegate_attributes
+            ))
+
+        if attr in delegate_attributes:
+            def make_wrapper(attribute, method):
+                @__graalpython__.builtin
+                def wrapper(*args, **kwargs):
+                    try:
+                        return method(*args, **kwargs)
+                    except NotImplementedError:
+                        delegate_module = __import__(delegate_name)
+                        return getattr(delegate_module, attribute)(*args, **kwargs)
+                return wrapper
+
+            new_globals[attr] = make_wrapper(attr, owner_globals[attr])
+
+    return new_globals
+
+
 @builtin
 def import_current_as_named_module_with_delegate(module_name, delegate_name, delegate_attributes=None,
-                                                 owner_globals=None):
+                                                 owner_globals=None, wrap_methods=True):
     owner_module = import_current_as_named_module(module_name, owner_globals=owner_globals)
+    if wrap_methods and owner_globals:
+        wrapped_globals = auto_wrap_methods(delegate_name, delegate_attributes, owner_globals)
+        owner_module.__dict__.update(**wrapped_globals)
     if delegate_attributes:
         lazy_attributes_from_delegate(delegate_name, delegate_attributes, owner_module)

graalpython/lib-graalpython/_struct.py

@@ -41,6 +41,6 @@
 __graalpython__.import_current_as_named_module_with_delegate(
     module_name="_struct",
     delegate_name="_cpython_struct",
-    delegate_attributes=['Struct', '_clearcache', 'calcsize', 'error', 'iter_unpack', 'pack', 'pack_into', 'unpack',
-                         'unpack_from'],
+    delegate_attributes=['Struct', 'StructError', '_clearcache', 'calcsize', 'error', 'iter_unpack', 'pack',
+                         'pack_into', 'unpack', 'unpack_from'],
     owner_globals=globals())