Stowage/cpython
2
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-10-19 16:03:42 +00:00
Code Issues Projects Releases Packages Wiki Activity

gh-126085: Add tp_iter to TypeAliasType to allow star unpacking (#127981)

Browse source
This commit is contained in:
Tomas R. 2025-03-04 20:34:59 +01:00 • committed by GitHub
parent d8a1cf469e
commit e091520fdb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 37 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

15
Objects/typevarobject.c
View file

@ -1,4 +1,4 @@
// TypeVar, TypeVarTuple, and ParamSpec
// TypeVar, TypeVarTuple, ParamSpec, and TypeAlias
#include "Python.h"
#include "pycore_object.h" // _PyObject_GC_TRACK/UNTRACK, PyAnnotateFormat
#include "pycore_typevarobject.h"
@ -394,7 +394,7 @@ caller(void)
}
static PyObject *
typevartuple_unpack(PyObject *tvt)
unpack(PyObject *self)
{
PyObject *typing = PyImport_ImportModule("typing");
if (typing == NULL) {
@ -405,7 +405,7 @@ typevartuple_unpack(PyObject *tvt)
Py_DECREF(typing);
return NULL;
}
PyObject *unpacked = PyObject_GetItem(unpack, tvt);
PyObject *unpacked = PyObject_GetItem(unpack, self);
Py_DECREF(typing);
Py_DECREF(unpack);
return unpacked;
@ -440,7 +440,7 @@ unpack_typevartuples(PyObject *params)
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);
PyObject *unpacked = unpack(param);
if (unpacked == NULL) {
Py_DECREF(new_params);
return NULL;
@ -1524,9 +1524,9 @@ typevartuple_dealloc(PyObject *self)
}
static PyObject *
typevartuple_iter(PyObject *self)
unpack_iter(PyObject *self)
{
PyObject *unpacked = typevartuple_unpack(self);
PyObject *unpacked = unpack(self);
if (unpacked == NULL) {
return NULL;
}
@ -1782,7 +1782,7 @@ PyType_Slot typevartuple_slots[] = {
{Py_tp_methods, typevartuple_methods},
{Py_tp_getset, typevartuple_getset},
{Py_tp_new, typevartuple},
{Py_tp_iter, typevartuple_iter},
{Py_tp_iter, unpack_iter},
{Py_tp_repr, typevartuple_repr},
{Py_tp_dealloc, typevartuple_dealloc},
{Py_tp_alloc, PyType_GenericAlloc},
@ -2158,6 +2158,7 @@ PyTypeObject _PyTypeAlias_Type = {
.tp_dealloc = typealias_dealloc,
.tp_new = typealias_new,
.tp_free = PyObject_GC_Del,
.tp_iter = unpack_iter,
.tp_traverse = typealias_traverse,
.tp_clear = typealias_clear,
.tp_repr = typealias_repr,