Add __new__ changes

eduardo-elizondo · eduardo-elizondo · commit c1a6d030b199 · 2019-09-16T23:31:34.000-07:00
diff --git a/Modules/posixmodule.c b/Modules/posixmodule.c
@@ -2077,16 +2077,12 @@ static PyStructSequence_Desc waitid_result_desc = {
 #endif
 
 static PyObject *
-statresult_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
+statresult_new(PyTypeObject *type, PyObject *args)
 {
     PyObject *sequence, *kwds;
     PyStructSequence *result;
     int i;
 
-    if (!_PyArg_NoKeywords("StatResult", kwargs)) {
-        return NULL;
-    }
-
     /* Remove the cls object from the argument list */
     sequence = PyTuple_GetSlice(args, 1, PyTuple_Size(args));
     if (!sequence) {
@@ -6253,20 +6249,24 @@ PyDoc_STRVAR(os_sched_param__doc__,
 "    A scheduling parameter.");
 
 static PyObject *
-os_sched_param(PyObject *module, PyObject *args, PyObject *kwargs)
+os_sched_param(PyTypeObject *type, PyObject *args, PyObject *kwargs)
 {
     static char* _keywords[] = {"sched_priority", NULL};
     static const char * _format = "O:sched_param";
-    PyObject *sched_priority, *res;
+    PyObject *sequence, *sched_priority, *res;
 
+    /* Remove the cls object from the argument list */
+    sequence = PyTuple_GetSlice(args, 1, PyTuple_Size(args));
+    if (!sequence) {
+        return NULL;
+    }
     int result = PyArg_ParseTupleAndKeywords(args, kwargs, _format, _keywords,
         &sched_priority);
+    Py_DECREF(sequence);
     if (!result) {
         return NULL;
     }
-    res = PyStructSequence_New(
-        (PyTypeObject*)_posixstate(module)->SchedParamType
-    );
+    res = PyStructSequence_New((PyTypeObject *)type);
     if (!res) {
         return NULL;
     }
@@ -6275,22 +6275,6 @@ os_sched_param(PyObject *module, PyObject *args, PyObject *kwargs)
     return res;
 }
 
-static PyObject *
-sched_param_new(PyTypeObject *type, PyObject *args, PyObject *kwargs) {
-    PyObject *module = PyState_FindModule(&posixmodule);
-    if (!module) {
-        return NULL;
-    }
-    /* Remove the cls object from the argument list */
-    PyObject *sequence = PyTuple_GetSlice(args, 1, PyTuple_Size(args));
-    if (!sequence) {
-        return NULL;
-    }
-    PyObject *result = os_sched_param(module, args, kwargs);
-    Py_DECREF(sequence);
-    return result;
-}
-
 static PyStructSequence_Field sched_param_fields[] = {
     {"sched_priority", "the scheduling priority"},
     {0}
@@ -13593,6 +13577,16 @@ os__remove_dll_directory_impl(PyObject *module, PyObject *cookie)
 
 #endif
 
+#if defined(HAVE_SCHED_SETPARAM) || defined(HAVE_SCHED_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDULER) || defined(POSIX_SPAWN_SETSCHEDPARAM)
+static PyMethodDef SchedParamType_dunder_new = {
+    "__new__", (PyCFunction)os_sched_param, METH_VARARGS | METH_KEYWORDS
+};
+#endif
+
+static PyMethodDef StatResultType_dunder_new = {
+    "__new__", (PyCFunction)statresult_new, METH_VARARGS
+};
+
 static PyMethodDef posix_methods[] = {
 
     OS_STAT_METHODDEF
@@ -14446,7 +14440,7 @@ static const char * const have_functions[] = {
 PyMODINIT_FUNC
 INITFUNC(void)
 {
-    PyObject *m, *v;
+    PyObject *m, *v, *dunder_new;
     PyObject *list;
     const char * const *trace;
 
@@ -14504,10 +14498,14 @@ INITFUNC(void)
     Py_INCREF(StatResultType);
     PyModule_AddObject(m, "stat_result", StatResultType);
     _posixstate(m)->StatResultType = StatResultType;
-
     /* Add a custom __new__ to the structsequence */
-    _posixstate(m)->structseq_new = ((PyTypeObject *)StatResultType)->tp_new;
-    ((PyTypeObject *)StatResultType)->tp_new = statresult_new;
+    _posixstate(m)->structseq_new = (newfunc)PyType_GetSlot((PyTypeObject *)StatResultType, Py_tp_new);
+    dunder_new = PyDescr_NewClassMethod((PyTypeObject *)StatResultType, &StatResultType_dunder_new);
+    if (dunder_new == NULL) {
+        return NULL;
+    }
+    PyObject_SetAttrString(StatResultType, "__new__", dunder_new);
+    Py_DECREF(dunder_new);
 
     statvfs_result_desc.name = "os.statvfs_result"; /* see issue #19209 */
     PyObject *StatVFSResultType = (PyObject *)PyStructSequence_NewType(&statvfs_result_desc);
@@ -14537,7 +14535,12 @@ INITFUNC(void)
     PyModule_AddObject(m, "sched_param", SchedParamType);
     _posixstate(m)->SchedParamType = SchedParamType;
     /* Add a custom __new__ to the structsequence */
-    ((PyTypeObject *)SchedParamType)->tp_new = sched_param_new;
+    dunder_new = PyDescr_NewClassMethod((PyTypeObject *)SchedParamType, &SchedParamType_dunder_new);
+    if (dunder_new == NULL) {
+        return NULL;
+    }
+    PyObject_SetAttrString((PyObject *)SchedParamType, "__new__", dunder_new);
+    Py_DECREF(dunder_new);
 #endif
 
     /* initialize TerminalSize_info */
