| // 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 nodefault_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(nodefault_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 = nodefault_doc, |
| .tp_methods = nodefault_methods, |
| .tp_new = nodefault_new, |
| }; |
| |
| PyObject _Py_NoDefaultStruct = _PyObject_HEAD_INIT(&_PyNoDefault_Type); |
| |
| typedef struct { |
| PyObject_HEAD |
| PyObject *value; |
| } constevaluatorobject; |
| |
| static void |
| constevaluator_dealloc(PyObject *self) |
| { |
| PyTypeObject *tp = Py_TYPE(self); |
| constevaluatorobject *ce = (constevaluatorobject *)self; |
| |
| _PyObject_GC_UNTRACK(self); |
| |
| Py_XDECREF(ce->value); |
| |
| Py_TYPE(self)->tp_free(self); |
| Py_DECREF(tp); |
| } |
| |
| static int |
| constevaluator_traverse(PyObject *self, visitproc visit, void *arg) |
| { |
| constevaluatorobject *ce = (constevaluatorobject *)self; |
| Py_VISIT(ce->value); |
| return 0; |
| } |
| |
| static int |
| constevaluator_clear(PyObject *self) |
| { |
| Py_CLEAR(((constevaluatorobject *)self)->value); |
| return 0; |
| } |
| |
| static PyObject * |
| constevaluator_repr(PyObject *self) |
| { |
| PyObject *value = ((constevaluatorobject *)self)->value; |
| return PyUnicode_FromFormat("<constevaluator %R>", value); |
| } |
| |
| static PyObject * |
| constevaluator_call(PyObject *self, PyObject *args, PyObject *kwargs) |
| { |
| if (!_PyArg_NoKeywords("constevaluator.__call__", kwargs)) { |
| return NULL; |
| } |
| int format; |
| if (!PyArg_ParseTuple(args, "i:constevaluator.__call__", &format)) { |
| return NULL; |
| } |
| PyObject *value = ((constevaluatorobject *)self)->value; |
| if (format == 3) { // STRING |
| PyUnicodeWriter *writer = PyUnicodeWriter_Create(5); // cannot be <5 |
| if (writer == NULL) { |
| return NULL; |
| } |
| if (PyTuple_Check(value)) { |
| if (PyUnicodeWriter_WriteChar(writer, '(') < 0) { |
| PyUnicodeWriter_Discard(writer); |
| return NULL; |
| } |
| for (Py_ssize_t i = 0; i < PyTuple_GET_SIZE(value); i++) { |
| PyObject *item = PyTuple_GET_ITEM(value, i); |
| if (i > 0) { |
| if (PyUnicodeWriter_WriteUTF8(writer, ", ", 2) < 0) { |
| PyUnicodeWriter_Discard(writer); |
| return NULL; |
| } |
| } |
| if (_Py_typing_type_repr(writer, item) < 0) { |
| PyUnicodeWriter_Discard(writer); |
| return NULL; |
| } |
| } |
| if (PyUnicodeWriter_WriteChar(writer, ')') < 0) { |
| PyUnicodeWriter_Discard(writer); |
| return NULL; |
| } |
| } |
| else { |
| if (_Py_typing_type_repr(writer, value) < 0) { |
| PyUnicodeWriter_Discard(writer); |
| return NULL; |
| } |
| } |
| return PyUnicodeWriter_Finish(writer); |
| } |
| return Py_NewRef(value); |
| } |
| |
| static PyObject * |
| constevaluator_alloc(PyObject *value) |
| { |
| PyTypeObject *tp = _PyInterpreterState_GET()->cached_objects.constevaluator_type; |
| assert(tp != NULL); |
| constevaluatorobject *ce = PyObject_GC_New(constevaluatorobject, tp); |
| if (ce == NULL) { |
| return NULL; |
| } |
| ce->value = Py_NewRef(value); |
| _PyObject_GC_TRACK(ce); |
| return (PyObject *)ce; |
| |
| } |
| |
| PyDoc_STRVAR(constevaluator_doc, |
| "_ConstEvaluator()\n" |
| "--\n\n" |
| "Internal type for implementing evaluation functions."); |
| |
| static PyType_Slot constevaluator_slots[] = { |
| {Py_tp_doc, (void *)constevaluator_doc}, |
| {Py_tp_dealloc, constevaluator_dealloc}, |
| {Py_tp_traverse, constevaluator_traverse}, |
| {Py_tp_clear, constevaluator_clear}, |
| {Py_tp_repr, constevaluator_repr}, |
| {Py_tp_call, constevaluator_call}, |
| {Py_tp_alloc, PyType_GenericAlloc}, |
| {Py_tp_free, PyObject_GC_Del}, |
| {0, NULL}, |
| }; |
| |
| PyType_Spec constevaluator_spec = { |
| .name = "_typing._ConstEvaluator", |
| .basicsize = sizeof(constevaluatorobject), |
| .flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC | Py_TPFLAGS_IMMUTABLETYPE |
| | Py_TPFLAGS_DISALLOW_INSTANTIATION, |
| .slots = constevaluator_slots, |
| }; |
| |
| int |
| _Py_typing_type_repr(PyUnicodeWriter *writer, PyObject *p) |
| { |
| PyObject *qualname = NULL; |
| PyObject *module = NULL; |
| PyObject *r = NULL; |
| int rc; |
| |
| if (p == Py_Ellipsis) { |
| // The Ellipsis object |
| r = PyUnicode_FromString("..."); |
| goto exit; |
| } |
| |
| if (p == (PyObject *)&_PyNone_Type) { |
| return PyUnicodeWriter_WriteUTF8(writer, "None", 4); |
| } |
| |
| if ((rc = PyObject_HasAttrWithError(p, &_Py_ID(__origin__))) > 0 && |
| (rc = PyObject_HasAttrWithError(p, &_Py_ID(__args__))) > 0) |
| { |
| // It looks like a GenericAlias |
| goto use_repr; |
| } |
| if (rc < 0) { |
| goto exit; |
| } |
| |
| if (PyObject_GetOptionalAttr(p, &_Py_ID(__qualname__), &qualname) < 0) { |
| goto exit; |
| } |
| if (qualname == NULL) { |
| goto use_repr; |
| } |
| if (PyObject_GetOptionalAttr(p, &_Py_ID(__module__), &module) < 0) { |
| goto exit; |
| } |
| if (module == NULL || module == Py_None) { |
| goto use_repr; |
| } |
| |
| // Looks like a class |
| if (PyUnicode_Check(module) && |
| _PyUnicode_EqualToASCIIString(module, "builtins")) |
| { |
| // builtins don't need a module name |
| r = PyObject_Str(qualname); |
| goto exit; |
| } |
| else { |
| r = PyUnicode_FromFormat("%S.%S", module, qualname); |
| goto exit; |
| } |
| |
| use_repr: |
| r = PyObject_Repr(p); |
| exit: |
| Py_XDECREF(qualname); |
| Py_XDECREF(module); |
| if (r == NULL) { |
| return -1; |
| } |
| rc = PyUnicodeWriter_WriteStr(writer, r); |
| Py_DECREF(r); |
| return rc; |
| } |
| |
| |
| 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 || PyStackRef_IsNull(f->f_funcobj)) { |
| Py_RETURN_NONE; |
| } |
| PyObject *r = PyFunction_GetModule(PyStackRef_AsPyObjectBorrow(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 PyObject * |
| typevar_evaluate_bound(typevarobject *self, void *Py_UNUSED(ignored)) |
| { |
| if (self->evaluate_bound != NULL) { |
| return Py_NewRef(self->evaluate_bound); |
| } |
| if (self->bound != NULL) { |
| return constevaluator_alloc(self->bound); |
| } |
| Py_RETURN_NONE; |
| } |
| |
| static PyObject * |
| typevar_evaluate_constraints(typevarobject *self, void *Py_UNUSED(ignored)) |
| { |
| if (self->evaluate_constraints != NULL) { |
| return Py_NewRef(self->evaluate_constraints); |
| } |
| if (self->constraints != NULL) { |
| return constevaluator_alloc(self->constraints); |
| } |
| Py_RETURN_NONE; |
| } |
| |
| static PyObject * |
| typevar_evaluate_default(typevarobject *self, void *Py_UNUSED(ignored)) |
| { |
| if (self->evaluate_default != NULL) { |
| return Py_NewRef(self->evaluate_default); |
| } |
| if (self->default_value != NULL) { |
| return constevaluator_alloc(self->default_value); |
| } |
| Py_RETURN_NONE; |
| } |
| |
| 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}, |
| {"evaluate_bound", (getter)typevar_evaluate_bound, NULL, NULL, NULL}, |
| {"evaluate_constraints", (getter)typevar_evaluate_constraints, NULL, NULL, NULL}, |
| {"evaluate_default", (getter)typevar_evaluate_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: tuple |
| 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=1b5b62e40c92c167]*/ |
| { |
| 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 PyObject * |
| paramspec_evaluate_default(paramspecobject *self, void *unused) |
| { |
| if (self->evaluate_default != NULL) { |
| return Py_NewRef(self->evaluate_default); |
| } |
| if (self->default_value != NULL) { |
| return constevaluator_alloc(self->default_value); |
| } |
| Py_RETURN_NONE; |
| } |
| |
| 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}, |
| {"evaluate_default", (getter)paramspec_evaluate_default, NULL, NULL, 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 PyObject * |
| typevartuple_evaluate_default(typevartupleobject *self, void *unused) |
| { |
| if (self->evaluate_default != NULL) { |
| return Py_NewRef(self->evaluate_default); |
| } |
| if (self->default_value != NULL) { |
| return constevaluator_alloc(self->default_value); |
| } |
| Py_RETURN_NONE; |
| } |
| |
| static PyGetSetDef typevartuple_getset[] = { |
| {"__default__", (getter)typevartuple_default, NULL, "The default value for this TypeVarTuple.", NULL}, |
| {"evaluate_default", (getter)typevartuple_evaluate_default, NULL, NULL, 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 PyObject * |
| get_type_param_default(PyThreadState *ts, PyObject *typeparam) { |
| // Does not modify refcount of existing objects. |
| if (Py_IS_TYPE(typeparam, ts->interp->cached_objects.typevar_type)) { |
| return typevar_default((typevarobject *)typeparam, NULL); |
| } |
| else if (Py_IS_TYPE(typeparam, ts->interp->cached_objects.paramspec_type)) { |
| return paramspec_default((paramspecobject *)typeparam, NULL); |
| } |
| else if (Py_IS_TYPE(typeparam, ts->interp->cached_objects.typevartuple_type)) { |
| return typevartuple_default((typevartupleobject *)typeparam, NULL); |
| } |
| else { |
| PyErr_Format(PyExc_TypeError, "Expected a type param, got %R", typeparam); |
| return 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_evaluate_value(PyObject *self, void *unused) |
| { |
| typealiasobject *ta = (typealiasobject *)self; |
| if (ta->compute_value != NULL) { |
| return Py_NewRef(ta->compute_value); |
| } |
| assert(ta->value != NULL); |
| return constevaluator_alloc(ta->value); |
| } |
| |
| 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}, |
| {"evaluate_value", typealias_evaluate_value, (setter)NULL, NULL, NULL}, |
| {"__module__", typealias_module, (setter)NULL, NULL, NULL}, |
| {0} |
| }; |
| |
| static PyObject * |
| typealias_check_type_params(PyObject *type_params, int *err) { |
| // Can return type_params or NULL without exception set. |
| // Does not change the reference count of type_params, |
| // sets `*err` to 1 when error happens and sets an exception, |
| // otherwise `*err` is set to 0. |
| *err = 0; |
| if (type_params == NULL) { |
| return NULL; |
| } |
| |
| assert(PyTuple_Check(type_params)); |
| Py_ssize_t length = PyTuple_GET_SIZE(type_params); |
| if (!length) { // 0-length tuples are the same as `NULL`. |
| return NULL; |
| } |
| |
| PyThreadState *ts = _PyThreadState_GET(); |
| int default_seen = 0; |
| for (Py_ssize_t index = 0; index < length; index++) { |
| PyObject *type_param = PyTuple_GET_ITEM(type_params, index); |
| PyObject *dflt = get_type_param_default(ts, type_param); |
| if (dflt == NULL) { |
| *err = 1; |
| return NULL; |
| } |
| if (dflt == &_Py_NoDefaultStruct) { |
| if (default_seen) { |
| *err = 1; |
| PyErr_Format(PyExc_TypeError, |
| "non-default type parameter '%R' " |
| "follows default type parameter", |
| type_param); |
| return NULL; |
| } |
| } else { |
| default_seen = 1; |
| Py_DECREF(dflt); |
| } |
| } |
| |
| return type_params; |
| } |
| |
| static PyObject * |
| typelias_convert_type_params(PyObject *type_params) |
| { |
| if ( |
| type_params == NULL |
| || Py_IsNone(type_params) |
| || (PyTuple_Check(type_params) && PyTuple_GET_SIZE(type_params) == 0) |
| ) { |
| return NULL; |
| } |
| else { |
| return type_params; |
| } |
| } |
| |
| 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_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; |
| } |
| |
| int err = 0; |
| PyObject *checked_params = typealias_check_type_params(type_params, &err); |
| if (err) { |
| return NULL; |
| } |
| |
| PyObject *module = caller(); |
| if (module == NULL) { |
| return NULL; |
| } |
| PyObject *ta = (PyObject *)typealias_alloc(name, checked_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 = typelias_convert_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); |
| MAKE_TYPE(constevaluator); |
| #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); |
| Py_CLEAR(interp->cached_objects.constevaluator_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; |
| } |
| } |