struct access to PyDateTime_CAPI

Author: timfel

graalpython/com.oracle.graal.python.cext/include/Python.h

@@ -119,6 +119,7 @@
 #include "pythread.h"
 #include "funcobject.h"
 #include "iterobject.h"
+#include "datetime.h"
 
 #define PY_TRUFFLE_CEXT ((void*)polyglot_import("python_cext"))
 #define PY_BUILTIN ((void*)polyglot_import("python_builtins"))

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

@@ -0,0 +1,10 @@
+#include "capi.h"
+
+POLYGLOT_DECLARE_TYPE(PyDateTime_CAPI);
+
+/** to be used from Java code only; returns the type ID for a PyDateTime_CAPI */
+extern PyObject* set_PyDateTime_CAPI_typeid(PyTypeObject* type) {
+    polyglot_invoke(PY_TRUFFLE_CEXT, "PyTruffle_Set_SulongType", type, polyglot_PyDateTime_CAPI_typeid());
+    return Py_True;
+}
+

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

@@ -481,6 +481,17 @@ Object doMemoryview(PMemoryView object, String key,
             throw new IllegalStateException("delegate of memoryview object is not native");
         }
 
+        protected static boolean isPyDateTimeCAPI(PythonObject object) {
+            return object.getPythonClass().getName().equals("PyDateTime_CAPI");
+        }
+
+        @Specialization(guards = "isPyDateTimeCAPI(object)")
+        Object doDatetimeCAPI(PythonObject object, String key,
+                        @Cached("create()") GetClassNode getClass,
+                        @Cached("create()") LookupAttributeInMRONode.Dynamic getAttrNode) {
+            return getToSulongNode().execute(getAttrNode.execute(getClass.execute(object), key));
+        }
+
         @Fallback
         Object doGeneric(Object object, String key) {
             // This is the preliminary generic case: There are native members we know that they

graalpython/lib-graalpython/python_cext.py

@@ -1196,6 +1196,7 @@ def DateTime_FromDateAndTimeAndFold(y, mon, d, h, m, s, usec, tz, fold, typ):
         def Time_FromTimeAndFold(h, m, s, us, tz, fold, typ):
             return typ(hour=h, minute=m, second=s, microsecond=us, tzinfo=tz, fold=fold)
 
+    import_c_func("set_PyDateTime_CAPI_typeid")(PyDateTime_CAPI)
     datetime.datetime_CAPI = PyCapsule("datetime.datetime_CAPI", PyDateTime_CAPI(), None)