gh-145119: Allow frozendict to be assigned to instance __dict__

Author: mohsinm-dev

Lib/test/test_descr.py

@@ -3570,6 +3570,35 @@ class Exception2(Base, Exception):
             self.assertEqual(e.a, 1)
             self.assertEqual(can_delete_dict(e), can_delete_dict(ValueError()))
 
+    def test_set_dict_to_frozendict(self):
+        # gh-145119: __dict__ accepts frozendict.
+        class C:
+            pass
+
+        obj = C()
+        obj.__dict__ = frozendict(x=1, y=2)
+        self.assertEqual(obj.x, 1)
+        self.assertEqual(obj.y, 2)
+        self.assertIn("x", dir(obj))
+        self.assertIn("y", dir(obj))
+        self.assertEqual(type(vars(obj)), frozendict)
+
+        with self.assertRaises(TypeError):
+            obj.z = 3
+        with self.assertRaises(TypeError):
+            del obj.x
+
+        class MyFrozenDict(frozendict):
+            pass
+
+        obj.__dict__ = MyFrozenDict(a=10)
+        self.assertEqual(obj.a, 10)
+        self.assertIn("a", dir(obj))
+
+        obj.__dict__ = {"w": 50}
+        obj.q = 99
+        self.assertEqual(obj.q, 99)
+
     def test_binary_operator_override(self):
         # Testing overrides of binary operations...
         class I(int):

Misc/NEWS.d/next/Core_and_Builtins/2026-02-22-23-50-00.gh-issue-145119.hN4sQ8.rst

@@ -0,0 +1,2 @@
+Allow :class:`frozendict` to be assigned to an instance's
+:attr:`~object.__dict__`, enabling immutable instances.

Objects/dictobject.c

@@ -7064,6 +7064,17 @@ int
 _PyDict_SetItem_LockHeld(PyDictObject *dict, PyObject *name, PyObject *value)
 {
     if (!PyDict_Check(dict)) {
+        if (PyFrozenDict_Check((PyObject *)dict)) {
+            if (value == NULL) {
+                PyErr_SetString(PyExc_TypeError,
+                                "'frozendict' object does not support item deletion");
+            }
+            else {
+                PyErr_SetString(PyExc_TypeError,
+                                "'frozendict' object does not support item assignment");
+            }
+            return -1;
+        }
         PyErr_BadInternalCall();
         return -1;
     }

Objects/typeobject.c

@@ -3999,7 +3999,7 @@ subtype_dict(PyObject *obj, void *context)
 int
 _PyObject_SetDict(PyObject *obj, PyObject *value)
 {
-    if (value != NULL && !PyDict_Check(value)) {
+    if (value != NULL && !PyAnyDict_Check(value)) {
         PyErr_Format(PyExc_TypeError,
                      "__dict__ must be set to a dictionary, "
                      "not a '%.200s'", Py_TYPE(value)->tp_name);
@@ -8305,15 +8305,24 @@ object___dir___impl(PyObject *self)
     if (dict == NULL) {
         dict = PyDict_New();
     }
-    else if (!PyDict_Check(dict)) {
-        Py_DECREF(dict);
-        dict = PyDict_New();
-    }
-    else {
+    else if (PyDict_Check(dict)) {
         /* Copy __dict__ to avoid mutating it. */
         PyObject *temp = PyDict_Copy(dict);
         Py_SETREF(dict, temp);
     }
+    else if (PyFrozenDict_Check(dict)) {
+        /* Convert frozendict to a mutable dict for merging. */
+        PyObject *temp = PyDict_New();
+        if (temp != NULL && PyDict_Update(temp, dict) < 0) {
+            Py_DECREF(temp);
+            temp = NULL;
+        }
+        Py_SETREF(dict, temp);
+    }
+    else {
+        Py_DECREF(dict);
+        dict = PyDict_New();
+    }
 
     if (dict == NULL)
         goto error;