annotate dunder getter/setter

Author: xuantengh

Objects/clinic/funcobject.c.h

@@ -680,41 +680,6 @@ func_set_qualname(PyObject *self, PyObject *value, void *Py_UNUSED(ignored))
     return 0;
 }
 
-static PyObject *
-func_get_annotate(PyObject *self, void *Py_UNUSED(ignored))
-{
-    PyFunctionObject *op = _PyFunction_CAST(self);
-    if (op->func_annotate == NULL) {
-        Py_RETURN_NONE;
-    }
-    return Py_NewRef(op->func_annotate);
-}
-
-static int
-func_set_annotate(PyObject *self, PyObject *value, void *Py_UNUSED(ignored))
-{
-    PyFunctionObject *op = _PyFunction_CAST(self);
-    if (value == NULL) {
-        PyErr_SetString(PyExc_TypeError,
-            "__annotate__ cannot be deleted");
-        return -1;
-    }
-    if (Py_IsNone(value)) {
-        Py_XSETREF(op->func_annotate, value);
-        return 0;
-    }
-    else if (PyCallable_Check(value)) {
-        Py_XSETREF(op->func_annotate, Py_XNewRef(value));
-        Py_CLEAR(op->func_annotations);
-        return 0;
-    }
-    else {
-        PyErr_SetString(PyExc_TypeError,
-            "__annotate__ must be callable or None");
-        return -1;
-    }
-}
-
 static PyObject *
 func_get_type_params(PyObject *self, void *Py_UNUSED(ignored))
 {
@@ -1101,12 +1066,62 @@ function___kwdefaults___set_impl(PyFunctionObject *self, PyObject *value)
     return 0;
 }
 
+/*[clinic input]
+@critical_section
+@getter
+function.__annotate__
+
+Get __annotate__ attribute for a function object.
+[clinic start generated code]*/
+
+static PyObject *
+function___annotate___get_impl(PyFunctionObject *self)
+/*[clinic end generated code: output=5ec7219ff2bda9e6 input=2932381f16ffeb75]*/
+{
+    if (self->func_annotate == NULL) {
+        Py_RETURN_NONE;
+    }
+    return Py_NewRef(self->func_annotate);
+}
+
+/*[clinic input]
+@critical_section
+@setter
+function.__annotate__
+[clinic start generated code]*/
+
+static int
+function___annotate___set_impl(PyFunctionObject *self, PyObject *value)
+/*[clinic end generated code: output=05b7dfc07ada66cd input=eb6225e358d97448]*/
+{
+    PyFunctionObject *op = _PyFunction_CAST(self);
+    if (value == NULL) {
+        PyErr_SetString(PyExc_TypeError,
+            "__annotate__ cannot be deleted");
+        return -1;
+    }
+    if (Py_IsNone(value)) {
+        Py_XSETREF(op->func_annotate, value);
+        return 0;
+    }
+    else if (PyCallable_Check(value)) {
+        Py_XSETREF(op->func_annotate, Py_XNewRef(value));
+        Py_CLEAR(op->func_annotations);
+        return 0;
+    }
+    else {
+        PyErr_SetString(PyExc_TypeError,
+            "__annotate__ must be callable or None");
+        return -1;
+    }
+}
+
 static PyGetSetDef func_getsetlist[] = {
     FUNCTION___CODE___GETSETDEF
     FUNCTION___DEFAULTS___GETSETDEF
     FUNCTION___KWDEFAULTS___GETSETDEF
     FUNCTION___ANNOTATIONS___GETSETDEF
-    {"__annotate__", func_get_annotate, func_set_annotate},
+    FUNCTION___ANNOTATE___GETSETDEF
     {"__dict__", PyObject_GenericGetDict, PyObject_GenericSetDict},
     {"__name__", func_get_name, func_set_name},
     {"__qualname__", func_get_qualname, func_set_qualname},