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

switch RangeIndex to PyObjectRef

Browse Source 
As `getitem` method in other types, get second argument as PyObjectRef.
This commit is contained in:
snowapril
2021-09-22 18:46:30 +09:00
parent e83b68b810
commit b197a178ea
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
vm/src/builtins/range.rs
View File

@@ -334,8 +334,8 @@ impl PyRange {
}
#[pymethod(magic)]
fn getitem(&self, subscript: RangeIndex, vm: &VirtualMachine) -> PyResult {
match subscript {
fn getitem(&self, subscript: PyObjectRef, vm: &VirtualMachine) -> PyResult {
match RangeIndex::try_from_object(vm, subscript)? {
RangeIndex::Slice(slice) => {
let (mut substart, mut substop, mut substep) =
slice.inner_indices(&self.compute_length(), vm)?;
@@ -391,8 +391,7 @@ impl AsMapping for PyRange {
#[inline]
fn subscript(zelf: PyObjectRef, needle: PyObjectRef, vm: &VirtualMachine) -> PyResult {
Self::downcast_ref(&zelf, vm)
.map(|zelf| zelf.getitem(RangeIndex::try_from_object(vm, needle)?, vm))?
Self::downcast_ref(&zelf, vm).map(|zelf| zelf.getitem(needle, vm))?
}
#[inline]