Use argument clinic

Author: markshannon

Include/internal/pycore_global_objects_fini_generated.h

@@ -411,6 +411,7 @@ struct _Py_global_strings {
         STRUCT_FOR_ID(errors)
         STRUCT_FOR_ID(event)
         STRUCT_FOR_ID(eventmask)
+        STRUCT_FOR_ID(events)
         STRUCT_FOR_ID(exc_type)
         STRUCT_FOR_ID(exc_value)
         STRUCT_FOR_ID(excepthook)
@@ -748,6 +749,7 @@ struct _Py_global_strings {
         STRUCT_FOR_ID(times)
         STRUCT_FOR_ID(timetuple)
         STRUCT_FOR_ID(timeunit)
+        STRUCT_FOR_ID(tool)
         STRUCT_FOR_ID(top)
         STRUCT_FOR_ID(trace_callback)
         STRUCT_FOR_ID(traceback)

Include/internal/pycore_global_strings.h

@@ -1570,43 +1570,6 @@ code_offset_to_position(PyObject *self, PyObject* offset)
         _source_offset_converter, &pi.pi_endcolumn);
 }
 
-static PyObject*
-code_get_local_events(PyObject *self, PyObject* tool) {
-    int tool_id;
-    int err = PyLong_AsInt32(tool, &tool_id);
-    if (err != 0) {
-        return NULL;
-    }
-    _PyMonitoringEventSet events;
-    err = _PyMonitoring_GetLocalEvents(self, tool_id, &events);
-    if (err != 0) {
-        return NULL;
-    }
-    return PyLong_FromUInt32(events);
-}
-
-static PyObject*
-code_set_local_events(PyObject *self, PyObject *const *args, Py_ssize_t nargs) {
-    int tool_id;
-    _PyMonitoringEventSet events;
-    if (!_PyArg_CheckPositional("__set_local_events__", nargs, 2, 2)) {
-        return NULL;
-    }
-    int err = PyLong_AsInt32(args[0], &tool_id);
-    if (err != 0) {
-        return NULL;
-    }
-    err = PyLong_AsUInt32(args[1], &events);
-    if (err != 0) {
-        return NULL;
-    }
-    err = _PyMonitoring_SetLocalEvents(self, tool_id, events);
-    if (err != 0) {
-        return NULL;
-    }
-    Py_RETURN_NONE;
-}
-
 
 /******************
  * "extra" frame eval info (see PEP 523)
@@ -2450,6 +2413,48 @@ code__varname_from_oparg_impl(PyCodeObject *self, int oparg)
     return Py_NewRef(name);
 }
 
+
+/*[clinic input]
+code.__get_local_events__ -> int
+
+    tool: int
+
+[clinic start generated code]*/
+
+static int
+code___get_local_events___impl(PyCodeObject *self, int tool)
+/*[clinic end generated code: output=e66c1af8a8c6c2aa input=660474646897971f]*/
+{
+    _PyMonitoringEventSet events;
+    int err = _PyMonitoring_GetLocalEvents((PyObject *)self, tool, &events);
+    if (err != 0) {
+        return err;
+    }
+    return events;
+}
+
+
+/*[clinic input]
+code.__set_local_events__
+
+
+    tool: int
+    events: int
+
+[clinic start generated code]*/
+
+static PyObject *
+code___set_local_events___impl(PyCodeObject *self, int tool, int events)
+/*[clinic end generated code: output=d80a9ef8386753f7 input=68cdcca03494b242]*/
+{
+    int err = _PyMonitoring_SetLocalEvents((PyObject *)self, tool, events);
+    if (err < 0) {
+        return NULL;
+    }
+    Py_RETURN_NONE;
+}
+
+
 /* XXX code objects need to participate in GC? */
 
 static struct PyMethodDef code_methods[] = {
@@ -2462,8 +2467,8 @@ static struct PyMethodDef code_methods[] = {
     {"__replace__", _PyCFunction_CAST(code_replace), METH_FASTCALL|METH_KEYWORDS,
      PyDoc_STR("__replace__($self, /, **changes)\n--\n\nThe same as replace().")},
     {"co_offset_to_position", code_offset_to_position, METH_O},
-    {"__get_local_events__", code_get_local_events, METH_O},
-    {"__set_local_events__", _PyCFunction_CAST(code_set_local_events), METH_FASTCALL},
+    CODE___GET_LOCAL_EVENTS___METHODDEF
+    CODE___SET_LOCAL_EVENTS___METHODDEF
     {NULL, NULL}                /* sentinel */
 };