fix mmap

youknowone · youknowone · commit 69e46c4ad510 · 2025-06-27T01:57:38.000+09:00
diff --git a/stdlib/src/mmap.rs b/stdlib/src/mmap.rs
@@ -431,12 +431,12 @@ mod mmap {
                     |mapping, _vm| Ok(PyMmap::mapping_downcast(mapping).__len__())
                 ),
                 subscript: atomic_func!(|mapping, needle, vm| {
-                    PyMmap::mapping_downcast(mapping)._getitem(needle, vm)
+                    PyMmap::mapping_downcast(mapping).getitem_inner(needle, vm)
                 }),
                 ass_subscript: atomic_func!(|mapping, needle, value, vm| {
                     let zelf = PyMmap::mapping_downcast(mapping);
                     if let Some(value) = value {
-                        PyMmap::_setitem(zelf, needle, value, vm)
+                        PyMmap::setitem_inner(zelf, needle, value, vm)
                     } else {
                         Err(vm
                             .new_type_error("mmap object doesn't support item deletion".to_owned()))
@@ -897,6 +897,21 @@ mod mmap {
             Ok(())
         }
 
+        #[pymethod]
+        fn __getitem__(&self, needle: PyObjectRef, vm: &VirtualMachine) -> PyResult<PyObjectRef> {
+            self.getitem_inner(&needle, vm)
+        }
+
+        #[pymethod]
+        fn __setitem__(
+            zelf: &Py<Self>,
+            needle: PyObjectRef,
+            value: PyObjectRef,
+            vm: &VirtualMachine,
+        ) -> PyResult<()> {
+            Self::setitem_inner(zelf, &needle, value, vm)
+        }
+
         #[pymethod]
         fn __enter__(zelf: &Py<Self>, vm: &VirtualMachine) -> PyResult<PyRef<Self>> {
             let _m = zelf.check_valid(vm)?;
@@ -963,14 +978,14 @@ mod mmap {
             Ok(PyBytes::from(result_buf).into_ref(&vm.ctx).into())
         }
 
-        fn _getitem(&self, needle: &PyObject, vm: &VirtualMachine) -> PyResult<PyObjectRef> {
+        fn getitem_inner(&self, needle: &PyObject, vm: &VirtualMachine) -> PyResult<PyObjectRef> {
             match SequenceIndex::try_from_borrowed_object(vm, needle, "mmap")? {
                 SequenceIndex::Int(i) => self.getitem_by_index(i, vm),
                 SequenceIndex::Slice(slice) => self.getitem_by_slice(&slice, vm),
             }
         }
 
-        fn _setitem(
+        fn setitem_inner(
             zelf: &Py<Self>,
             needle: &PyObject,
             value: PyObjectRef,
