mschoi/RustPython
1
0
Fork 0
forked from Rust-related/RustPython
Code Pull Requests Actions Activity

clean up enumerate

Browse Source
This commit is contained in:
Jeong YunWon
2023-03-19 02:12:10 +09:00
parent 29073cf711
commit 29055ab4a6
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
vm/src/builtins/enumerate.rs
View File

@@ -8,7 +8,7 @@ use crate::{
function::OptionalArg,
protocol::{PyIter, PyIterReturn},
types::{Constructor, IterNext, IterNextIterable},
AsObject, Context, Py, PyObjectRef, PyPayload, PyRef, PyResult, VirtualMachine,
AsObject, Context, Py, PyObjectRef, PyPayload, PyResult, VirtualMachine,
};
use num_bigint::BigInt;
use num_traits::Zero;
@@ -51,17 +51,21 @@ impl Constructor for PyEnumerate {
}
}
#[pyclass(with(IterNext, Constructor), flags(BASETYPE))]
#[pyclass(with(Py, IterNext, Constructor), flags(BASETYPE))]
impl PyEnumerate {
#[pyclassmethod(magic)]
fn class_getitem(cls: PyTypeRef, args: PyObjectRef, vm: &VirtualMachine) -> PyGenericAlias {
PyGenericAlias::new(cls, args, vm)
}
}
#[pyclass]
impl Py<PyEnumerate> {
#[pymethod(magic)]
fn reduce(zelf: PyRef<Self>) -> (PyTypeRef, (PyIter, BigInt)) {
fn reduce(&self) -> (PyTypeRef, (PyIter, BigInt)) {
(
zelf.class().to_owned(),
(zelf.iterator.clone(), zelf.counter.read().clone()),
self.class().to_owned(),
(self.iterator.clone(), self.counter.read().clone()),
)
}
}