Update staticmethod.rs

HyunP-dev · HyunP-dev · commit 8fce0e323c3b · 2022-05-22T14:04:31.000+09:00
diff --git a/vm/src/builtins/staticmethod.rs b/vm/src/builtins/staticmethod.rs
@@ -3,8 +3,9 @@ use crate::{
     builtins::builtinfunc::PyBuiltinMethod,
     class::PyClassImpl,
     function::{FuncArgs, IntoPyNativeFunc},
-    types::{Callable, Constructor, GetDescriptor},
+    types::{Callable, Constructor, GetDescriptor, Initializer},
     Context, PyObjectRef, PyPayload, PyRef, PyResult, VirtualMachine,
+    AsObject
 };
 
 #[pyclass(module = false, name = "staticmethod")]
@@ -62,7 +63,17 @@ impl PyStaticMethod {
     }
 }
 
-#[pyimpl(with(Callable, GetDescriptor, Constructor), flags(BASETYPE, HAS_DICT))]
+impl Initializer for PyStaticMethod {
+    type Args = PyObjectRef;
+
+    fn init(zelf: PyRef<Self>, args: Self::Args, vm: &VirtualMachine) -> PyResult<()> {
+        zelf.as_object().set_attr("__doc__", args.get_attr("__doc__", vm)?, vm)?;
+        // zelf.as_object().set_attr("__format__", args.get_attr("__format__", vm)?, vm);
+        Ok(())
+    }
+}
+
+#[pyimpl(with(Callable, GetDescriptor, Constructor, Initializer), flags(BASETYPE, HAS_DICT))]
 impl PyStaticMethod {
     #[pyproperty(magic)]
     fn isabstractmethod(&self, vm: &VirtualMachine) -> PyObjectRef {
@@ -72,11 +83,46 @@ impl PyStaticMethod {
         }
     }
 
+    #[pyproperty(magic)]
+    fn func(&self, vm: &VirtualMachine) -> PyObjectRef {
+        self.callable.clone()
+    }
+
+    #[pyproperty(magic)]
+    fn wrapped(&self, vm: &VirtualMachine) -> PyObjectRef {
+        self.callable.clone()
+    }
+
     #[pyproperty(magic, setter)]
     fn set_isabstractmethod(&self, value: PyObjectRef, vm: &VirtualMachine) -> PyResult<()> {
         self.callable.set_attr("__isabstractmethod__", value, vm)?;
         Ok(())
     }
+
+    #[pymethod(magic)]
+    fn repr(&self, vm: &VirtualMachine) -> PyResult<String> {
+        Ok(format!("<staticmethod({})>", self.callable.repr(vm)?))
+    }
+
+    #[pyproperty(magic)]
+    fn module(&self, vm: &VirtualMachine) -> PyResult {
+        self.callable.get_attr("__module__", vm)
+    }
+
+    #[pyproperty(magic)]
+    fn name(&self, vm: &VirtualMachine) -> PyResult {
+        self.callable.get_attr("__name__", vm)
+    }
+
+    #[pyproperty(magic)]
+    fn qualname(&self, vm: &VirtualMachine) -> PyResult {
+        self.callable.get_attr("__qualname__", vm)
+    }
+
+    #[pyproperty(magic)]
+    fn annotations(&self, vm: &VirtualMachine) -> PyResult {
+        self.callable.get_attr("__annotations__", vm)
+    }
 }
 
 impl Callable for PyStaticMethod {
