Implement C API function PyModule_GetDef

Author: fangerer

graalpython/com.oracle.graal.python.cext/src/moduleobject.c

@@ -150,6 +150,14 @@ PyObject* PyModule_New(const char *name) {
     return UPCALL_CEXT_O(_jls_PyModule_NewObject, polyglot_from_string(name, SRC_CS));
 }
 
+PyModuleDef* PyModule_GetDef(PyObject* m) {
+    if (!PyModule_Check(m)) {
+        PyErr_BadArgument();
+        return NULL;
+    }
+    return ((PyModuleObject *)m)->md_def;
+}
+
 void* PyModule_GetState(PyObject *m) {
     if (!PyModule_Check(m)) {
         PyErr_BadArgument();

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/cext/capi/DynamicObjectNativeWrapper.java

@@ -875,10 +875,9 @@ static Object doTpDict(PythonClass object, @SuppressWarnings("unused") PythonNat
             return toSulongNode.execute(dict);
         }
 
-        @Specialization(guards = "eq(MD_DEF, key)", limit = "1")
-        static Object doMdDef(@SuppressWarnings("unused") PythonObject object, DynamicObjectNativeWrapper nativeWrapper, @SuppressWarnings("unused") String key,
-                        @CachedLibrary("nativeWrapper.getNativeMemberStore()") HashingStorageLibrary lib) {
-            return lib.getItem(nativeWrapper.getNativeMemberStore(), MD_DEF);
+        @Specialization(guards = "eq(MD_DEF, key)")
+        static Object doMdDef(PythonModule object, @SuppressWarnings("unused") DynamicObjectNativeWrapper nativeWrapper, @SuppressWarnings("unused") String key) {
+            return object.getNativeModuleDef();
         }
 
         @Specialization(guards = "eq(BUF_DELEGATE, key)")
@@ -1274,11 +1273,9 @@ static void doTpSubclasses(PythonClass object, @SuppressWarnings("unused") Pytho
                 hashLib.forEach(storage, eachNode, new SubclassAddState(storage, hashLib, subclasses));
             }
 
-            @Specialization(guards = "eq(MD_DEF, key)", limit = "1")
-            static void doMdDef(@SuppressWarnings("unused") PythonObject object, DynamicObjectNativeWrapper nativeWrapper, @SuppressWarnings("unused") String key, Object value,
-                            @CachedLanguage PythonLanguage lang,
-                            @CachedLibrary("nativeWrapper.createNativeMemberStore(lang)") HashingStorageLibrary lib) {
-                lib.setItem(nativeWrapper.createNativeMemberStore(lang), MD_DEF.getMemberName(), value);
+            @Specialization(guards = "eq(MD_DEF, key)")
+            static void doMdDef(PythonModule object, @SuppressWarnings("unused") DynamicObjectNativeWrapper nativeWrapper, @SuppressWarnings("unused") String key, Object value) {
+                object.setNativeModuleDef(value);
             }
 
             @Specialization(guards = "eq(TP_DICT, key)", limit = "1")