apply mapping protocol in itemprotocol

Snowapril · Snowapril · commit 63f0dd58a9a2 · 2021-09-25T14:11:18.000+09:00
diff --git a/vm/src/pyobject.rs b/vm/src/pyobject.rs
@@ -12,7 +12,7 @@ use crate::builtins::bytes;
 use crate::builtins::code;
 use crate::builtins::code::PyCodeRef;
 use crate::builtins::complex::PyComplex;
-use crate::builtins::dict::{PyDict, PyDictRef};
+use crate::builtins::dict::{PyDict, PyDictRef, PyMapping};
 use crate::builtins::float::PyFloat;
 use crate::builtins::function::PyBoundMethod;
 use crate::builtins::getset::{IntoPyGetterFunc, IntoPySetterFunc, PyGetSet};
@@ -646,6 +646,12 @@ where
     T: IntoPyObject,
 {
     fn get_item(&self, key: T, vm: &VirtualMachine) -> PyResult {
+        if let Ok(map) = PyMapping::try_from_borrowed_object(vm, self) {
+            if let Some(getitem) = map.subscript {
+                return getitem(self.clone(), key.into_pyobject(vm), vm);
+            }
+        }
+
         match vm.get_special_method(self.clone(), "__getitem__")? {
             Ok(special_method) => return special_method.invoke((key,), vm),
             Err(obj) => {
@@ -663,6 +669,12 @@ where
     }
 
     fn set_item(&self, key: T, value: PyObjectRef, vm: &VirtualMachine) -> PyResult<()> {
+        if let Ok(map) = PyMapping::try_from_borrowed_object(vm, self) {
+            if let Some(setitem) = map.ass_subscript {
+                return setitem(self.clone(), key.into_pyobject(vm), value, vm);
+            }
+        }
+
         vm.get_special_method(self.clone(), "__setitem__")?
             .map_err(|obj| {
                 vm.new_type_error(format!(
