PyUnicodeWriter_WriteRepr(NULL) now writes "<NULL>"

docs/api.rst

@@ -262,6 +262,7 @@ Not supported:
 * ``PyInitConfig_SetStrList()``
 * ``PyType_GetBaseByToken()``
 * ``PyUnicodeWriter_DecodeUTF8Stateful()``
+* ``PyUnicodeWriter_WriteUCS4()``
 * ``Py_InitializeFromInitConfig()``
 
 

pythoncapi_compat.h

@@ -1425,6 +1425,11 @@ PyUnicodeWriter_WriteStr(PyUnicodeWriter *writer, PyObject *obj)
 static inline int
 PyUnicodeWriter_WriteRepr(PyUnicodeWriter *writer, PyObject *obj)
 {
+    if (obj == NULL) {
+        return _PyUnicodeWriter_WriteASCIIString((_PyUnicodeWriter*)writer,
+                                                 "<NULL>", 6);
+    }
+
     PyObject *str = PyObject_Repr(obj);
     if (str == NULL) {
         return -1;

tests/test_pythoncapi_compat_cext.c

@@ -1908,13 +1908,17 @@ test_unicodewriter(PyObject *Py_UNUSED(self), PyObject *Py_UNUSED(args))
         goto error;
     }
     Py_CLEAR(str);
+    if (PyUnicodeWriter_WriteRepr(writer, NULL) < 0) {
+        goto error;
+    }
 
     {
         PyObject *result = PyUnicodeWriter_Finish(writer);
         if (result == NULL) {
             return NULL;
         }
-        assert(PyUnicode_EqualToUTF8(result, "var=long non-ASCII valu\xC3\xA9 'repr'"));
+        const char *expected = "var=long non-ASCII valu\xC3\xA9 'repr'<NULL>";
+        assert(PyUnicode_EqualToUTF8(result, expected));
         Py_DECREF(result);
     }