Add PyThreadState_GetUnchecked() (#77)

vstinner · web-flow · commit f78c780daa7b · 2023-10-03T19:10:59.000+02:00
diff --git a/docs/api.rst b/docs/api.rst
@@ -107,6 +107,12 @@ Python 3.13
 
    See `PyObject_ClearManagedDict() documentation <https://docs.python.org/dev/c-api/object.html#c.PyObject_ClearManagedDict>`__.
 
+.. c:function:: PyThreadState* PyThreadState_GetUnchecked(void)
+
+   See `PyThreadState_GetUnchecked() documentation <https://docs.python.org/dev/c-api/init.html#c.PyThreadState_GetUnchecked>`__.
+
+   Available on Python 3.5.2 and newer.
+
 
 Python 3.12
 -----------
diff --git a/docs/changelog.rst b/docs/changelog.rst
@@ -1,8 +1,12 @@
 Changelog
 =========
 
-* 2023-10-03: Add ``PyObject_VisitManagedDict()`` and
-  ``PyObject_ClearManagedDict()`` functions.
+* 2023-10-03: Add functions:
+
+  * ``PyObject_VisitManagedDict()``
+  * ``PyObject_ClearManagedDict()``
+  * ``PyThreadState_GetUnchecked()``
+
 * 2023-09-29: Add functions:
 
   * ``PyMapping_HasKeyWithError()``
diff --git a/pythoncapi_compat.h b/pythoncapi_compat.h
@@ -929,6 +929,16 @@ PyObject_ClearManagedDict(PyObject *obj)
 }
 #endif
 
+// gh-108867 added PyThreadState_GetUnchecked() to Python 3.13.0a1
+// Python 3.5.2 added _PyThreadState_UncheckedGet().
+#if PY_VERSION_HEX >= 0x03050200 && PY_VERSION_HEX < 0x030D00A1
+static inline PyThreadState*
+PyThreadState_GetUnchecked(void)
+{
+    return _PyThreadState_UncheckedGet();
+}
+#endif
+
 
 #ifdef __cplusplus
 }
diff --git a/tests/test_pythoncapi_compat_cext.c b/tests/test_pythoncapi_compat_cext.c
@@ -318,6 +318,11 @@ test_thread_state(PyObject *Py_UNUSED(module), PyObject* Py_UNUSED(ignored))
     PyThreadState_LeaveTracing(tstate);
 #endif
 
+#if PY_VERSION_HEX >= 0x03050200
+    // PyThreadState_GetUnchecked()
+    assert(PyThreadState_GetUnchecked() == tstate);
+#endif
+
     Py_RETURN_NONE;
 }
 
