Rust-related/RustPython
1
0
Fork 1
mirror of https://github.com/RustPython/RustPython.git synced 2026-06-09 22:49:57 +09:00
Code Issues Packages Projects Releases Wiki Activity

apply mapping protocol in itemprotocol

Browse Source
This commit is contained in:
snowapril
2021-09-21 00:14:49 +09:00
parent fc761b4e5b
commit e83b68b810
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
vm/src/pyobject.rs
View File

@@ -10,6 +10,7 @@ use crate::{
builtinfunc::PyNativeFuncDef,
bytearray, bytes,
code::{self, PyCode},
dict::PyMapping,
getset::{IntoPyGetterFunc, IntoPySetterFunc, PyGetSet},
namespace::PyNamespace,
object, pystr,
@@ -639,6 +640,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) => {
@@ -656,6 +663,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!(