Skip to content

Commit 625496c

Browse files
youknowoneyouknowone
committed
__class_getitem__
1 parent b6acfe0 commit 625496c

File tree

14 files changed

+119
-24
lines changed

14 files changed

+119
-24
lines changed

‎stdlib/src/array.rs‎

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
// spell-checker:ignore typecode tofile tolist fromfile
2-
32
use rustpython_vm::{PyRef, VirtualMachine, builtins::PyModule};
43

54
pub(crate) fn make_module(vm: &VirtualMachine) -> PyRef<PyModule> {
@@ -44,8 +43,8 @@ mod array {
4443
AsObject, Py, PyObject, PyObjectRef, PyPayload, PyRef, PyResult, VirtualMachine,
4544
atomic_func,
4645
builtins::{
47-
PositionIterInternal, PyByteArray, PyBytes, PyBytesRef, PyDictRef, PyFloat, PyInt,
48-
PyList, PyListRef, PyStr, PyStrRef, PyTupleRef, PyTypeRef,
46+
PositionIterInternal, PyByteArray, PyBytes, PyBytesRef, PyDictRef, PyFloat,
47+
PyGenericAlias, PyInt, PyList, PyListRef, PyStr, PyStrRef, PyTupleRef, PyTypeRef,
4948
},
5049
class_or_notimplemented,
5150
convert::{ToPyObject, ToPyResult, TryFromBorrowedObject, TryFromObject},
@@ -1195,6 +1194,15 @@ mod array {
11951194

11961195
false
11971196
}
1197+
1198+
#[pyclassmethod]
1199+
fn __class_getitem__(
1200+
cls: PyTypeRef,
1201+
args: PyObjectRef,
1202+
vm: &VirtualMachine,
1203+
) -> PyGenericAlias {
1204+
PyGenericAlias::from_args(cls, args, vm)
1205+
}
11981206
}
11991207

12001208
impl Comparable for PyArray {

‎stdlib/src/contextvars.rs‎

Lines changed: 15 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ thread_local! {
2424
mod _contextvars {
2525
use crate::vm::{
2626
AsObject, Py, PyObjectRef, PyPayload, PyRef, PyResult, VirtualMachine, atomic_func,
27-
builtins::{PyStrRef, PyTypeRef},
27+
builtins::{PyGenericAlias, PyStrRef, PyTypeRef},
2828
class::StaticType,
2929
common::hash::PyHash,
3030
function::{ArgCallable, FuncArgs, OptionalArg},
@@ -478,11 +478,11 @@ mod _contextvars {
478478

479479
#[pyclassmethod]
480480
fn __class_getitem__(
481-
_cls: PyTypeRef,
482-
_key: PyStrRef,
483-
_vm: &VirtualMachine,
484-
) -> PyResult<()> {
485-
unimplemented!("ContextVar.__class_getitem__() is currently under construction")
481+
cls: PyTypeRef,
482+
args: PyObjectRef,
483+
vm: &VirtualMachine,
484+
) -> PyGenericAlias {
485+
PyGenericAlias::from_args(cls, args, vm)
486486
}
487487
}
488488

@@ -571,6 +571,15 @@ mod _contextvars {
571571
None => ContextTokenMissing::static_type().to_owned().into(),
572572
}
573573
}
574+
575+
#[pyclassmethod]
576+
fn __class_getitem__(
577+
cls: PyTypeRef,
578+
args: PyObjectRef,
579+
vm: &VirtualMachine,
580+
) -> PyGenericAlias {
581+
PyGenericAlias::from_args(cls, args, vm)
582+
}
574583
}
575584

576585
impl Constructor for ContextToken {

‎vm/src/builtins/bytearray.rs‎

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
//! Implementation of the python bytearray object.
22
use super::{
3-
PositionIterInternal, PyBytes, PyBytesRef, PyDictRef, PyIntRef, PyStrRef, PyTuple, PyTupleRef,
4-
PyType, PyTypeRef,
3+
PositionIterInternal, PyBytes, PyBytesRef, PyDictRef, PyGenericAlias, PyIntRef, PyStrRef,
4+
PyTuple, PyTupleRef, PyType, PyTypeRef,
55
};
66
use crate::{
77
AsObject, Context, Py, PyObject, PyObjectRef, PyPayload, PyRef, PyResult, TryFromObject,
@@ -571,6 +571,11 @@ impl PyByteArray {
571571
fn reverse(&self) {
572572
self.borrow_buf_mut().reverse();
573573
}
574+
575+
#[pyclassmethod]
576+
fn __class_getitem__(cls: PyTypeRef, args: PyObjectRef, vm: &VirtualMachine) -> PyGenericAlias {
577+
PyGenericAlias::from_args(cls, args, vm)
578+
}
574579
}
575580

576581
#[pyclass]

‎vm/src/builtins/bytes.rs‎

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
use super::{
2-
PositionIterInternal, PyDictRef, PyIntRef, PyStrRef, PyTuple, PyTupleRef, PyType, PyTypeRef,
2+
PositionIterInternal, PyDictRef, PyGenericAlias, PyIntRef, PyStrRef, PyTuple, PyTupleRef,
3+
PyType, PyTypeRef,
34
};
45
use crate::{
56
AsObject, Context, Py, PyObject, PyObjectRef, PyPayload, PyRef, PyResult,
@@ -481,6 +482,11 @@ impl PyBytes {
481482
let param: Vec<PyObjectRef> = self.elements().map(|x| x.to_pyobject(vm)).collect();
482483
PyTuple::new_ref(param, &vm.ctx)
483484
}
485+
486+
#[pyclassmethod]
487+
fn __class_getitem__(cls: PyTypeRef, args: PyObjectRef, vm: &VirtualMachine) -> PyGenericAlias {
488+
PyGenericAlias::from_args(cls, args, vm)
489+
}
484490
}
485491

486492
#[pyclass]

‎vm/src/builtins/classmethod.rs‎

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
use super::{PyBoundMethod, PyStr, PyType, PyTypeRef};
1+
use super::{PyBoundMethod, PyGenericAlias, PyStr, PyType, PyTypeRef};
22
use crate::{
33
AsObject, Context, Py, PyObjectRef, PyPayload, PyRef, PyResult, VirtualMachine,
44
class::PyClassImpl,
@@ -170,6 +170,11 @@ impl PyClassMethod {
170170
.set_attr("__isabstractmethod__", value, vm)?;
171171
Ok(())
172172
}
173+
174+
#[pyclassmethod]
175+
fn __class_getitem__(cls: PyTypeRef, args: PyObjectRef, vm: &VirtualMachine) -> PyGenericAlias {
176+
PyGenericAlias::from_args(cls, args, vm)
177+
}
173178
}
174179

175180
impl Representable for PyClassMethod {

‎vm/src/builtins/coroutine.rs‎

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
use super::{PyCode, PyStrRef, PyType};
1+
use super::{PyCode, PyGenericAlias, PyStrRef, PyType, PyTypeRef};
22
use crate::{
33
AsObject, Context, Py, PyObjectRef, PyPayload, PyRef, PyResult, VirtualMachine,
44
class::PyClassImpl,
@@ -72,6 +72,11 @@ impl PyCoroutine {
7272
fn cr_origin(&self, _vm: &VirtualMachine) -> Option<(PyStrRef, usize, PyStrRef)> {
7373
None
7474
}
75+
76+
#[pyclassmethod]
77+
fn __class_getitem__(cls: PyTypeRef, args: PyObjectRef, vm: &VirtualMachine) -> PyGenericAlias {
78+
PyGenericAlias::from_args(cls, args, vm)
79+
}
7580
}
7681

7782
#[pyclass]

‎vm/src/builtins/generator.rs‎

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
* The mythical generator.
33
*/
44

5-
use super::{PyCode, PyStrRef, PyType};
5+
use super::{PyCode, PyGenericAlias, PyStrRef, PyType, PyTypeRef};
66
use crate::{
77
AsObject, Context, Py, PyObjectRef, PyPayload, PyRef, PyResult, VirtualMachine,
88
class::PyClassImpl,
@@ -67,6 +67,11 @@ impl PyGenerator {
6767
fn gi_yieldfrom(&self, _vm: &VirtualMachine) -> Option<PyObjectRef> {
6868
self.inner.frame().yield_from_target()
6969
}
70+
71+
#[pyclassmethod]
72+
fn __class_getitem__(cls: PyTypeRef, args: PyObjectRef, vm: &VirtualMachine) -> PyGenericAlias {
73+
PyGenericAlias::from_args(cls, args, vm)
74+
}
7075
}
7176

7277
#[pyclass]

‎vm/src/builtins/memory.rs‎

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
use super::{
2-
PositionIterInternal, PyBytes, PyBytesRef, PyInt, PyListRef, PySlice, PyStr, PyStrRef, PyTuple,
3-
PyTupleRef, PyType, PyTypeRef,
2+
PositionIterInternal, PyBytes, PyBytesRef, PyGenericAlias, PyInt, PyListRef, PySlice, PyStr,
3+
PyStrRef, PyTuple, PyTupleRef, PyType, PyTypeRef,
44
};
55
use crate::{
66
AsObject, Context, Py, PyObject, PyObjectRef, PyPayload, PyRef, PyResult,
@@ -550,6 +550,11 @@ impl Py<PyMemoryView> {
550550
Representable
551551
))]
552552
impl PyMemoryView {
553+
#[pyclassmethod]
554+
fn __class_getitem__(cls: PyTypeRef, args: PyObjectRef, vm: &VirtualMachine) -> PyGenericAlias {
555+
PyGenericAlias::from_args(cls, args, vm)
556+
}
557+
553558
#[pymethod]
554559
pub fn release(&self) {
555560
if self.released.compare_exchange(false, true).is_ok() {

‎vm/src/builtins/range.rs‎

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
use super::{
2-
PyInt, PyIntRef, PySlice, PyTupleRef, PyType, PyTypeRef, builtins_iter, tuple::tuple_hash,
2+
PyGenericAlias, PyInt, PyIntRef, PySlice, PyTupleRef, PyType, PyTypeRef, builtins_iter,
3+
tuple::tuple_hash,
34
};
45
use crate::{
56
AsObject, Context, Py, PyObject, PyObjectRef, PyPayload, PyRef, PyResult, TryFromObject,
@@ -183,6 +184,11 @@ pub fn init(context: &Context) {
183184
Representable
184185
))]
185186
impl PyRange {
187+
#[pyclassmethod]
188+
fn __class_getitem__(cls: PyTypeRef, args: PyObjectRef, vm: &VirtualMachine) -> PyGenericAlias {
189+
PyGenericAlias::from_args(cls, args, vm)
190+
}
191+
186192
fn new(cls: PyTypeRef, stop: ArgIndex, vm: &VirtualMachine) -> PyResult<PyRef<Self>> {
187193
PyRange {
188194
start: vm.ctx.new_pyref(0),

‎vm/src/builtins/slice.rs‎

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
// sliceobject.{h,c} in CPython
22
// spell-checker:ignore sliceobject
3-
use super::{PyStrRef, PyTupleRef, PyType, PyTypeRef};
3+
use super::{PyGenericAlias, PyStrRef, PyTupleRef, PyType, PyTypeRef};
44
use crate::{
55
AsObject, Context, Py, PyObject, PyObjectRef, PyPayload, PyRef, PyResult, VirtualMachine,
66
class::PyClassImpl,
@@ -30,6 +30,11 @@ impl PyPayload for PySlice {
3030

3131
#[pyclass(with(Comparable, Representable, Hashable))]
3232
impl PySlice {
33+
#[pyclassmethod]
34+
fn __class_getitem__(cls: PyTypeRef, args: PyObjectRef, vm: &VirtualMachine) -> PyGenericAlias {
35+
PyGenericAlias::from_args(cls, args, vm)
36+
}
37+
3338
#[pygetset]
3439
fn start(&self, vm: &VirtualMachine) -> PyObjectRef {
3540
self.start.clone().to_pyobject(vm)

0 commit comments

Comments
 (0)