mirror of https://github.com/python/cpython
1989 lines
57 KiB
C
1989 lines
57 KiB
C
// TypeVar, TypeVarTuple, and ParamSpec
|
|
#include "Python.h"
|
|
#include "pycore_object.h" // _PyObject_GC_TRACK/UNTRACK
|
|
#include "pycore_typevarobject.h"
|
|
#include "pycore_unionobject.h" // _Py_union_type_or
|
|
|
|
|
|
/*[clinic input]
|
|
class typevar "typevarobject *" "&_PyTypeVar_Type"
|
|
class paramspec "paramspecobject *" "&_PyParamSpec_Type"
|
|
class paramspecargs "paramspecattrobject *" "&_PyParamSpecArgs_Type"
|
|
class paramspeckwargs "paramspecattrobject *" "&_PyParamSpecKwargs_Type"
|
|
class typevartuple "typevartupleobject *" "&_PyTypeVarTuple_Type"
|
|
class typealias "typealiasobject *" "&_PyTypeAlias_Type"
|
|
class Generic "PyObject *" "&PyGeneric_Type"
|
|
[clinic start generated code]*/
|
|
/*[clinic end generated code: output=da39a3ee5e6b4b0d input=aa86741931a0f55c]*/
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
PyObject *name;
|
|
PyObject *bound;
|
|
PyObject *evaluate_bound;
|
|
PyObject *constraints;
|
|
PyObject *evaluate_constraints;
|
|
PyObject *default_value;
|
|
PyObject *evaluate_default;
|
|
bool covariant;
|
|
bool contravariant;
|
|
bool infer_variance;
|
|
} typevarobject;
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
PyObject *name;
|
|
PyObject *default_value;
|
|
PyObject *evaluate_default;
|
|
} typevartupleobject;
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
PyObject *name;
|
|
PyObject *bound;
|
|
PyObject *default_value;
|
|
PyObject *evaluate_default;
|
|
bool covariant;
|
|
bool contravariant;
|
|
bool infer_variance;
|
|
} paramspecobject;
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
PyObject *name;
|
|
PyObject *type_params;
|
|
PyObject *compute_value;
|
|
PyObject *value;
|
|
PyObject *module;
|
|
} typealiasobject;
|
|
|
|
#include "clinic/typevarobject.c.h"
|
|
|
|
/* NoDefault is a marker object to indicate that a parameter has no default. */
|
|
|
|
static PyObject *
|
|
NoDefault_repr(PyObject *op)
|
|
{
|
|
return PyUnicode_FromString("typing.NoDefault");
|
|
}
|
|
|
|
static PyObject *
|
|
NoDefault_reduce(PyObject *op, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
return PyUnicode_FromString("NoDefault");
|
|
}
|
|
|
|
static PyMethodDef notimplemented_methods[] = {
|
|
{"__reduce__", NoDefault_reduce, METH_NOARGS, NULL},
|
|
{NULL, NULL}
|
|
};
|
|
|
|
static PyObject *
|
|
nodefault_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
|
|
{
|
|
if (PyTuple_GET_SIZE(args) || (kwargs && PyDict_GET_SIZE(kwargs))) {
|
|
PyErr_SetString(PyExc_TypeError, "NoDefaultType takes no arguments");
|
|
return NULL;
|
|
}
|
|
return &_Py_NoDefaultStruct;
|
|
}
|
|
|
|
static void
|
|
nodefault_dealloc(PyObject *nodefault)
|
|
{
|
|
/* This should never get called, but we also don't want to SEGV if
|
|
* we accidentally decref NoDefault out of existence. Instead,
|
|
* since NoDefault is an immortal object, re-set the reference count.
|
|
*/
|
|
_Py_SetImmortal(nodefault);
|
|
}
|
|
|
|
PyDoc_STRVAR(notimplemented_doc,
|
|
"NoDefaultType()\n"
|
|
"--\n\n"
|
|
"The type of the NoDefault singleton.");
|
|
|
|
PyTypeObject _PyNoDefault_Type = {
|
|
PyVarObject_HEAD_INIT(&PyType_Type, 0)
|
|
"NoDefaultType",
|
|
.tp_dealloc = nodefault_dealloc,
|
|
.tp_repr = NoDefault_repr,
|
|
.tp_flags = Py_TPFLAGS_DEFAULT,
|
|
.tp_doc = notimplemented_doc,
|
|
.tp_methods = notimplemented_methods,
|
|
.tp_new = nodefault_new,
|
|
};
|
|
|
|
PyObject _Py_NoDefaultStruct = _PyObject_HEAD_INIT(&_PyNoDefault_Type);
|
|
|
|
|
|
static PyObject *
|
|
call_typing_func_object(const char *name, PyObject **args, size_t nargs)
|
|
{
|
|
PyObject *typing = PyImport_ImportModule("typing");
|
|
if (typing == NULL) {
|
|
return NULL;
|
|
}
|
|
PyObject *func = PyObject_GetAttrString(typing, name);
|
|
if (func == NULL) {
|
|
Py_DECREF(typing);
|
|
return NULL;
|
|
}
|
|
PyObject *result = PyObject_Vectorcall(func, args, nargs, NULL);
|
|
Py_DECREF(func);
|
|
Py_DECREF(typing);
|
|
return result;
|
|
}
|
|
|
|
static PyObject *
|
|
type_check(PyObject *arg, const char *msg)
|
|
{
|
|
// Calling typing.py here leads to bootstrapping problems
|
|
if (Py_IsNone(arg)) {
|
|
return Py_NewRef(Py_TYPE(arg));
|
|
}
|
|
PyObject *message_str = PyUnicode_FromString(msg);
|
|
if (message_str == NULL) {
|
|
return NULL;
|
|
}
|
|
PyObject *args[2] = {arg, message_str};
|
|
PyObject *result = call_typing_func_object("_type_check", args, 2);
|
|
Py_DECREF(message_str);
|
|
return result;
|
|
}
|
|
|
|
/*
|
|
* Return a typing.Union. This is used as the nb_or (|) operator for
|
|
* TypeVar and ParamSpec. We use this rather than _Py_union_type_or
|
|
* (which would produce a types.Union) because historically TypeVar
|
|
* supported unions with string forward references, and we want to
|
|
* preserve that behavior. _Py_union_type_or only allows a small set
|
|
* of types.
|
|
*/
|
|
static PyObject *
|
|
make_union(PyObject *self, PyObject *other)
|
|
{
|
|
PyObject *args[2] = {self, other};
|
|
PyObject *result = call_typing_func_object("_make_union", args, 2);
|
|
return result;
|
|
}
|
|
|
|
static PyObject *
|
|
caller(void)
|
|
{
|
|
_PyInterpreterFrame *f = _PyThreadState_GET()->current_frame;
|
|
if (f == NULL) {
|
|
Py_RETURN_NONE;
|
|
}
|
|
if (f == NULL || f->f_funcobj == NULL) {
|
|
Py_RETURN_NONE;
|
|
}
|
|
PyObject *r = PyFunction_GetModule(f->f_funcobj);
|
|
if (!r) {
|
|
PyErr_Clear();
|
|
Py_RETURN_NONE;
|
|
}
|
|
return Py_NewRef(r);
|
|
}
|
|
|
|
static PyObject *
|
|
typevartuple_unpack(PyObject *tvt)
|
|
{
|
|
PyObject *typing = PyImport_ImportModule("typing");
|
|
if (typing == NULL) {
|
|
return NULL;
|
|
}
|
|
PyObject *unpack = PyObject_GetAttrString(typing, "Unpack");
|
|
if (unpack == NULL) {
|
|
Py_DECREF(typing);
|
|
return NULL;
|
|
}
|
|
PyObject *unpacked = PyObject_GetItem(unpack, tvt);
|
|
Py_DECREF(typing);
|
|
Py_DECREF(unpack);
|
|
return unpacked;
|
|
}
|
|
|
|
static int
|
|
contains_typevartuple(PyTupleObject *params)
|
|
{
|
|
Py_ssize_t n = PyTuple_GET_SIZE(params);
|
|
PyTypeObject *tp = _PyInterpreterState_GET()->cached_objects.typevartuple_type;
|
|
for (Py_ssize_t i = 0; i < n; i++) {
|
|
PyObject *param = PyTuple_GET_ITEM(params, i);
|
|
if (Py_IS_TYPE(param, tp)) {
|
|
return 1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *
|
|
unpack_typevartuples(PyObject *params)
|
|
{
|
|
assert(PyTuple_Check(params));
|
|
// TypeVarTuple must be unpacked when passed to Generic, so we do that here.
|
|
if (contains_typevartuple((PyTupleObject *)params)) {
|
|
Py_ssize_t n = PyTuple_GET_SIZE(params);
|
|
PyObject *new_params = PyTuple_New(n);
|
|
if (new_params == NULL) {
|
|
return NULL;
|
|
}
|
|
PyTypeObject *tp = _PyInterpreterState_GET()->cached_objects.typevartuple_type;
|
|
for (Py_ssize_t i = 0; i < n; i++) {
|
|
PyObject *param = PyTuple_GET_ITEM(params, i);
|
|
if (Py_IS_TYPE(param, tp)) {
|
|
PyObject *unpacked = typevartuple_unpack(param);
|
|
if (unpacked == NULL) {
|
|
Py_DECREF(new_params);
|
|
return NULL;
|
|
}
|
|
PyTuple_SET_ITEM(new_params, i, unpacked);
|
|
}
|
|
else {
|
|
PyTuple_SET_ITEM(new_params, i, Py_NewRef(param));
|
|
}
|
|
}
|
|
return new_params;
|
|
}
|
|
else {
|
|
return Py_NewRef(params);
|
|
}
|
|
}
|
|
|
|
static void
|
|
typevar_dealloc(PyObject *self)
|
|
{
|
|
PyTypeObject *tp = Py_TYPE(self);
|
|
typevarobject *tv = (typevarobject *)self;
|
|
|
|
_PyObject_GC_UNTRACK(self);
|
|
|
|
Py_DECREF(tv->name);
|
|
Py_XDECREF(tv->bound);
|
|
Py_XDECREF(tv->evaluate_bound);
|
|
Py_XDECREF(tv->constraints);
|
|
Py_XDECREF(tv->evaluate_constraints);
|
|
Py_XDECREF(tv->default_value);
|
|
Py_XDECREF(tv->evaluate_default);
|
|
PyObject_ClearManagedDict(self);
|
|
PyObject_ClearWeakRefs(self);
|
|
|
|
Py_TYPE(self)->tp_free(self);
|
|
Py_DECREF(tp);
|
|
}
|
|
|
|
static int
|
|
typevar_traverse(PyObject *self, visitproc visit, void *arg)
|
|
{
|
|
Py_VISIT(Py_TYPE(self));
|
|
typevarobject *tv = (typevarobject *)self;
|
|
Py_VISIT(tv->bound);
|
|
Py_VISIT(tv->evaluate_bound);
|
|
Py_VISIT(tv->constraints);
|
|
Py_VISIT(tv->evaluate_constraints);
|
|
Py_VISIT(tv->default_value);
|
|
Py_VISIT(tv->evaluate_default);
|
|
PyObject_VisitManagedDict(self, visit, arg);
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
typevar_clear(typevarobject *self)
|
|
{
|
|
Py_CLEAR(self->bound);
|
|
Py_CLEAR(self->evaluate_bound);
|
|
Py_CLEAR(self->constraints);
|
|
Py_CLEAR(self->evaluate_constraints);
|
|
Py_CLEAR(self->default_value);
|
|
Py_CLEAR(self->evaluate_default);
|
|
PyObject_ClearManagedDict((PyObject *)self);
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *
|
|
typevar_repr(PyObject *self)
|
|
{
|
|
typevarobject *tv = (typevarobject *)self;
|
|
|
|
if (tv->infer_variance) {
|
|
return Py_NewRef(tv->name);
|
|
}
|
|
|
|
char variance = tv->covariant ? '+' : tv->contravariant ? '-' : '~';
|
|
return PyUnicode_FromFormat("%c%U", variance, tv->name);
|
|
}
|
|
|
|
static PyMemberDef typevar_members[] = {
|
|
{"__name__", _Py_T_OBJECT, offsetof(typevarobject, name), Py_READONLY},
|
|
{"__covariant__", Py_T_BOOL, offsetof(typevarobject, covariant), Py_READONLY},
|
|
{"__contravariant__", Py_T_BOOL, offsetof(typevarobject, contravariant), Py_READONLY},
|
|
{"__infer_variance__", Py_T_BOOL, offsetof(typevarobject, infer_variance), Py_READONLY},
|
|
{0}
|
|
};
|
|
|
|
static PyObject *
|
|
typevar_bound(typevarobject *self, void *Py_UNUSED(ignored))
|
|
{
|
|
if (self->bound != NULL) {
|
|
return Py_NewRef(self->bound);
|
|
}
|
|
if (self->evaluate_bound == NULL) {
|
|
Py_RETURN_NONE;
|
|
}
|
|
PyObject *bound = PyObject_CallNoArgs(self->evaluate_bound);
|
|
self->bound = Py_XNewRef(bound);
|
|
return bound;
|
|
}
|
|
|
|
static PyObject *
|
|
typevar_default(typevarobject *self, void *unused)
|
|
{
|
|
if (self->default_value != NULL) {
|
|
return Py_NewRef(self->default_value);
|
|
}
|
|
if (self->evaluate_default == NULL) {
|
|
return &_Py_NoDefaultStruct;
|
|
}
|
|
PyObject *default_value = PyObject_CallNoArgs(self->evaluate_default);
|
|
self->default_value = Py_XNewRef(default_value);
|
|
return default_value;
|
|
}
|
|
|
|
static PyObject *
|
|
typevar_constraints(typevarobject *self, void *Py_UNUSED(ignored))
|
|
{
|
|
if (self->constraints != NULL) {
|
|
return Py_NewRef(self->constraints);
|
|
}
|
|
if (self->evaluate_constraints == NULL) {
|
|
return PyTuple_New(0);
|
|
}
|
|
PyObject *constraints = PyObject_CallNoArgs(self->evaluate_constraints);
|
|
self->constraints = Py_XNewRef(constraints);
|
|
return constraints;
|
|
}
|
|
|
|
static PyGetSetDef typevar_getset[] = {
|
|
{"__bound__", (getter)typevar_bound, NULL, NULL, NULL},
|
|
{"__constraints__", (getter)typevar_constraints, NULL, NULL, NULL},
|
|
{"__default__", (getter)typevar_default, NULL, NULL, NULL},
|
|
{0}
|
|
};
|
|
|
|
static typevarobject *
|
|
typevar_alloc(PyObject *name, PyObject *bound, PyObject *evaluate_bound,
|
|
PyObject *constraints, PyObject *evaluate_constraints,
|
|
PyObject *default_value,
|
|
bool covariant, bool contravariant, bool infer_variance,
|
|
PyObject *module)
|
|
{
|
|
PyTypeObject *tp = _PyInterpreterState_GET()->cached_objects.typevar_type;
|
|
assert(tp != NULL);
|
|
typevarobject *tv = PyObject_GC_New(typevarobject, tp);
|
|
if (tv == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
tv->name = Py_NewRef(name);
|
|
|
|
tv->bound = Py_XNewRef(bound);
|
|
tv->evaluate_bound = Py_XNewRef(evaluate_bound);
|
|
tv->constraints = Py_XNewRef(constraints);
|
|
tv->evaluate_constraints = Py_XNewRef(evaluate_constraints);
|
|
tv->default_value = Py_XNewRef(default_value);
|
|
tv->evaluate_default = NULL;
|
|
|
|
tv->covariant = covariant;
|
|
tv->contravariant = contravariant;
|
|
tv->infer_variance = infer_variance;
|
|
_PyObject_GC_TRACK(tv);
|
|
|
|
if (module != NULL) {
|
|
if (PyObject_SetAttrString((PyObject *)tv, "__module__", module) < 0) {
|
|
Py_DECREF(tv);
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
return tv;
|
|
}
|
|
|
|
/*[clinic input]
|
|
@classmethod
|
|
typevar.__new__ as typevar_new
|
|
|
|
name: object(subclass_of="&PyUnicode_Type")
|
|
*constraints: object
|
|
bound: object = None
|
|
default as default_value: object(c_default="&_Py_NoDefaultStruct") = typing.NoDefault
|
|
covariant: bool = False
|
|
contravariant: bool = False
|
|
infer_variance: bool = False
|
|
|
|
Create a TypeVar.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
typevar_new_impl(PyTypeObject *type, PyObject *name, PyObject *constraints,
|
|
PyObject *bound, PyObject *default_value, int covariant,
|
|
int contravariant, int infer_variance)
|
|
/*[clinic end generated code: output=d2b248ff074eaab6 input=836f97f631d7293a]*/
|
|
{
|
|
if (covariant && contravariant) {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"Bivariant types are not supported.");
|
|
return NULL;
|
|
}
|
|
|
|
if (infer_variance && (covariant || contravariant)) {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"Variance cannot be specified with infer_variance.");
|
|
return NULL;
|
|
}
|
|
|
|
if (Py_IsNone(bound)) {
|
|
bound = NULL;
|
|
}
|
|
if (bound != NULL) {
|
|
bound = type_check(bound, "Bound must be a type.");
|
|
if (bound == NULL) {
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
assert(PyTuple_CheckExact(constraints));
|
|
Py_ssize_t n_constraints = PyTuple_GET_SIZE(constraints);
|
|
if (n_constraints == 1) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"A single constraint is not allowed");
|
|
Py_XDECREF(bound);
|
|
return NULL;
|
|
} else if (n_constraints == 0) {
|
|
constraints = NULL;
|
|
} else if (bound != NULL) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"Constraints cannot be combined with bound=...");
|
|
Py_XDECREF(bound);
|
|
return NULL;
|
|
}
|
|
PyObject *module = caller();
|
|
if (module == NULL) {
|
|
Py_XDECREF(bound);
|
|
return NULL;
|
|
}
|
|
|
|
PyObject *tv = (PyObject *)typevar_alloc(name, bound, NULL,
|
|
constraints, NULL,
|
|
default_value,
|
|
covariant, contravariant,
|
|
infer_variance, module);
|
|
Py_XDECREF(bound);
|
|
Py_XDECREF(module);
|
|
return tv;
|
|
}
|
|
|
|
/*[clinic input]
|
|
typevar.__typing_subst__ as typevar_typing_subst
|
|
|
|
arg: object
|
|
/
|
|
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
typevar_typing_subst(typevarobject *self, PyObject *arg)
|
|
/*[clinic end generated code: output=0773735e8ce18968 input=9e87b57f0fc59b92]*/
|
|
{
|
|
PyObject *args[2] = {(PyObject *)self, arg};
|
|
PyObject *result = call_typing_func_object("_typevar_subst", args, 2);
|
|
return result;
|
|
}
|
|
|
|
/*[clinic input]
|
|
typevar.__typing_prepare_subst__ as typevar_typing_prepare_subst
|
|
|
|
alias: object
|
|
args: object
|
|
/
|
|
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
typevar_typing_prepare_subst_impl(typevarobject *self, PyObject *alias,
|
|
PyObject *args)
|
|
/*[clinic end generated code: output=82c3f4691e0ded22 input=201a750415d14ffb]*/
|
|
{
|
|
PyObject *params = PyObject_GetAttrString(alias, "__parameters__");
|
|
if (params == NULL) {
|
|
return NULL;
|
|
}
|
|
Py_ssize_t i = PySequence_Index(params, (PyObject *)self);
|
|
if (i == -1) {
|
|
Py_DECREF(params);
|
|
return NULL;
|
|
}
|
|
Py_ssize_t args_len = PySequence_Length(args);
|
|
if (args_len == -1) {
|
|
Py_DECREF(params);
|
|
return NULL;
|
|
}
|
|
if (i < args_len) {
|
|
// We already have a value for our TypeVar
|
|
Py_DECREF(params);
|
|
return Py_NewRef(args);
|
|
}
|
|
else if (i == args_len) {
|
|
// If the TypeVar has a default, use it.
|
|
PyObject *dflt = typevar_default(self, NULL);
|
|
if (dflt == NULL) {
|
|
Py_DECREF(params);
|
|
return NULL;
|
|
}
|
|
if (dflt != &_Py_NoDefaultStruct) {
|
|
PyObject *new_args = PyTuple_Pack(1, dflt);
|
|
Py_DECREF(dflt);
|
|
if (new_args == NULL) {
|
|
Py_DECREF(params);
|
|
return NULL;
|
|
}
|
|
PyObject *result = PySequence_Concat(args, new_args);
|
|
Py_DECREF(params);
|
|
Py_DECREF(new_args);
|
|
return result;
|
|
}
|
|
}
|
|
Py_DECREF(params);
|
|
PyErr_Format(PyExc_TypeError,
|
|
"Too few arguments for %S; actual %d, expected at least %d",
|
|
alias, args_len, i + 1);
|
|
return NULL;
|
|
}
|
|
|
|
/*[clinic input]
|
|
typevar.__reduce__ as typevar_reduce
|
|
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
typevar_reduce_impl(typevarobject *self)
|
|
/*[clinic end generated code: output=02e5c55d7cf8a08f input=de76bc95f04fb9ff]*/
|
|
{
|
|
return Py_NewRef(self->name);
|
|
}
|
|
|
|
|
|
/*[clinic input]
|
|
typevar.has_default as typevar_has_default
|
|
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
typevar_has_default_impl(typevarobject *self)
|
|
/*[clinic end generated code: output=76bf0b8dc98b97dd input=31024aa030761cf6]*/
|
|
{
|
|
if (self->evaluate_default != NULL ||
|
|
(self->default_value != &_Py_NoDefaultStruct && self->default_value != NULL)) {
|
|
Py_RETURN_TRUE;
|
|
}
|
|
Py_RETURN_FALSE;
|
|
}
|
|
|
|
static PyObject *
|
|
typevar_mro_entries(PyObject *self, PyObject *args)
|
|
{
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"Cannot subclass an instance of TypeVar");
|
|
return NULL;
|
|
}
|
|
|
|
static PyMethodDef typevar_methods[] = {
|
|
TYPEVAR_TYPING_SUBST_METHODDEF
|
|
TYPEVAR_TYPING_PREPARE_SUBST_METHODDEF
|
|
TYPEVAR_REDUCE_METHODDEF
|
|
TYPEVAR_HAS_DEFAULT_METHODDEF
|
|
{"__mro_entries__", typevar_mro_entries, METH_O},
|
|
{0}
|
|
};
|
|
|
|
PyDoc_STRVAR(typevar_doc,
|
|
"Type variable.\n\
|
|
\n\
|
|
The preferred way to construct a type variable is via the dedicated\n\
|
|
syntax for generic functions, classes, and type aliases::\n\
|
|
\n\
|
|
class Sequence[T]: # T is a TypeVar\n\
|
|
...\n\
|
|
\n\
|
|
This syntax can also be used to create bound and constrained type\n\
|
|
variables::\n\
|
|
\n\
|
|
# S is a TypeVar bound to str\n\
|
|
class StrSequence[S: str]:\n\
|
|
...\n\
|
|
\n\
|
|
# A is a TypeVar constrained to str or bytes\n\
|
|
class StrOrBytesSequence[A: (str, bytes)]:\n\
|
|
...\n\
|
|
\n\
|
|
Type variables can also have defaults:\n\
|
|
\n\
|
|
class IntDefault[T = int]:\n\
|
|
...\n\
|
|
\n\
|
|
However, if desired, reusable type variables can also be constructed\n\
|
|
manually, like so::\n\
|
|
\n\
|
|
T = TypeVar('T') # Can be anything\n\
|
|
S = TypeVar('S', bound=str) # Can be any subtype of str\n\
|
|
A = TypeVar('A', str, bytes) # Must be exactly str or bytes\n\
|
|
D = TypeVar('D', default=int) # Defaults to int\n\
|
|
\n\
|
|
Type variables exist primarily for the benefit of static type\n\
|
|
checkers. They serve as the parameters for generic types as well\n\
|
|
as for generic function and type alias definitions.\n\
|
|
\n\
|
|
The variance of type variables is inferred by type checkers when they\n\
|
|
are created through the type parameter syntax and when\n\
|
|
``infer_variance=True`` is passed. Manually created type variables may\n\
|
|
be explicitly marked covariant or contravariant by passing\n\
|
|
``covariant=True`` or ``contravariant=True``. By default, manually\n\
|
|
created type variables are invariant. See PEP 484 and PEP 695 for more\n\
|
|
details.\n\
|
|
");
|
|
|
|
static PyType_Slot typevar_slots[] = {
|
|
{Py_tp_doc, (void *)typevar_doc},
|
|
{Py_tp_methods, typevar_methods},
|
|
{Py_nb_or, make_union},
|
|
{Py_tp_new, typevar_new},
|
|
{Py_tp_dealloc, typevar_dealloc},
|
|
{Py_tp_alloc, PyType_GenericAlloc},
|
|
{Py_tp_free, PyObject_GC_Del},
|
|
{Py_tp_traverse, typevar_traverse},
|
|
{Py_tp_clear, typevar_clear},
|
|
{Py_tp_repr, typevar_repr},
|
|
{Py_tp_members, typevar_members},
|
|
{Py_tp_getset, typevar_getset},
|
|
{0, NULL},
|
|
};
|
|
|
|
PyType_Spec typevar_spec = {
|
|
.name = "typing.TypeVar",
|
|
.basicsize = sizeof(typevarobject),
|
|
.flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC | Py_TPFLAGS_IMMUTABLETYPE
|
|
| Py_TPFLAGS_MANAGED_DICT | Py_TPFLAGS_MANAGED_WEAKREF,
|
|
.slots = typevar_slots,
|
|
};
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
PyObject *__origin__;
|
|
} paramspecattrobject;
|
|
|
|
static void
|
|
paramspecattr_dealloc(PyObject *self)
|
|
{
|
|
PyTypeObject *tp = Py_TYPE(self);
|
|
paramspecattrobject *psa = (paramspecattrobject *)self;
|
|
|
|
_PyObject_GC_UNTRACK(self);
|
|
|
|
Py_XDECREF(psa->__origin__);
|
|
|
|
Py_TYPE(self)->tp_free(self);
|
|
Py_DECREF(tp);
|
|
}
|
|
|
|
static int
|
|
paramspecattr_traverse(PyObject *self, visitproc visit, void *arg)
|
|
{
|
|
paramspecattrobject *psa = (paramspecattrobject *)self;
|
|
Py_VISIT(psa->__origin__);
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
paramspecattr_clear(paramspecattrobject *self)
|
|
{
|
|
Py_CLEAR(self->__origin__);
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *
|
|
paramspecattr_richcompare(PyObject *a, PyObject *b, int op)
|
|
{
|
|
if (!Py_IS_TYPE(a, Py_TYPE(b))) {
|
|
Py_RETURN_NOTIMPLEMENTED;
|
|
}
|
|
if (op != Py_EQ && op != Py_NE) {
|
|
Py_RETURN_NOTIMPLEMENTED;
|
|
}
|
|
return PyObject_RichCompare(
|
|
((paramspecattrobject *)a)->__origin__,
|
|
((paramspecattrobject *)b)->__origin__,
|
|
op
|
|
);
|
|
}
|
|
|
|
static PyMemberDef paramspecattr_members[] = {
|
|
{"__origin__", _Py_T_OBJECT, offsetof(paramspecattrobject, __origin__), Py_READONLY},
|
|
{0}
|
|
};
|
|
|
|
static paramspecattrobject *
|
|
paramspecattr_new(PyTypeObject *tp, PyObject *origin)
|
|
{
|
|
paramspecattrobject *psa = PyObject_GC_New(paramspecattrobject, tp);
|
|
if (psa == NULL) {
|
|
return NULL;
|
|
}
|
|
psa->__origin__ = Py_NewRef(origin);
|
|
_PyObject_GC_TRACK(psa);
|
|
return psa;
|
|
}
|
|
|
|
static PyObject *
|
|
paramspecargs_repr(PyObject *self)
|
|
{
|
|
paramspecattrobject *psa = (paramspecattrobject *)self;
|
|
|
|
PyTypeObject *tp = _PyInterpreterState_GET()->cached_objects.paramspec_type;
|
|
if (Py_IS_TYPE(psa->__origin__, tp)) {
|
|
return PyUnicode_FromFormat("%U.args",
|
|
((paramspecobject *)psa->__origin__)->name);
|
|
}
|
|
return PyUnicode_FromFormat("%R.args", psa->__origin__);
|
|
}
|
|
|
|
|
|
/*[clinic input]
|
|
@classmethod
|
|
paramspecargs.__new__ as paramspecargs_new
|
|
|
|
origin: object
|
|
|
|
Create a ParamSpecArgs object.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
paramspecargs_new_impl(PyTypeObject *type, PyObject *origin)
|
|
/*[clinic end generated code: output=9a1463dc8942fe4e input=3596a0bb6183c208]*/
|
|
{
|
|
return (PyObject *)paramspecattr_new(type, origin);
|
|
}
|
|
|
|
static PyObject *
|
|
paramspecargs_mro_entries(PyObject *self, PyObject *args)
|
|
{
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"Cannot subclass an instance of ParamSpecArgs");
|
|
return NULL;
|
|
}
|
|
|
|
static PyMethodDef paramspecargs_methods[] = {
|
|
{"__mro_entries__", paramspecargs_mro_entries, METH_O},
|
|
{0}
|
|
};
|
|
|
|
PyDoc_STRVAR(paramspecargs_doc,
|
|
"The args for a ParamSpec object.\n\
|
|
\n\
|
|
Given a ParamSpec object P, P.args is an instance of ParamSpecArgs.\n\
|
|
\n\
|
|
ParamSpecArgs objects have a reference back to their ParamSpec::\n\
|
|
\n\
|
|
>>> P = ParamSpec(\"P\")\n\
|
|
>>> P.args.__origin__ is P\n\
|
|
True\n\
|
|
\n\
|
|
This type is meant for runtime introspection and has no special meaning\n\
|
|
to static type checkers.\n\
|
|
");
|
|
|
|
static PyType_Slot paramspecargs_slots[] = {
|
|
{Py_tp_doc, (void *)paramspecargs_doc},
|
|
{Py_tp_methods, paramspecargs_methods},
|
|
{Py_tp_new, paramspecargs_new},
|
|
{Py_tp_dealloc, paramspecattr_dealloc},
|
|
{Py_tp_alloc, PyType_GenericAlloc},
|
|
{Py_tp_free, PyObject_GC_Del},
|
|
{Py_tp_traverse, paramspecattr_traverse},
|
|
{Py_tp_clear, (inquiry)paramspecattr_clear},
|
|
{Py_tp_repr, paramspecargs_repr},
|
|
{Py_tp_members, paramspecattr_members},
|
|
{Py_tp_richcompare, paramspecattr_richcompare},
|
|
{0, NULL},
|
|
};
|
|
|
|
PyType_Spec paramspecargs_spec = {
|
|
.name = "typing.ParamSpecArgs",
|
|
.basicsize = sizeof(paramspecattrobject),
|
|
.flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC | Py_TPFLAGS_IMMUTABLETYPE
|
|
| Py_TPFLAGS_MANAGED_WEAKREF,
|
|
.slots = paramspecargs_slots,
|
|
};
|
|
|
|
static PyObject *
|
|
paramspeckwargs_repr(PyObject *self)
|
|
{
|
|
paramspecattrobject *psk = (paramspecattrobject *)self;
|
|
|
|
PyTypeObject *tp = _PyInterpreterState_GET()->cached_objects.paramspec_type;
|
|
if (Py_IS_TYPE(psk->__origin__, tp)) {
|
|
return PyUnicode_FromFormat("%U.kwargs",
|
|
((paramspecobject *)psk->__origin__)->name);
|
|
}
|
|
return PyUnicode_FromFormat("%R.kwargs", psk->__origin__);
|
|
}
|
|
|
|
/*[clinic input]
|
|
@classmethod
|
|
paramspeckwargs.__new__ as paramspeckwargs_new
|
|
|
|
origin: object
|
|
|
|
Create a ParamSpecKwargs object.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
paramspeckwargs_new_impl(PyTypeObject *type, PyObject *origin)
|
|
/*[clinic end generated code: output=277b11967ebaf4ab input=981bca9b0cf9e40a]*/
|
|
{
|
|
return (PyObject *)paramspecattr_new(type, origin);
|
|
}
|
|
|
|
static PyObject *
|
|
paramspeckwargs_mro_entries(PyObject *self, PyObject *args)
|
|
{
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"Cannot subclass an instance of ParamSpecKwargs");
|
|
return NULL;
|
|
}
|
|
|
|
static PyMethodDef paramspeckwargs_methods[] = {
|
|
{"__mro_entries__", paramspeckwargs_mro_entries, METH_O},
|
|
{0}
|
|
};
|
|
|
|
PyDoc_STRVAR(paramspeckwargs_doc,
|
|
"The kwargs for a ParamSpec object.\n\
|
|
\n\
|
|
Given a ParamSpec object P, P.kwargs is an instance of ParamSpecKwargs.\n\
|
|
\n\
|
|
ParamSpecKwargs objects have a reference back to their ParamSpec::\n\
|
|
\n\
|
|
>>> P = ParamSpec(\"P\")\n\
|
|
>>> P.kwargs.__origin__ is P\n\
|
|
True\n\
|
|
\n\
|
|
This type is meant for runtime introspection and has no special meaning\n\
|
|
to static type checkers.\n\
|
|
");
|
|
|
|
static PyType_Slot paramspeckwargs_slots[] = {
|
|
{Py_tp_doc, (void *)paramspeckwargs_doc},
|
|
{Py_tp_methods, paramspeckwargs_methods},
|
|
{Py_tp_new, paramspeckwargs_new},
|
|
{Py_tp_dealloc, paramspecattr_dealloc},
|
|
{Py_tp_alloc, PyType_GenericAlloc},
|
|
{Py_tp_free, PyObject_GC_Del},
|
|
{Py_tp_traverse, paramspecattr_traverse},
|
|
{Py_tp_clear, (inquiry)paramspecattr_clear},
|
|
{Py_tp_repr, paramspeckwargs_repr},
|
|
{Py_tp_members, paramspecattr_members},
|
|
{Py_tp_richcompare, paramspecattr_richcompare},
|
|
{0, NULL},
|
|
};
|
|
|
|
PyType_Spec paramspeckwargs_spec = {
|
|
.name = "typing.ParamSpecKwargs",
|
|
.basicsize = sizeof(paramspecattrobject),
|
|
.flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC | Py_TPFLAGS_IMMUTABLETYPE
|
|
| Py_TPFLAGS_MANAGED_WEAKREF,
|
|
.slots = paramspeckwargs_slots,
|
|
};
|
|
|
|
static void
|
|
paramspec_dealloc(PyObject *self)
|
|
{
|
|
PyTypeObject *tp = Py_TYPE(self);
|
|
paramspecobject *ps = (paramspecobject *)self;
|
|
|
|
_PyObject_GC_UNTRACK(self);
|
|
|
|
Py_DECREF(ps->name);
|
|
Py_XDECREF(ps->bound);
|
|
Py_XDECREF(ps->default_value);
|
|
Py_XDECREF(ps->evaluate_default);
|
|
PyObject_ClearManagedDict(self);
|
|
PyObject_ClearWeakRefs(self);
|
|
|
|
Py_TYPE(self)->tp_free(self);
|
|
Py_DECREF(tp);
|
|
}
|
|
|
|
static int
|
|
paramspec_traverse(PyObject *self, visitproc visit, void *arg)
|
|
{
|
|
Py_VISIT(Py_TYPE(self));
|
|
paramspecobject *ps = (paramspecobject *)self;
|
|
Py_VISIT(ps->bound);
|
|
Py_VISIT(ps->default_value);
|
|
Py_VISIT(ps->evaluate_default);
|
|
PyObject_VisitManagedDict(self, visit, arg);
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
paramspec_clear(paramspecobject *self)
|
|
{
|
|
Py_CLEAR(self->bound);
|
|
Py_CLEAR(self->default_value);
|
|
Py_CLEAR(self->evaluate_default);
|
|
PyObject_ClearManagedDict((PyObject *)self);
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *
|
|
paramspec_repr(PyObject *self)
|
|
{
|
|
paramspecobject *ps = (paramspecobject *)self;
|
|
|
|
if (ps->infer_variance) {
|
|
return Py_NewRef(ps->name);
|
|
}
|
|
|
|
char variance = ps->covariant ? '+' : ps->contravariant ? '-' : '~';
|
|
return PyUnicode_FromFormat("%c%U", variance, ps->name);
|
|
}
|
|
|
|
static PyMemberDef paramspec_members[] = {
|
|
{"__name__", _Py_T_OBJECT, offsetof(paramspecobject, name), Py_READONLY},
|
|
{"__bound__", _Py_T_OBJECT, offsetof(paramspecobject, bound), Py_READONLY},
|
|
{"__covariant__", Py_T_BOOL, offsetof(paramspecobject, covariant), Py_READONLY},
|
|
{"__contravariant__", Py_T_BOOL, offsetof(paramspecobject, contravariant), Py_READONLY},
|
|
{"__infer_variance__", Py_T_BOOL, offsetof(paramspecobject, infer_variance), Py_READONLY},
|
|
{0}
|
|
};
|
|
|
|
static PyObject *
|
|
paramspec_args(PyObject *self, void *unused)
|
|
{
|
|
PyTypeObject *tp = _PyInterpreterState_GET()->cached_objects.paramspecargs_type;
|
|
return (PyObject *)paramspecattr_new(tp, self);
|
|
}
|
|
|
|
static PyObject *
|
|
paramspec_kwargs(PyObject *self, void *unused)
|
|
{
|
|
PyTypeObject *tp = _PyInterpreterState_GET()->cached_objects.paramspeckwargs_type;
|
|
return (PyObject *)paramspecattr_new(tp, self);
|
|
}
|
|
|
|
static PyObject *
|
|
paramspec_default(paramspecobject *self, void *unused)
|
|
{
|
|
if (self->default_value != NULL) {
|
|
return Py_NewRef(self->default_value);
|
|
}
|
|
if (self->evaluate_default == NULL) {
|
|
return &_Py_NoDefaultStruct;
|
|
}
|
|
PyObject *default_value = PyObject_CallNoArgs(self->evaluate_default);
|
|
self->default_value = Py_XNewRef(default_value);
|
|
return default_value;
|
|
}
|
|
|
|
static PyGetSetDef paramspec_getset[] = {
|
|
{"args", (getter)paramspec_args, NULL, PyDoc_STR("Represents positional arguments."), NULL},
|
|
{"kwargs", (getter)paramspec_kwargs, NULL, PyDoc_STR("Represents keyword arguments."), NULL},
|
|
{"__default__", (getter)paramspec_default, NULL, "The default value for this ParamSpec.", NULL},
|
|
{0},
|
|
};
|
|
|
|
static paramspecobject *
|
|
paramspec_alloc(PyObject *name, PyObject *bound, PyObject *default_value, bool covariant,
|
|
bool contravariant, bool infer_variance, PyObject *module)
|
|
{
|
|
PyTypeObject *tp = _PyInterpreterState_GET()->cached_objects.paramspec_type;
|
|
paramspecobject *ps = PyObject_GC_New(paramspecobject, tp);
|
|
if (ps == NULL) {
|
|
return NULL;
|
|
}
|
|
ps->name = Py_NewRef(name);
|
|
ps->bound = Py_XNewRef(bound);
|
|
ps->covariant = covariant;
|
|
ps->contravariant = contravariant;
|
|
ps->infer_variance = infer_variance;
|
|
ps->default_value = Py_XNewRef(default_value);
|
|
ps->evaluate_default = NULL;
|
|
_PyObject_GC_TRACK(ps);
|
|
if (module != NULL) {
|
|
if (PyObject_SetAttrString((PyObject *)ps, "__module__", module) < 0) {
|
|
Py_DECREF(ps);
|
|
return NULL;
|
|
}
|
|
}
|
|
return ps;
|
|
}
|
|
|
|
/*[clinic input]
|
|
@classmethod
|
|
paramspec.__new__ as paramspec_new
|
|
|
|
name: object(subclass_of="&PyUnicode_Type")
|
|
*
|
|
bound: object = None
|
|
default as default_value: object(c_default="&_Py_NoDefaultStruct") = typing.NoDefault
|
|
covariant: bool = False
|
|
contravariant: bool = False
|
|
infer_variance: bool = False
|
|
|
|
Create a ParamSpec object.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
paramspec_new_impl(PyTypeObject *type, PyObject *name, PyObject *bound,
|
|
PyObject *default_value, int covariant, int contravariant,
|
|
int infer_variance)
|
|
/*[clinic end generated code: output=47ca9d63fa5a094d input=495e1565bc067ab9]*/
|
|
{
|
|
if (covariant && contravariant) {
|
|
PyErr_SetString(PyExc_ValueError, "Bivariant types are not supported.");
|
|
return NULL;
|
|
}
|
|
if (infer_variance && (covariant || contravariant)) {
|
|
PyErr_SetString(PyExc_ValueError, "Variance cannot be specified with infer_variance.");
|
|
return NULL;
|
|
}
|
|
if (bound != NULL) {
|
|
bound = type_check(bound, "Bound must be a type.");
|
|
if (bound == NULL) {
|
|
return NULL;
|
|
}
|
|
}
|
|
PyObject *module = caller();
|
|
if (module == NULL) {
|
|
Py_XDECREF(bound);
|
|
return NULL;
|
|
}
|
|
PyObject *ps = (PyObject *)paramspec_alloc(
|
|
name, bound, default_value, covariant, contravariant, infer_variance, module);
|
|
Py_XDECREF(bound);
|
|
Py_DECREF(module);
|
|
return ps;
|
|
}
|
|
|
|
|
|
/*[clinic input]
|
|
paramspec.__typing_subst__ as paramspec_typing_subst
|
|
|
|
arg: object
|
|
/
|
|
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
paramspec_typing_subst(paramspecobject *self, PyObject *arg)
|
|
/*[clinic end generated code: output=4c5b4aaada1c5814 input=2d5b5e3d4a717189]*/
|
|
{
|
|
PyObject *args[2] = {(PyObject *)self, arg};
|
|
PyObject *result = call_typing_func_object("_paramspec_subst", args, 2);
|
|
return result;
|
|
}
|
|
|
|
/*[clinic input]
|
|
paramspec.__typing_prepare_subst__ as paramspec_typing_prepare_subst
|
|
|
|
alias: object
|
|
args: object
|
|
/
|
|
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
paramspec_typing_prepare_subst_impl(paramspecobject *self, PyObject *alias,
|
|
PyObject *args)
|
|
/*[clinic end generated code: output=95449d630a2adb9a input=6df6f9fef3e150da]*/
|
|
{
|
|
PyObject *args_array[3] = {(PyObject *)self, alias, args};
|
|
PyObject *result = call_typing_func_object(
|
|
"_paramspec_prepare_subst", args_array, 3);
|
|
return result;
|
|
}
|
|
|
|
/*[clinic input]
|
|
paramspec.__reduce__ as paramspec_reduce
|
|
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
paramspec_reduce_impl(paramspecobject *self)
|
|
/*[clinic end generated code: output=b83398674416db27 input=5bf349f0d5dd426c]*/
|
|
{
|
|
return Py_NewRef(self->name);
|
|
}
|
|
|
|
/*[clinic input]
|
|
paramspec.has_default as paramspec_has_default
|
|
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
paramspec_has_default_impl(paramspecobject *self)
|
|
/*[clinic end generated code: output=daaae7467a6a4368 input=2112e97eeb76cd59]*/
|
|
{
|
|
if (self->evaluate_default != NULL ||
|
|
(self->default_value != &_Py_NoDefaultStruct && self->default_value != NULL)) {
|
|
Py_RETURN_TRUE;
|
|
}
|
|
Py_RETURN_FALSE;
|
|
}
|
|
|
|
static PyObject *
|
|
paramspec_mro_entries(PyObject *self, PyObject *args)
|
|
{
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"Cannot subclass an instance of ParamSpec");
|
|
return NULL;
|
|
}
|
|
|
|
static PyMethodDef paramspec_methods[] = {
|
|
PARAMSPEC_TYPING_SUBST_METHODDEF
|
|
PARAMSPEC_TYPING_PREPARE_SUBST_METHODDEF
|
|
PARAMSPEC_HAS_DEFAULT_METHODDEF
|
|
PARAMSPEC_REDUCE_METHODDEF
|
|
{"__mro_entries__", paramspec_mro_entries, METH_O},
|
|
{0}
|
|
};
|
|
|
|
PyDoc_STRVAR(paramspec_doc,
|
|
"Parameter specification variable.\n\
|
|
\n\
|
|
The preferred way to construct a parameter specification is via the\n\
|
|
dedicated syntax for generic functions, classes, and type aliases,\n\
|
|
where the use of '**' creates a parameter specification::\n\
|
|
\n\
|
|
type IntFunc[**P] = Callable[P, int]\n\
|
|
\n\
|
|
The following syntax creates a parameter specification that defaults\n\
|
|
to a callable accepting two positional-only arguments of types int\n\
|
|
and str:\n\
|
|
\n\
|
|
type IntFuncDefault[**P = (int, str)] = Callable[P, int]\n\
|
|
\n\
|
|
For compatibility with Python 3.11 and earlier, ParamSpec objects\n\
|
|
can also be created as follows::\n\
|
|
\n\
|
|
P = ParamSpec('P')\n\
|
|
DefaultP = ParamSpec('DefaultP', default=(int, str))\n\
|
|
\n\
|
|
Parameter specification variables exist primarily for the benefit of\n\
|
|
static type checkers. They are used to forward the parameter types of\n\
|
|
one callable to another callable, a pattern commonly found in\n\
|
|
higher-order functions and decorators. They are only valid when used\n\
|
|
in ``Concatenate``, or as the first argument to ``Callable``, or as\n\
|
|
parameters for user-defined Generics. See class Generic for more\n\
|
|
information on generic types.\n\
|
|
\n\
|
|
An example for annotating a decorator::\n\
|
|
\n\
|
|
def add_logging[**P, T](f: Callable[P, T]) -> Callable[P, T]:\n\
|
|
'''A type-safe decorator to add logging to a function.'''\n\
|
|
def inner(*args: P.args, **kwargs: P.kwargs) -> T:\n\
|
|
logging.info(f'{f.__name__} was called')\n\
|
|
return f(*args, **kwargs)\n\
|
|
return inner\n\
|
|
\n\
|
|
@add_logging\n\
|
|
def add_two(x: float, y: float) -> float:\n\
|
|
'''Add two numbers together.'''\n\
|
|
return x + y\n\
|
|
\n\
|
|
Parameter specification variables can be introspected. e.g.::\n\
|
|
\n\
|
|
>>> P = ParamSpec(\"P\")\n\
|
|
>>> P.__name__\n\
|
|
'P'\n\
|
|
\n\
|
|
Note that only parameter specification variables defined in the global\n\
|
|
scope can be pickled.\n\
|
|
");
|
|
|
|
static PyType_Slot paramspec_slots[] = {
|
|
{Py_tp_doc, (void *)paramspec_doc},
|
|
{Py_tp_members, paramspec_members},
|
|
{Py_tp_methods, paramspec_methods},
|
|
{Py_tp_getset, paramspec_getset},
|
|
// Unions of ParamSpecs have no defined meaning, but they were allowed
|
|
// by the Python implementation, so we allow them here too.
|
|
{Py_nb_or, make_union},
|
|
{Py_tp_new, paramspec_new},
|
|
{Py_tp_dealloc, paramspec_dealloc},
|
|
{Py_tp_alloc, PyType_GenericAlloc},
|
|
{Py_tp_free, PyObject_GC_Del},
|
|
{Py_tp_traverse, paramspec_traverse},
|
|
{Py_tp_clear, paramspec_clear},
|
|
{Py_tp_repr, paramspec_repr},
|
|
{0, 0},
|
|
};
|
|
|
|
PyType_Spec paramspec_spec = {
|
|
.name = "typing.ParamSpec",
|
|
.basicsize = sizeof(paramspecobject),
|
|
.flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC | Py_TPFLAGS_IMMUTABLETYPE
|
|
| Py_TPFLAGS_MANAGED_DICT | Py_TPFLAGS_MANAGED_WEAKREF,
|
|
.slots = paramspec_slots,
|
|
};
|
|
|
|
static void
|
|
typevartuple_dealloc(PyObject *self)
|
|
{
|
|
PyTypeObject *tp = Py_TYPE(self);
|
|
_PyObject_GC_UNTRACK(self);
|
|
typevartupleobject *tvt = (typevartupleobject *)self;
|
|
|
|
Py_DECREF(tvt->name);
|
|
Py_XDECREF(tvt->default_value);
|
|
Py_XDECREF(tvt->evaluate_default);
|
|
PyObject_ClearManagedDict(self);
|
|
PyObject_ClearWeakRefs(self);
|
|
|
|
Py_TYPE(self)->tp_free(self);
|
|
Py_DECREF(tp);
|
|
}
|
|
|
|
static PyObject *
|
|
typevartuple_iter(PyObject *self)
|
|
{
|
|
PyObject *unpacked = typevartuple_unpack(self);
|
|
if (unpacked == NULL) {
|
|
return NULL;
|
|
}
|
|
PyObject *tuple = PyTuple_Pack(1, unpacked);
|
|
if (tuple == NULL) {
|
|
Py_DECREF(unpacked);
|
|
return NULL;
|
|
}
|
|
PyObject *result = PyObject_GetIter(tuple);
|
|
Py_DECREF(unpacked);
|
|
Py_DECREF(tuple);
|
|
return result;
|
|
}
|
|
|
|
static PyObject *
|
|
typevartuple_repr(PyObject *self)
|
|
{
|
|
typevartupleobject *tvt = (typevartupleobject *)self;
|
|
|
|
return Py_NewRef(tvt->name);
|
|
}
|
|
|
|
static PyMemberDef typevartuple_members[] = {
|
|
{"__name__", _Py_T_OBJECT, offsetof(typevartupleobject, name), Py_READONLY},
|
|
{0}
|
|
};
|
|
|
|
static typevartupleobject *
|
|
typevartuple_alloc(PyObject *name, PyObject *module, PyObject *default_value)
|
|
{
|
|
PyTypeObject *tp = _PyInterpreterState_GET()->cached_objects.typevartuple_type;
|
|
typevartupleobject *tvt = PyObject_GC_New(typevartupleobject, tp);
|
|
if (tvt == NULL) {
|
|
return NULL;
|
|
}
|
|
tvt->name = Py_NewRef(name);
|
|
tvt->default_value = Py_XNewRef(default_value);
|
|
tvt->evaluate_default = NULL;
|
|
_PyObject_GC_TRACK(tvt);
|
|
if (module != NULL) {
|
|
if (PyObject_SetAttrString((PyObject *)tvt, "__module__", module) < 0) {
|
|
Py_DECREF(tvt);
|
|
return NULL;
|
|
}
|
|
}
|
|
return tvt;
|
|
}
|
|
|
|
/*[clinic input]
|
|
@classmethod
|
|
typevartuple.__new__
|
|
|
|
name: object(subclass_of="&PyUnicode_Type")
|
|
*
|
|
default as default_value: object(c_default="&_Py_NoDefaultStruct") = typing.NoDefault
|
|
|
|
Create a new TypeVarTuple with the given name.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
typevartuple_impl(PyTypeObject *type, PyObject *name,
|
|
PyObject *default_value)
|
|
/*[clinic end generated code: output=9d6b76dfe95aae51 input=e149739929a866d0]*/
|
|
{
|
|
PyObject *module = caller();
|
|
if (module == NULL) {
|
|
return NULL;
|
|
}
|
|
PyObject *result = (PyObject *)typevartuple_alloc(name, module, default_value);
|
|
Py_DECREF(module);
|
|
return result;
|
|
}
|
|
|
|
/*[clinic input]
|
|
typevartuple.__typing_subst__ as typevartuple_typing_subst
|
|
|
|
arg: object
|
|
/
|
|
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
typevartuple_typing_subst(typevartupleobject *self, PyObject *arg)
|
|
/*[clinic end generated code: output=237054c6d7484eea input=3fcf2dfd9eee7945]*/
|
|
{
|
|
PyErr_SetString(PyExc_TypeError, "Substitution of bare TypeVarTuple is not supported");
|
|
return NULL;
|
|
}
|
|
|
|
/*[clinic input]
|
|
typevartuple.__typing_prepare_subst__ as typevartuple_typing_prepare_subst
|
|
|
|
alias: object
|
|
args: object
|
|
/
|
|
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
typevartuple_typing_prepare_subst_impl(typevartupleobject *self,
|
|
PyObject *alias, PyObject *args)
|
|
/*[clinic end generated code: output=ff999bc5b02036c1 input=685b149b0fc47556]*/
|
|
{
|
|
PyObject *args_array[3] = {(PyObject *)self, alias, args};
|
|
PyObject *result = call_typing_func_object(
|
|
"_typevartuple_prepare_subst", args_array, 3);
|
|
return result;
|
|
}
|
|
|
|
/*[clinic input]
|
|
typevartuple.__reduce__ as typevartuple_reduce
|
|
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
typevartuple_reduce_impl(typevartupleobject *self)
|
|
/*[clinic end generated code: output=3215bc0477913d20 input=3018a4d66147e807]*/
|
|
{
|
|
return Py_NewRef(self->name);
|
|
}
|
|
|
|
|
|
/*[clinic input]
|
|
typevartuple.has_default as typevartuple_has_default
|
|
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
typevartuple_has_default_impl(typevartupleobject *self)
|
|
/*[clinic end generated code: output=4895f602f56a5e29 input=9ef3250ddb2c1851]*/
|
|
{
|
|
if (self->evaluate_default != NULL ||
|
|
(self->default_value != &_Py_NoDefaultStruct && self->default_value != NULL)) {
|
|
Py_RETURN_TRUE;
|
|
}
|
|
Py_RETURN_FALSE;
|
|
}
|
|
|
|
static PyObject *
|
|
typevartuple_mro_entries(PyObject *self, PyObject *args)
|
|
{
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"Cannot subclass an instance of TypeVarTuple");
|
|
return NULL;
|
|
}
|
|
|
|
static int
|
|
typevartuple_traverse(PyObject *self, visitproc visit, void *arg)
|
|
{
|
|
Py_VISIT(Py_TYPE(self));
|
|
Py_VISIT(((typevartupleobject *)self)->default_value);
|
|
Py_VISIT(((typevartupleobject *)self)->evaluate_default);
|
|
PyObject_VisitManagedDict(self, visit, arg);
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
typevartuple_clear(PyObject *self)
|
|
{
|
|
Py_CLEAR(((typevartupleobject *)self)->default_value);
|
|
Py_CLEAR(((typevartupleobject *)self)->evaluate_default);
|
|
PyObject_ClearManagedDict(self);
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *
|
|
typevartuple_default(typevartupleobject *self, void *unused)
|
|
{
|
|
if (self->default_value != NULL) {
|
|
return Py_NewRef(self->default_value);
|
|
}
|
|
if (self->evaluate_default == NULL) {
|
|
return &_Py_NoDefaultStruct;
|
|
}
|
|
PyObject *default_value = PyObject_CallNoArgs(self->evaluate_default);
|
|
self->default_value = Py_XNewRef(default_value);
|
|
return default_value;
|
|
}
|
|
|
|
static PyGetSetDef typevartuple_getset[] = {
|
|
{"__default__", (getter)typevartuple_default, NULL, "The default value for this TypeVarTuple.", NULL},
|
|
{0},
|
|
};
|
|
|
|
static PyMethodDef typevartuple_methods[] = {
|
|
TYPEVARTUPLE_TYPING_SUBST_METHODDEF
|
|
TYPEVARTUPLE_TYPING_PREPARE_SUBST_METHODDEF
|
|
TYPEVARTUPLE_REDUCE_METHODDEF
|
|
TYPEVARTUPLE_HAS_DEFAULT_METHODDEF
|
|
{"__mro_entries__", typevartuple_mro_entries, METH_O},
|
|
{0}
|
|
};
|
|
|
|
PyDoc_STRVAR(typevartuple_doc,
|
|
"Type variable tuple. A specialized form of type variable that enables\n\
|
|
variadic generics.\n\
|
|
\n\
|
|
The preferred way to construct a type variable tuple is via the\n\
|
|
dedicated syntax for generic functions, classes, and type aliases,\n\
|
|
where a single '*' indicates a type variable tuple::\n\
|
|
\n\
|
|
def move_first_element_to_last[T, *Ts](tup: tuple[T, *Ts]) -> tuple[*Ts, T]:\n\
|
|
return (*tup[1:], tup[0])\n\
|
|
\n\
|
|
Type variables tuples can have default values:\n\
|
|
\n\
|
|
type AliasWithDefault[*Ts = (str, int)] = tuple[*Ts]\n\
|
|
\n\
|
|
For compatibility with Python 3.11 and earlier, TypeVarTuple objects\n\
|
|
can also be created as follows::\n\
|
|
\n\
|
|
Ts = TypeVarTuple('Ts') # Can be given any name\n\
|
|
DefaultTs = TypeVarTuple('Ts', default=(str, int))\n\
|
|
\n\
|
|
Just as a TypeVar (type variable) is a placeholder for a single type,\n\
|
|
a TypeVarTuple is a placeholder for an *arbitrary* number of types. For\n\
|
|
example, if we define a generic class using a TypeVarTuple::\n\
|
|
\n\
|
|
class C[*Ts]: ...\n\
|
|
\n\
|
|
Then we can parameterize that class with an arbitrary number of type\n\
|
|
arguments::\n\
|
|
\n\
|
|
C[int] # Fine\n\
|
|
C[int, str] # Also fine\n\
|
|
C[()] # Even this is fine\n\
|
|
\n\
|
|
For more details, see PEP 646.\n\
|
|
\n\
|
|
Note that only TypeVarTuples defined in the global scope can be\n\
|
|
pickled.\n\
|
|
");
|
|
|
|
PyType_Slot typevartuple_slots[] = {
|
|
{Py_tp_doc, (void *)typevartuple_doc},
|
|
{Py_tp_members, typevartuple_members},
|
|
{Py_tp_methods, typevartuple_methods},
|
|
{Py_tp_getset, typevartuple_getset},
|
|
{Py_tp_new, typevartuple},
|
|
{Py_tp_iter, typevartuple_iter},
|
|
{Py_tp_repr, typevartuple_repr},
|
|
{Py_tp_dealloc, typevartuple_dealloc},
|
|
{Py_tp_alloc, PyType_GenericAlloc},
|
|
{Py_tp_free, PyObject_GC_Del},
|
|
{Py_tp_traverse, typevartuple_traverse},
|
|
{Py_tp_clear, typevartuple_clear},
|
|
{0, 0},
|
|
};
|
|
|
|
PyType_Spec typevartuple_spec = {
|
|
.name = "typing.TypeVarTuple",
|
|
.basicsize = sizeof(typevartupleobject),
|
|
.flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_IMMUTABLETYPE | Py_TPFLAGS_MANAGED_DICT
|
|
| Py_TPFLAGS_HAVE_GC | Py_TPFLAGS_MANAGED_WEAKREF,
|
|
.slots = typevartuple_slots,
|
|
};
|
|
|
|
PyObject *
|
|
_Py_make_typevar(PyObject *name, PyObject *evaluate_bound, PyObject *evaluate_constraints)
|
|
{
|
|
return (PyObject *)typevar_alloc(name, NULL, evaluate_bound, NULL, evaluate_constraints,
|
|
NULL, false, false, true, NULL);
|
|
}
|
|
|
|
PyObject *
|
|
_Py_make_paramspec(PyThreadState *Py_UNUSED(ignored), PyObject *v)
|
|
{
|
|
assert(PyUnicode_Check(v));
|
|
return (PyObject *)paramspec_alloc(v, NULL, NULL, false, false, true, NULL);
|
|
}
|
|
|
|
PyObject *
|
|
_Py_make_typevartuple(PyThreadState *Py_UNUSED(ignored), PyObject *v)
|
|
{
|
|
assert(PyUnicode_Check(v));
|
|
return (PyObject *)typevartuple_alloc(v, NULL, NULL);
|
|
}
|
|
|
|
static void
|
|
typealias_dealloc(PyObject *self)
|
|
{
|
|
PyTypeObject *tp = Py_TYPE(self);
|
|
_PyObject_GC_UNTRACK(self);
|
|
typealiasobject *ta = (typealiasobject *)self;
|
|
Py_DECREF(ta->name);
|
|
Py_XDECREF(ta->type_params);
|
|
Py_XDECREF(ta->compute_value);
|
|
Py_XDECREF(ta->value);
|
|
Py_XDECREF(ta->module);
|
|
Py_TYPE(self)->tp_free(self);
|
|
Py_DECREF(tp);
|
|
}
|
|
|
|
static PyObject *
|
|
typealias_get_value(typealiasobject *ta)
|
|
{
|
|
if (ta->value != NULL) {
|
|
return Py_NewRef(ta->value);
|
|
}
|
|
PyObject *result = PyObject_CallNoArgs(ta->compute_value);
|
|
if (result == NULL) {
|
|
return NULL;
|
|
}
|
|
ta->value = Py_NewRef(result);
|
|
return result;
|
|
}
|
|
|
|
static PyObject *
|
|
typealias_repr(PyObject *self)
|
|
{
|
|
typealiasobject *ta = (typealiasobject *)self;
|
|
return Py_NewRef(ta->name);
|
|
}
|
|
|
|
static PyMemberDef typealias_members[] = {
|
|
{"__name__", _Py_T_OBJECT, offsetof(typealiasobject, name), Py_READONLY},
|
|
{0}
|
|
};
|
|
|
|
static PyObject *
|
|
typealias_value(PyObject *self, void *unused)
|
|
{
|
|
typealiasobject *ta = (typealiasobject *)self;
|
|
return typealias_get_value(ta);
|
|
}
|
|
|
|
static PyObject *
|
|
typealias_parameters(PyObject *self, void *unused)
|
|
{
|
|
typealiasobject *ta = (typealiasobject *)self;
|
|
if (ta->type_params == NULL) {
|
|
return PyTuple_New(0);
|
|
}
|
|
return unpack_typevartuples(ta->type_params);
|
|
}
|
|
|
|
static PyObject *
|
|
typealias_type_params(PyObject *self, void *unused)
|
|
{
|
|
typealiasobject *ta = (typealiasobject *)self;
|
|
if (ta->type_params == NULL) {
|
|
return PyTuple_New(0);
|
|
}
|
|
return Py_NewRef(ta->type_params);
|
|
}
|
|
|
|
static PyObject *
|
|
typealias_module(PyObject *self, void *unused)
|
|
{
|
|
typealiasobject *ta = (typealiasobject *)self;
|
|
if (ta->module != NULL) {
|
|
return Py_NewRef(ta->module);
|
|
}
|
|
if (ta->compute_value != NULL) {
|
|
PyObject* mod = PyFunction_GetModule(ta->compute_value);
|
|
if (mod != NULL) {
|
|
// PyFunction_GetModule() returns a borrowed reference,
|
|
// and it may return NULL (e.g., for functions defined
|
|
// in an exec()'ed block).
|
|
return Py_NewRef(mod);
|
|
}
|
|
}
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyGetSetDef typealias_getset[] = {
|
|
{"__parameters__", typealias_parameters, (setter)NULL, NULL, NULL},
|
|
{"__type_params__", typealias_type_params, (setter)NULL, NULL, NULL},
|
|
{"__value__", typealias_value, (setter)NULL, NULL, NULL},
|
|
{"__module__", typealias_module, (setter)NULL, NULL, NULL},
|
|
{0}
|
|
};
|
|
|
|
static typealiasobject *
|
|
typealias_alloc(PyObject *name, PyObject *type_params, PyObject *compute_value,
|
|
PyObject *value, PyObject *module)
|
|
{
|
|
typealiasobject *ta = PyObject_GC_New(typealiasobject, &_PyTypeAlias_Type);
|
|
if (ta == NULL) {
|
|
return NULL;
|
|
}
|
|
ta->name = Py_NewRef(name);
|
|
ta->type_params = Py_IsNone(type_params) ? NULL : Py_XNewRef(type_params);
|
|
ta->compute_value = Py_XNewRef(compute_value);
|
|
ta->value = Py_XNewRef(value);
|
|
ta->module = Py_XNewRef(module);
|
|
_PyObject_GC_TRACK(ta);
|
|
return ta;
|
|
}
|
|
|
|
static int
|
|
typealias_traverse(typealiasobject *self, visitproc visit, void *arg)
|
|
{
|
|
Py_VISIT(self->type_params);
|
|
Py_VISIT(self->compute_value);
|
|
Py_VISIT(self->value);
|
|
Py_VISIT(self->module);
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
typealias_clear(typealiasobject *self)
|
|
{
|
|
Py_CLEAR(self->type_params);
|
|
Py_CLEAR(self->compute_value);
|
|
Py_CLEAR(self->value);
|
|
Py_CLEAR(self->module);
|
|
return 0;
|
|
}
|
|
|
|
/*[clinic input]
|
|
typealias.__reduce__ as typealias_reduce
|
|
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
typealias_reduce_impl(typealiasobject *self)
|
|
/*[clinic end generated code: output=913724f92ad3b39b input=4f06fbd9472ec0f1]*/
|
|
{
|
|
return Py_NewRef(self->name);
|
|
}
|
|
|
|
static PyObject *
|
|
typealias_subscript(PyObject *self, PyObject *args)
|
|
{
|
|
if (((typealiasobject *)self)->type_params == NULL) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"Only generic type aliases are subscriptable");
|
|
return NULL;
|
|
}
|
|
return Py_GenericAlias(self, args);
|
|
}
|
|
|
|
static PyMethodDef typealias_methods[] = {
|
|
TYPEALIAS_REDUCE_METHODDEF
|
|
{0}
|
|
};
|
|
|
|
|
|
/*[clinic input]
|
|
@classmethod
|
|
typealias.__new__ as typealias_new
|
|
|
|
name: object(subclass_of="&PyUnicode_Type")
|
|
value: object
|
|
*
|
|
type_params: object = NULL
|
|
|
|
Create a TypeAliasType.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
typealias_new_impl(PyTypeObject *type, PyObject *name, PyObject *value,
|
|
PyObject *type_params)
|
|
/*[clinic end generated code: output=8920ce6bdff86f00 input=df163c34e17e1a35]*/
|
|
{
|
|
if (type_params != NULL && !PyTuple_Check(type_params)) {
|
|
PyErr_SetString(PyExc_TypeError, "type_params must be a tuple");
|
|
return NULL;
|
|
}
|
|
PyObject *module = caller();
|
|
if (module == NULL) {
|
|
return NULL;
|
|
}
|
|
PyObject *ta = (PyObject *)typealias_alloc(name, type_params, NULL, value,
|
|
module);
|
|
Py_DECREF(module);
|
|
return ta;
|
|
}
|
|
|
|
PyDoc_STRVAR(typealias_doc,
|
|
"Type alias.\n\
|
|
\n\
|
|
Type aliases are created through the type statement::\n\
|
|
\n\
|
|
type Alias = int\n\
|
|
\n\
|
|
In this example, Alias and int will be treated equivalently by static\n\
|
|
type checkers.\n\
|
|
\n\
|
|
At runtime, Alias is an instance of TypeAliasType. The __name__\n\
|
|
attribute holds the name of the type alias. The value of the type alias\n\
|
|
is stored in the __value__ attribute. It is evaluated lazily, so the\n\
|
|
value is computed only if the attribute is accessed.\n\
|
|
\n\
|
|
Type aliases can also be generic::\n\
|
|
\n\
|
|
type ListOrSet[T] = list[T] | set[T]\n\
|
|
\n\
|
|
In this case, the type parameters of the alias are stored in the\n\
|
|
__type_params__ attribute.\n\
|
|
\n\
|
|
See PEP 695 for more information.\n\
|
|
");
|
|
|
|
static PyNumberMethods typealias_as_number = {
|
|
.nb_or = _Py_union_type_or,
|
|
};
|
|
|
|
static PyMappingMethods typealias_as_mapping = {
|
|
.mp_subscript = typealias_subscript,
|
|
};
|
|
|
|
PyTypeObject _PyTypeAlias_Type = {
|
|
PyVarObject_HEAD_INIT(&PyType_Type, 0)
|
|
.tp_name = "typing.TypeAliasType",
|
|
.tp_basicsize = sizeof(typealiasobject),
|
|
.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_IMMUTABLETYPE | Py_TPFLAGS_HAVE_GC,
|
|
.tp_doc = typealias_doc,
|
|
.tp_members = typealias_members,
|
|
.tp_methods = typealias_methods,
|
|
.tp_getset = typealias_getset,
|
|
.tp_alloc = PyType_GenericAlloc,
|
|
.tp_dealloc = typealias_dealloc,
|
|
.tp_new = typealias_new,
|
|
.tp_free = PyObject_GC_Del,
|
|
.tp_traverse = (traverseproc)typealias_traverse,
|
|
.tp_clear = (inquiry)typealias_clear,
|
|
.tp_repr = typealias_repr,
|
|
.tp_as_number = &typealias_as_number,
|
|
.tp_as_mapping = &typealias_as_mapping,
|
|
};
|
|
|
|
PyObject *
|
|
_Py_make_typealias(PyThreadState* unused, PyObject *args)
|
|
{
|
|
assert(PyTuple_Check(args));
|
|
assert(PyTuple_GET_SIZE(args) == 3);
|
|
PyObject *name = PyTuple_GET_ITEM(args, 0);
|
|
assert(PyUnicode_Check(name));
|
|
PyObject *type_params = PyTuple_GET_ITEM(args, 1);
|
|
PyObject *compute_value = PyTuple_GET_ITEM(args, 2);
|
|
assert(PyFunction_Check(compute_value));
|
|
return (PyObject *)typealias_alloc(name, type_params, compute_value, NULL, NULL);
|
|
}
|
|
|
|
PyDoc_STRVAR(generic_doc,
|
|
"Abstract base class for generic types.\n\
|
|
\n\
|
|
On Python 3.12 and newer, generic classes implicitly inherit from\n\
|
|
Generic when they declare a parameter list after the class's name::\n\
|
|
\n\
|
|
class Mapping[KT, VT]:\n\
|
|
def __getitem__(self, key: KT) -> VT:\n\
|
|
...\n\
|
|
# Etc.\n\
|
|
\n\
|
|
On older versions of Python, however, generic classes have to\n\
|
|
explicitly inherit from Generic.\n\
|
|
\n\
|
|
After a class has been declared to be generic, it can then be used as\n\
|
|
follows::\n\
|
|
\n\
|
|
def lookup_name[KT, VT](mapping: Mapping[KT, VT], key: KT, default: VT) -> VT:\n\
|
|
try:\n\
|
|
return mapping[key]\n\
|
|
except KeyError:\n\
|
|
return default\n\
|
|
");
|
|
|
|
PyDoc_STRVAR(generic_class_getitem_doc,
|
|
"Parameterizes a generic class.\n\
|
|
\n\
|
|
At least, parameterizing a generic class is the *main* thing this\n\
|
|
method does. For example, for some generic class `Foo`, this is called\n\
|
|
when we do `Foo[int]` - there, with `cls=Foo` and `params=int`.\n\
|
|
\n\
|
|
However, note that this method is also called when defining generic\n\
|
|
classes in the first place with `class Foo[T]: ...`.\n\
|
|
");
|
|
|
|
static PyObject *
|
|
call_typing_args_kwargs(const char *name, PyTypeObject *cls, PyObject *args, PyObject *kwargs)
|
|
{
|
|
PyObject *typing = NULL, *func = NULL, *new_args = NULL;
|
|
typing = PyImport_ImportModule("typing");
|
|
if (typing == NULL) {
|
|
goto error;
|
|
}
|
|
func = PyObject_GetAttrString(typing, name);
|
|
if (func == NULL) {
|
|
goto error;
|
|
}
|
|
assert(PyTuple_Check(args));
|
|
Py_ssize_t nargs = PyTuple_GET_SIZE(args);
|
|
new_args = PyTuple_New(nargs + 1);
|
|
if (new_args == NULL) {
|
|
goto error;
|
|
}
|
|
PyTuple_SET_ITEM(new_args, 0, Py_NewRef((PyObject *)cls));
|
|
for (Py_ssize_t i = 0; i < nargs; i++) {
|
|
PyObject *arg = PyTuple_GET_ITEM(args, i);
|
|
PyTuple_SET_ITEM(new_args, i + 1, Py_NewRef(arg));
|
|
}
|
|
PyObject *result = PyObject_Call(func, new_args, kwargs);
|
|
Py_DECREF(typing);
|
|
Py_DECREF(func);
|
|
Py_DECREF(new_args);
|
|
return result;
|
|
error:
|
|
Py_XDECREF(typing);
|
|
Py_XDECREF(func);
|
|
Py_XDECREF(new_args);
|
|
return NULL;
|
|
}
|
|
|
|
static PyObject *
|
|
generic_init_subclass(PyTypeObject *cls, PyObject *args, PyObject *kwargs)
|
|
{
|
|
return call_typing_args_kwargs("_generic_init_subclass", cls, args, kwargs);
|
|
}
|
|
|
|
static PyObject *
|
|
generic_class_getitem(PyTypeObject *cls, PyObject *args, PyObject *kwargs)
|
|
{
|
|
return call_typing_args_kwargs("_generic_class_getitem", cls, args, kwargs);
|
|
}
|
|
|
|
PyObject *
|
|
_Py_subscript_generic(PyThreadState* unused, PyObject *params)
|
|
{
|
|
params = unpack_typevartuples(params);
|
|
|
|
PyInterpreterState *interp = _PyInterpreterState_GET();
|
|
if (interp->cached_objects.generic_type == NULL) {
|
|
PyErr_SetString(PyExc_SystemError, "Cannot find Generic type");
|
|
return NULL;
|
|
}
|
|
PyObject *args[2] = {(PyObject *)interp->cached_objects.generic_type, params};
|
|
PyObject *result = call_typing_func_object("_GenericAlias", args, 2);
|
|
Py_DECREF(params);
|
|
return result;
|
|
}
|
|
|
|
static PyMethodDef generic_methods[] = {
|
|
{"__class_getitem__", (PyCFunction)(void (*)(void))generic_class_getitem,
|
|
METH_VARARGS | METH_KEYWORDS | METH_CLASS,
|
|
generic_class_getitem_doc},
|
|
{"__init_subclass__", (PyCFunction)(void (*)(void))generic_init_subclass,
|
|
METH_VARARGS | METH_KEYWORDS | METH_CLASS,
|
|
PyDoc_STR("Function to initialize subclasses.")},
|
|
{NULL} /* Sentinel */
|
|
};
|
|
|
|
static void
|
|
generic_dealloc(PyObject *self)
|
|
{
|
|
PyTypeObject *tp = Py_TYPE(self);
|
|
_PyObject_GC_UNTRACK(self);
|
|
Py_TYPE(self)->tp_free(self);
|
|
Py_DECREF(tp);
|
|
}
|
|
|
|
static int
|
|
generic_traverse(PyObject *self, visitproc visit, void *arg)
|
|
{
|
|
Py_VISIT(Py_TYPE(self));
|
|
return 0;
|
|
}
|
|
|
|
static PyType_Slot generic_slots[] = {
|
|
{Py_tp_doc, (void *)generic_doc},
|
|
{Py_tp_methods, generic_methods},
|
|
{Py_tp_dealloc, generic_dealloc},
|
|
{Py_tp_alloc, PyType_GenericAlloc},
|
|
{Py_tp_free, PyObject_GC_Del},
|
|
{Py_tp_traverse, generic_traverse},
|
|
{0, NULL},
|
|
};
|
|
|
|
PyType_Spec generic_spec = {
|
|
.name = "typing.Generic",
|
|
.basicsize = sizeof(PyObject),
|
|
.flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC,
|
|
.slots = generic_slots,
|
|
};
|
|
|
|
int _Py_initialize_generic(PyInterpreterState *interp)
|
|
{
|
|
#define MAKE_TYPE(name) \
|
|
do { \
|
|
PyTypeObject *name ## _type = (PyTypeObject *)PyType_FromSpec(&name ## _spec); \
|
|
if (name ## _type == NULL) { \
|
|
return -1; \
|
|
} \
|
|
interp->cached_objects.name ## _type = name ## _type; \
|
|
} while(0)
|
|
|
|
MAKE_TYPE(generic);
|
|
MAKE_TYPE(typevar);
|
|
MAKE_TYPE(typevartuple);
|
|
MAKE_TYPE(paramspec);
|
|
MAKE_TYPE(paramspecargs);
|
|
MAKE_TYPE(paramspeckwargs);
|
|
#undef MAKE_TYPE
|
|
return 0;
|
|
}
|
|
|
|
void _Py_clear_generic_types(PyInterpreterState *interp)
|
|
{
|
|
Py_CLEAR(interp->cached_objects.generic_type);
|
|
Py_CLEAR(interp->cached_objects.typevar_type);
|
|
Py_CLEAR(interp->cached_objects.typevartuple_type);
|
|
Py_CLEAR(interp->cached_objects.paramspec_type);
|
|
Py_CLEAR(interp->cached_objects.paramspecargs_type);
|
|
Py_CLEAR(interp->cached_objects.paramspeckwargs_type);
|
|
}
|
|
|
|
PyObject *
|
|
_Py_set_typeparam_default(PyThreadState *ts, PyObject *typeparam, PyObject *evaluate_default)
|
|
{
|
|
if (Py_IS_TYPE(typeparam, ts->interp->cached_objects.typevar_type)) {
|
|
Py_XSETREF(((typevarobject *)typeparam)->evaluate_default, Py_NewRef(evaluate_default));
|
|
return Py_NewRef(typeparam);
|
|
}
|
|
else if (Py_IS_TYPE(typeparam, ts->interp->cached_objects.paramspec_type)) {
|
|
Py_XSETREF(((paramspecobject *)typeparam)->evaluate_default, Py_NewRef(evaluate_default));
|
|
return Py_NewRef(typeparam);
|
|
}
|
|
else if (Py_IS_TYPE(typeparam, ts->interp->cached_objects.typevartuple_type)) {
|
|
Py_XSETREF(((typevartupleobject *)typeparam)->evaluate_default, Py_NewRef(evaluate_default));
|
|
return Py_NewRef(typeparam);
|
|
}
|
|
else {
|
|
PyErr_Format(PyExc_TypeError, "Expected a type param, got %R", typeparam);
|
|
return NULL;
|
|
}
|
|
}
|