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

Implement "__class_getitem__" (barely working)

Browse Source
This commit is contained in:
Padraic Fanning
2021-05-17 22:30:08 -04:00
parent 3921ea405c
commit 48de03506c
2 changed files with 25 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
vm/src/builtins/object.rs
View File

@@ -174,6 +174,11 @@ impl PyBaseObject {
#[pyclassmethod(magic)]
fn init_subclass(_cls: PyTypeRef) {}
#[pyclassmethod(magic)]
fn class_getitem(cls: PyTypeRef, _args: FuncArgs) -> PyObjectRef {
cls.into_object()
}
#[pymethod(magic)]
pub fn dir(obj: PyObjectRef, vm: &VirtualMachine) -> PyResult<PyList> {
let attributes: PyAttributes = obj.class().get_attributes();

28
vm/src/pyobject.rs
View File

@@ -685,14 +685,26 @@ where
T: IntoPyObject,
{
fn get_item(&self, key: T, vm: &VirtualMachine) -> PyResult {
vm.get_special_method(self.clone(), "__getitem__")?
.map_err(|obj| {
vm.new_type_error(format!(
"'{}' object is not subscriptable",
obj.class().name
))
})?
.invoke((key,), vm)
match vm.get_special_method(self.clone(), "__getitem__")? {
Ok(special_method) => special_method.invoke((key,), vm),
Err(obj) => {
if obj.isinstance(&vm.ctx.types.type_type) {
vm.get_special_method(obj, "__class_getitem__")?
.map_err(|obj2| {
vm.new_type_error(format!(
"'{}' object is not subscriptable",
obj2.class().name
))
})?
.invoke((key,), vm)
} else {
Err(vm.new_type_error(format!(
"'{}' object is not subscriptable",
obj.class().name
)))
}
}
}
}
fn set_item(&self, key: T, value: PyObjectRef, vm: &VirtualMachine) -> PyResult<()> {