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

Add arrayiter __reduce__

Browse Source 
Co-authored-by: Yaminyam <siontama@gmail.com>
This commit is contained in:
hyezzang
2022-07-11 19:19:18 +09:00
parent bed4877f8d
commit 0f24b18312
1 changed files with 16 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
stdlib/src/array.rs
View File

@@ -6,15 +6,15 @@ mod array {
common::{
atomic::{self, AtomicUsize},
lock::{
PyMappedRwLockReadGuard, PyMappedRwLockWriteGuard, PyRwLock, PyRwLockReadGuard,
PyRwLockWriteGuard,
PyMappedRwLockReadGuard, PyMappedRwLockWriteGuard, PyMutex, PyRwLock,
PyRwLockReadGuard, PyRwLockWriteGuard,
},
str::wchar_t,
},
vm::{
builtins::{
PyByteArray, PyBytes, PyBytesRef, PyDictRef, PyFloat, PyInt, PyIntRef, PyList,
PyListRef, PyStr, PyStrRef, PyTupleRef, PyTypeRef,
PositionIterInternal, PyByteArray, PyBytes, PyBytesRef, PyDictRef, PyFloat, PyInt,
PyIntRef, PyList, PyListRef, PyStr, PyStrRef, PyTupleRef, PyTypeRef,
},
class_or_notimplemented,
convert::{ToPyObject, ToPyResult, TryFromObject},
@@ -1260,7 +1260,8 @@ mod array {
fn iter(zelf: PyRef<Self>, vm: &VirtualMachine) -> PyResult {
Ok(PyArrayIterator {
position: AtomicUsize::new(0),
array: zelf,
array: zelf.clone(),
internal: PyMutex::new(PositionIterInternal::new(zelf.clone(), 0)),
}
.into_pyobject(vm))
}
@@ -1281,10 +1282,19 @@ mod array {
pub struct PyArrayIterator {
position: AtomicUsize,
array: PyArrayRef,
internal: PyMutex<PositionIterInternal<PyArrayRef>>,
}
#[pyimpl(with(IterNext))]
impl PyArrayIterator {}
impl PyArrayIterator {
#[pymethod(magic)]
fn reduce(&self, vm: &VirtualMachine) -> PyTupleRef {
self.internal
.lock()
.builtins_iter_reduce(|x| x.clone().into(), vm)
}
}
impl IterNextIterable for PyArrayIterator {}
impl IterNext for PyArrayIterator {