gh-134819: Add sys.set_object_tags and sys.get_object_tags

Author: corona10

Lib/test/test_sys.py

@@ -753,23 +753,25 @@ def test_43581(self):
         self.assertEqual(sys.__stdout__.encoding, sys.__stderr__.encoding)
 
     def test_intern(self):
-        has_is_interned = (test.support.check_impl_detail(cpython=True)
-                           or hasattr(sys, '_is_interned'))
         self.assertRaises(TypeError, sys.intern)
         self.assertRaises(TypeError, sys.intern, b'abc')
+        has_is_interned = (test.support.check_impl_detail(cpython=True)
+                           or hasattr(sys, '_is_interned'))
         if has_is_interned:
             self.assertRaises(TypeError, sys._is_interned)
             self.assertRaises(TypeError, sys._is_interned, b'abc')
+
+        def _is_interned(obj):
+            tags = sys.get_object_tags(obj)
+            return tags.get("interned", False)
+
         s = "never interned before" + str(random.randrange(0, 10**9))
         self.assertTrue(sys.intern(s) is s)
-        if has_is_interned:
-            self.assertIs(sys._is_interned(s), True)
+        self.assertIs(_is_interned(s), True)
         s2 = s.swapcase().swapcase()
-        if has_is_interned:
-            self.assertIs(sys._is_interned(s2), False)
+        self.assertIs(_is_interned(s2), False)
         self.assertTrue(sys.intern(s2) is s)
-        if has_is_interned:
-            self.assertIs(sys._is_interned(s2), False)
+        self.assertIs(_is_interned(s2), False)
 
         # Subclasses of string can't be interned, because they
         # provide too much opportunity for insane things to happen.
@@ -781,8 +783,7 @@ def __hash__(self):
                 return 123
 
         self.assertRaises(TypeError, sys.intern, S("abc"))
-        if has_is_interned:
-            self.assertIs(sys._is_interned(S("abc")), False)
+        self.assertIs(_is_interned(S("abc")), False)
 
     @support.cpython_only
     @requires_subinterpreters
@@ -847,7 +848,8 @@ def test_subinterp_intern_singleton(self):
                     assert id(s) == {id(s)}
                     t = sys.intern(s)
                     '''))
-                self.assertTrue(sys._is_interned(s))
+                tags = sys.get_object_tags(s)
+                self.assertTrue(tags.get("interned", False))
 
     def test_sys_flags(self):
         self.assertTrue(sys.flags)

Python/clinic/sysmodule.c.h

@@ -1052,6 +1052,65 @@ sys__is_immortal_impl(PyObject *module, PyObject *op)
     return PyUnstable_IsImmortal(op);
 }
 
+
+/*[clinic input]
+sys.get_object_tags -> object
+
+  op: object
+  /
+Return the tags of the given object.
+[clinic start generated code]*/
+
+static PyObject *
+sys_get_object_tags(PyObject *module, PyObject *op)
+/*[clinic end generated code: output=a68da7f1805c9216 input=75993fb67096e2ff]*/
+{
+    assert(op != NULL);
+    PyObject *dict = PyDict_New();
+    if (dict == NULL) {
+        return NULL;
+    }
+    if (PyUnstable_IsImmortal(op)) {
+        if (PyDict_SetItemString(dict, "immortal", Py_True) < 0) {
+            Py_DECREF(dict);
+            return NULL;
+        }
+    }
+    else {
+        if (PyDict_SetItemString(dict, "immortal", Py_False) < 0) {
+            Py_DECREF(dict);
+            return NULL;
+        }
+    }
+
+    if (PyUnicode_CHECK_INTERNED(op)) {
+        if (PyDict_SetItemString(dict, "interned", Py_True) < 0) {
+            Py_DECREF(dict);
+            return NULL;
+        }
+    }
+    else {
+        if (PyDict_SetItemString(dict, "interned", Py_False) < 0) {
+            Py_DECREF(dict);
+            return NULL;
+        }
+    }
+
+    if (PyUnstable_Object_EnableDeferredRefcount(op)) {
+        if (PyDict_SetItemString(dict, "deferred_refcount", Py_True) < 0) {
+            Py_DECREF(dict);
+            return NULL;
+        }
+    }
+    else {
+        if (PyDict_SetItemString(dict, "deferred_refcount", Py_False) < 0) {
+            Py_DECREF(dict);
+            return NULL;
+        }
+    }
+    return dict;
+}
+
 /*
  * Cached interned string objects used for calling the profile and
  * trace functions.
@@ -2796,6 +2855,7 @@ static PyMethodDef sys_methods[] = {
     SYS__IS_IMMORTAL_METHODDEF
     SYS_INTERN_METHODDEF
     SYS__IS_INTERNED_METHODDEF
+    SYS_GET_OBJECT_TAGS_METHODDEF
     SYS_IS_FINALIZING_METHODDEF
     SYS_MDEBUG_METHODDEF
     SYS_SETSWITCHINTERVAL_METHODDEF