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

PyDeque::extend to use _extend always

Browse Source
This commit is contained in:
Jeong YunWon
2021-08-12 00:01:22 +09:00
parent 9c5dace49b
commit cd548b8fc3
1 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
vm/src/stdlib/collections.rs
View File

@@ -137,17 +137,13 @@ mod _collections {
fn extend(zelf: PyRef<Self>, iter: PyObjectRef, vm: &VirtualMachine) -> PyResult<()> {
// TODO: use length_hint here and for extendleft
if zelf.is(&iter) {
let iter =
PyIterable::try_from_object(vm, PyDeque::iter(zelf.copy().into_ref(vm), vm)?)?;
for elem in iter.iter(vm)? {
zelf.append(elem?)
}
Ok(())
let copied: Vec<PyObjectRef> = vm.extract_elements(zelf.as_object())?;
zelf._extend(copied.into_iter().map(Ok))?;
} else {
let iter = PyIterable::try_from_object(vm, iter)?;
zelf._extend(iter.iter(vm)?)
zelf._extend(iter.iter(vm)?)?;
}
Ok(())
}
#[pymethod]