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

Impl PySequence contains, count and index

Browse Source
This commit is contained in:
Kangzhi Shi
2021-10-16 14:05:44 +02:00
committed by Jeong YunWon
parent e40894a274
commit bb7469ce9b
1 changed files with 57 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
vm/src/protocol/sequence.rs
View File

@@ -222,6 +222,63 @@ impl PySequence {
list.extend(self.obj.clone(), vm)?;
Ok(list.into())
}
pub fn contains(&self, target: &PyObjectRef, vm: &VirtualMachine) -> PyResult<bool> {
if let Some(f) = self.methods().contains {
return f(&self.obj, target, vm);
}
let iter = self.obj.clone().get_iter(vm)?;
let iter = iter.iter(vm)?;
for elem in iter {
let elem = elem?;
if vm.bool_eq(&elem, target)? {
return Ok(true);
}
}
Ok(false)
}
pub fn count(&self, target: &PyObjectRef, vm: &VirtualMachine) -> PyResult<usize> {
let mut n = 0;
let iter = self.obj.clone().get_iter(vm)?;
let iter = iter.iter(vm)?;
for elem in iter {
let elem = elem?;
if vm.bool_eq(&elem, target)? {
if n == isize::MAX as usize {
return Err(vm.new_overflow_error("index exceeds C integer size".to_string()));
}
n += 1;
}
}
Ok(n)
}
pub fn index(&self, target: &PyObjectRef, vm: &VirtualMachine) -> PyResult<usize> {
let mut index: isize = -1;
let iter = self.obj.clone().get_iter(vm)?;
let iter = iter.iter(vm)?;
for elem in iter {
if index == isize::MAX {
return Err(vm.new_overflow_error("index exceeds C integer size".to_string()));
}
index += 1;
let elem = elem?;
if vm.bool_eq(&elem, target)? {
return Ok(index as usize);
}
}
Err(vm.new_value_error("sequence.index(x): x not in sequence".to_string()))
}
}
pub(crate) fn try_add_for_concat(