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 or make public get_elements for sequences/dicts

Browse Source
This commit is contained in:
Daniel Watkins
2018-08-30 18:51:54 -04:00
parent e0b71b8d93
commit 934612314b
3 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
vm/src/objdict.rs
View File

@@ -26,7 +26,7 @@ pub fn new(dict_type: PyObjectRef) -> PyObjectRef {
)
}
fn get_elements(obj: &PyObjectRef) -> HashMap<String, PyObjectRef> {
pub fn get_elements(obj: &PyObjectRef) -> HashMap<String, PyObjectRef> {
if let PyObjectKind::Dict { elements } = &obj.borrow().kind {
elements.clone()
} else {

2
vm/src/objlist.rs
View File

@@ -26,7 +26,7 @@ pub fn set_item(
}
}
fn get_elements(obj: PyObjectRef) -> Vec<PyObjectRef> {
pub fn get_elements(obj: PyObjectRef) -> Vec<PyObjectRef> {
if let PyObjectKind::List { elements } = &obj.borrow().kind {
elements.to_vec()
} else {

8
vm/src/objsequence.rs
View File

@@ -98,3 +98,11 @@ pub fn get_item(
))),
}
}
pub fn get_elements(obj: PyObjectRef) -> Vec<PyObjectRef> {
if let PyObjectKind::Tuple { elements } = &obj.borrow().kind {
elements.to_vec()
} else {
panic!("Cannot extract list elements from non-list");
}
}