From 758d52909f3373da0ffcd38fb3789a9a69a6fdfc Mon Sep 17 00:00:00 2001
From: Moreal <dev.moreal@gmail.com>
Date: Fri, 28 Jan 2022 23:03:08 +0900
Subject: [PATCH] Add `type.__name__` setter

---
 vm/src/builtins/pytype.rs | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/vm/src/builtins/pytype.rs b/vm/src/builtins/pytype.rs
index 1cafd20930..396bb204ff 100644
--- a/vm/src/builtins/pytype.rs
+++ b/vm/src/builtins/pytype.rs
@@ -545,6 +545,29 @@ impl PyType {
         ))
     }
 
+    #[pyproperty(magic, setter)]
+    fn set_name(&self, value: PyObjectRef, vm: &VirtualMachine) -> PyResult<()> {
+        if !self.slots.flags.has_feature(PyTypeFlags::HEAPTYPE) {
+            Err(vm.new_type_error(format!(
+                "cannot set '{}' attribute of immutable type '{}'",
+                "__name__",
+                self.name()
+            )))
+        } else {
+            match value.downcast_ref::<PyStr>().map(|m| m.as_str()) {
+                Some(name) => {
+                    *self.slots.name.write() = Some(name.to_string());
+                    Ok(())
+                }
+                None => Err(vm.new_type_error(format!(
+                    "can only assign string to {}.__name__, not '{}'",
+                    self.name(),
+                    value.class().name()
+                ))),
+            }
+        }
+    }
+
     #[pyproperty(magic)]
     fn text_signature(&self) -> Option<String> {
         self.slots