Add PySys_GetAttr() function

Author: vstinner

docs/api.rst

@@ -26,6 +26,25 @@ Latest version of the header file:
 `pythoncapi_compat.h <https://raw.githubusercontent.com/python/pythoncapi-compat/main/pythoncapi_compat.h>`_.
 
 
+Python 3.15
+-----------
+
+.. c:function:: PyObject* PySys_GetAttr(const char *name)
+
+ See `PySys_GetAttr() documentation <https://docs.python.org/dev/c-api/sys.html#c.PySys_GetAttr>`__.
+
+.. c:function:: PyObject* PySys_GetAttrString(const char *name)
+
+ See `PySys_GetAttrString() documentation <https://docs.python.org/dev/c-api/sys.html#c.PySys_GetAttrString>`__.
+
+.. c:function:: PyObject* PySys_GetOptionalAttr(const char *name)
+
+ See `PySys_GetOptionalAttr() documentation <https://docs.python.org/dev/c-api/sys.html#c.PySys_GetOptionalAttr>`__.
+
+.. c:function:: PyObject* PySys_GetOptionalAttrString(const char *name)
+
+ See `PySys_GetOptionalAttrString() documentation <https://docs.python.org/dev/c-api/sys.html#c.PySys_GetOptionalAttrString>`__.
+
 Python 3.14
 -----------
 

docs/changelog.rst

@@ -1,6 +1,13 @@
 Changelog
 =========
 
+* 2025-06-03: Add functions:
+
+ * ``PySys_GetAttr()``
+ * ``PySys_GetAttrString()``
+ * ``PySys_GetOptionalAttr()``
+ * ``PySys_GetOptionalAttrString()``
+
 * 2025-01-19: Add ``PyConfig_Get()`` functions.
 * 2025-01-06: Add ``Py_fopen()`` and ``Py_fclose()`` functions.
 * 2024-12-16: Add ``structmember.h`` constants:

pythoncapi_compat.h

@@ -2199,6 +2199,70 @@ PyConfig_GetInt(const char *name, int *value)
 #endif // PY_VERSION_HEX > 0x03090000 && !defined(PYPY_VERSION)
 
 
+#if PY_VERSION_HEX < 0x030F0000
+static inline PyObject*
+PySys_GetAttrString(const char *name)
+{
+#if PY_VERSION_HEX >= 0x03000000
+ PyObject *value = Py_XNewRef(PySys_GetObject(name));
+#else
+ PyObject *value = Py_XNewRef(PySys_GetObject((char*)name));
+#endif
+ if (value != NULL) {
+ return value;
+ }
+ if (!PyErr_Occurred()) {
+ PyErr_Format(PyExc_RuntimeError, "lost sys.%s", name);
+ }
+ return NULL;
+}
+
+static inline PyObject*
+PySys_GetAttr(PyObject *name)
+{
+#if PY_VERSION_HEX >= 0x03000000
+ const char *name_str = PyUnicode_AsUTF8(name);
+#else
+ const char *name_str = PyString_AsString(name);
+#endif
+ if (name_str == NULL) {
+ return NULL;
+ }
+
+ return PySys_GetAttrString(name_str);
+}
+
+static inline int
+PySys_GetOptionalAttrString(const char *name, PyObject **value)
+{
+#if PY_VERSION_HEX >= 0x03000000
+ *value = Py_XNewRef(PySys_GetObject(name));
+#else
+ *value = Py_XNewRef(PySys_GetObject((char*)name));
+#endif
+ if (*value != NULL) {
+ return 1;
+ }
+ return 0;
+}
+
+static inline int
+PySys_GetOptionalAttr(PyObject *name, PyObject **value)
+{
+#if PY_VERSION_HEX >= 0x03000000
+ const char *name_str = PyUnicode_AsUTF8(name);
+#else
+ const char *name_str = PyString_AsString(name);
+#endif
+ if (name_str == NULL) {
+ return -1;
+ }
+
+ return PySys_GetOptionalAttrString(name_str, value);
+}
+#endif // PY_VERSION_HEX < 0x030F00A1
+
+
 #ifdef __cplusplus
 }
 #endif

tests/test_pythoncapi_compat_cext.c

@@ -2181,6 +2181,64 @@ test_config(PyObject *Py_UNUSED(module), PyObject *Py_UNUSED(args))
 #endif
 
 
+static PyObject *
+test_sys(PyObject *Py_UNUSED(module), PyObject *Py_UNUSED(args))
+{
+ const char *stdout_str = "stdout";
+ PyObject *stdout_obj = create_string(stdout_str);
+#if PYTHON3
+ PyObject *sys_stdout = PySys_GetObject(stdout_str); // borrowed ref
+#else
+ PyObject *sys_stdout = PySys_GetObject((char*)stdout_str); // borrowed ref
+#endif
+ const char *nonexistent_str = "nonexistent";
+ PyObject *nonexistent_obj = create_string(nonexistent_str);
+ PyObject *value;
+
+ // get sys.stdout
+ value = PySys_GetAttr(stdout_obj);
+ assert(value == sys_stdout);
+ Py_DECREF(value);
+
+ value = PySys_GetAttrString(stdout_str);
+ assert(value == sys_stdout);
+ Py_DECREF(value);
+
+ value = UNINITIALIZED_OBJ;
+ assert(PySys_GetOptionalAttr(stdout_obj, &value) == 1);
+ assert(value == sys_stdout);
+ Py_DECREF(value);
+
+ value = UNINITIALIZED_OBJ;
+ assert(PySys_GetOptionalAttrString(stdout_str, &value) == 1);
+ assert(value == sys_stdout);
+ Py_DECREF(value);
+
+ // non existent attribute
+ value = PySys_GetAttr(nonexistent_obj);
+ assert(value == NULL);
+ assert(PyErr_ExceptionMatches(PyExc_RuntimeError));
+ PyErr_Clear();
+
+ value = PySys_GetAttrString(nonexistent_str);
+ assert(value == NULL);
+ assert(PyErr_ExceptionMatches(PyExc_RuntimeError));
+ PyErr_Clear();
+
+ value = UNINITIALIZED_OBJ;
+ assert(PySys_GetOptionalAttr(nonexistent_obj, &value) == 0);
+ assert(value == NULL);
+
+ value = UNINITIALIZED_OBJ;
+ assert(PySys_GetOptionalAttrString(nonexistent_str, &value) == 0);
+ assert(value == NULL);
+
+ Py_DECREF(stdout_obj);
+ Py_DECREF(nonexistent_obj);
+ Py_RETURN_NONE;
+}
+
+
 static struct PyMethodDef methods[] = {
  {"test_object", test_object, METH_NOARGS, _Py_NULL},
  {"test_py_is", test_py_is, METH_NOARGS, _Py_NULL},
@@ -2232,6 +2290,7 @@ static struct PyMethodDef methods[] = {
 #if 0x03090000 <= PY_VERSION_HEX && !defined(PYPY_VERSION)
  {"test_config", test_config, METH_NOARGS, _Py_NULL},
 #endif
+ {"test_sys", test_sys, METH_NOARGS, _Py_NULL},
  {_Py_NULL, _Py_NULL, 0, _Py_NULL}
 };