Function subscript_generic

crates/vm/src/builtins/genericalias.rs:727–741

Creates a GenericAlias from type parameters, equivalent to _Py_subscript_generic. This is used for PEP 695 classes to create Generic[T] from type parameters. _Py_subscript_generic

(type_params: PyObjectRef, vm: &VirtualMachine)

/// This is used for PEP 695 classes to create Generic[T] from type parameters.
// _Py_subscript_generic
pub fn subscript_generic(type_params: PyObjectRef, vm: &VirtualMachine) -> PyResult {
    let typing_module = vm.import("typing", 0)?;
    let generic_type = typing_module.get_attr("Generic", vm)?;
    let generic_alias_class = typing_module.get_attr("_GenericAlias", vm)?;

    let params = if let Ok(tuple) = type_params.try_to_ref::<PyTuple>(vm) {
        tuple.to_owned()
    } else {
        PyTuple::new_ref(vec![type_params], &vm.ctx)
    };

    let args = crate::stdlib::_typing::unpack_typevartuples(&params, vm)?;

    generic_alias_class.call((generic_type, args.to_pyobject(vm)), vm)
}

pub fn init(context: &'static Context) {
    PyGenericAlias::extend_class(context, context.types.generic_alias_type);