Add weakproxy mapping

Author: yonmilk

Lib/test/test_array.py

@@ -1028,8 +1028,6 @@ def test_buffer(self):
         self.assertRaises(BufferError, operator.delitem, a, slice(0, 1))
         self.assertEqual(m.tobytes(), expected)
 
-    # TODO: RUSTPYTHON
-    @unittest.expectedFailure
     def test_weakref(self):
         s = array.array(self.typecode, self.example)
         p = weakref.proxy(s)

Lib/test/test_weakref.py

@@ -394,8 +394,6 @@ def check_proxy(self, o, proxy):
         self.assertFalse(hasattr(o, 'foo'),
                      "object does not reflect attribute removal via proxy")
 
-    # TODO: RUSTPYTHON
-    @unittest.expectedFailure
     def test_proxy_deletion(self):
         # Test clearing of SF bug #762891
         class Foo:

vm/src/builtins/weakproxy.rs

@@ -2,8 +2,8 @@ use super::{PyStrRef, PyType, PyTypeRef, PyWeak};
 use crate::{
     class::PyClassImpl,
     function::OptionalArg,
-    protocol::{PySequence, PySequenceMethods},
-    types::{AsSequence, Constructor, GetAttr, SetAttr},
+    protocol::{PyMappingMethods, PySequence, PySequenceMethods},
+    types::{AsMapping, AsSequence, Constructor, GetAttr, SetAttr},
     Context, Py, PyObjectRef, PyPayload, PyRef, PyResult, VirtualMachine,
 };
 
@@ -58,7 +58,7 @@ crate::common::static_cell! {
     static WEAK_SUBCLASS: PyTypeRef;
 }
 
-#[pyimpl(with(GetAttr, SetAttr, Constructor, AsSequence))]
+#[pyimpl(with(GetAttr, SetAttr, Constructor, AsSequence, AsMapping))]
 impl PyWeakProxy {
     #[pymethod(magic)]
     fn str(&self, vm: &VirtualMachine) -> PyResult<PyStrRef> {
@@ -81,6 +81,32 @@ impl PyWeakProxy {
             None => Err(new_reference_error(vm)),
         }
     }
+
+    fn getitem(&self, needle: PyObjectRef, vm: &VirtualMachine) -> PyResult {
+        match self.weak.upgrade() {
+            Some(obj) => obj.get_item(&*needle, vm),
+            None => Err(new_reference_error(vm)),
+        }
+    }
+
+    fn setitem(
+        &self,
+        needle: PyObjectRef,
+        value: PyObjectRef,
+        vm: &VirtualMachine,
+    ) -> PyResult<()> {
+        match self.weak.upgrade() {
+            Some(obj) => obj.set_item(&*needle, value, vm),
+            None => Err(new_reference_error(vm)),
+        }
+    }
+
+    fn delitem(&self, needle: PyObjectRef, vm: &VirtualMachine) -> PyResult<()> {
+        match self.weak.upgrade() {
+            Some(obj) => obj.del_item(&*needle, vm),
+            None => Err(new_reference_error(vm)),
+        }
+    }
 }
 
 fn new_reference_error(vm: &VirtualMachine) -> PyRef<super::PyBaseException> {
@@ -125,6 +151,23 @@ impl AsSequence for PyWeakProxy {
     };
 }
 
+impl AsMapping for PyWeakProxy {
+    const AS_MAPPING: PyMappingMethods = PyMappingMethods {
+        length: Some(|mapping, vm| Self::mapping_downcast(mapping).len(vm)),
+        subscript: Some(|mapping, needle, vm| {
+            Self::mapping_downcast(mapping).getitem(needle.to_owned(), vm)
+        }),
+        ass_subscript: Some(|mapping, needle, value, vm| {
+            let zelf = Self::mapping_downcast(mapping);
+            if let Some(value) = value {
+                zelf.setitem(needle.to_owned(), value, vm)
+            } else {
+                zelf.delitem(needle.to_owned(), vm)
+            }
+        }),
+    };
+}
+
 pub fn init(context: &Context) {
     PyWeakProxy::extend_class(context, context.types.weakproxy_type);
 }