[Python-checkins] gh-104600: Make function.__type_params__ writable (#104601)
JelleZijlstra
webhook-mailer at python.org
Thu May 18 19:45:44 EDT 2023
https://github.com/python/cpython/commit/3fadd7d5857842fc5cddd4c496b73161b0bcb421
commit: 3fadd7d5857842fc5cddd4c496b73161b0bcb421
branch: main
author: Jelle Zijlstra <jelle.zijlstra at gmail.com>
committer: JelleZijlstra <jelle.zijlstra at gmail.com>
date: 2023-05-18T16:45:37-07:00
summary:
gh-104600: Make function.__type_params__ writable (#104601)
files:
A Misc/NEWS.d/next/Library/2023-05-17-21-01-48.gh-issue-104600.E6CK35.rst
M Lib/functools.py
M Lib/test/test_funcattrs.py
M Lib/test/test_functools.py
M Lib/test/test_type_params.py
M Objects/funcobject.c
diff --git a/Lib/functools.py b/Lib/functools.py
index aaf4291150fb..72b2103e7a55 100644
--- a/Lib/functools.py
+++ b/Lib/functools.py
@@ -30,7 +30,7 @@
# wrapper functions that can handle naive introspection
WRAPPER_ASSIGNMENTS = ('__module__', '__name__', '__qualname__', '__doc__',
- '__annotations__')
+ '__annotations__', '__type_params__')
WRAPPER_UPDATES = ('__dict__',)
def update_wrapper(wrapper,
wrapped,
diff --git a/Lib/test/test_funcattrs.py b/Lib/test/test_funcattrs.py
index 77977d0ae966..e08d72877d8a 100644
--- a/Lib/test/test_funcattrs.py
+++ b/Lib/test/test_funcattrs.py
@@ -1,5 +1,6 @@
import textwrap
import types
+import typing
import unittest
@@ -190,6 +191,20 @@ def test___qualname__(self):
# __qualname__ must be a string
self.cannot_set_attr(self.b, '__qualname__', 7, TypeError)
+ def test___type_params__(self):
+ def generic[T](): pass
+ def not_generic(): pass
+ T, = generic.__type_params__
+ self.assertIsInstance(T, typing.TypeVar)
+ self.assertEqual(generic.__type_params__, (T,))
+ self.assertEqual(not_generic.__type_params__, ())
+ with self.assertRaises(TypeError):
+ del not_generic.__type_params__
+ with self.assertRaises(TypeError):
+ not_generic.__type_params__ = 42
+ not_generic.__type_params__ = (T,)
+ self.assertEqual(not_generic.__type_params__, (T,))
+
def test___code__(self):
num_one, num_two = 7, 8
def a(): pass
diff --git a/Lib/test/test_functools.py b/Lib/test/test_functools.py
index af286052a7d5..d668fa4c3adf 100644
--- a/Lib/test/test_functools.py
+++ b/Lib/test/test_functools.py
@@ -617,7 +617,7 @@ def check_wrapper(self, wrapper, wrapped,
def _default_update(self):
- def f(a:'This is a new annotation'):
+ def f[T](a:'This is a new annotation'):
"""This is a test"""
pass
f.attr = 'This is also a test'
@@ -630,12 +630,14 @@ def wrapper(b:'This is the prior annotation'):
def test_default_update(self):
wrapper, f = self._default_update()
self.check_wrapper(wrapper, f)
+ T, = f.__type_params__
self.assertIs(wrapper.__wrapped__, f)
self.assertEqual(wrapper.__name__, 'f')
self.assertEqual(wrapper.__qualname__, f.__qualname__)
self.assertEqual(wrapper.attr, 'This is also a test')
self.assertEqual(wrapper.__annotations__['a'], 'This is a new annotation')
self.assertNotIn('b', wrapper.__annotations__)
+ self.assertEqual(wrapper.__type_params__, (T,))
@unittest.skipIf(sys.flags.optimize >= 2,
"Docstrings are omitted with -O2 and above")
diff --git a/Lib/test/test_type_params.py b/Lib/test/test_type_params.py
index 96bd1fa0bab9..466e3bd43a68 100644
--- a/Lib/test/test_type_params.py
+++ b/Lib/test/test_type_params.py
@@ -843,5 +843,5 @@ def func[A]():
func.__type_params__ = ()
"""
- with self.assertRaisesRegex(AttributeError, "attribute '__type_params__' of 'function' objects is not writable"):
- run_code(code)
+ ns = run_code(code)
+ self.assertEqual(ns["func"].__type_params__, ())
diff --git a/Misc/NEWS.d/next/Library/2023-05-17-21-01-48.gh-issue-104600.E6CK35.rst b/Misc/NEWS.d/next/Library/2023-05-17-21-01-48.gh-issue-104600.E6CK35.rst
new file mode 100644
index 000000000000..64f81e140d96
--- /dev/null
+++ b/Misc/NEWS.d/next/Library/2023-05-17-21-01-48.gh-issue-104600.E6CK35.rst
@@ -0,0 +1,2 @@
+:func:`functools.update_wrapper` now sets the ``__type_params__`` attribute
+(added by :pep:`695`).
diff --git a/Objects/funcobject.c b/Objects/funcobject.c
index 69898bf722d6..753038600aa8 100644
--- a/Objects/funcobject.c
+++ b/Objects/funcobject.c
@@ -665,6 +665,20 @@ func_get_type_params(PyFunctionObject *op, void *Py_UNUSED(ignored))
return Py_NewRef(op->func_typeparams);
}
+static int
+func_set_type_params(PyFunctionObject *op, PyObject *value, void *Py_UNUSED(ignored))
+{
+ /* Not legal to del f.__type_params__ or to set it to anything
+ * other than a tuple object. */
+ if (value == NULL || !PyTuple_Check(value)) {
+ PyErr_SetString(PyExc_TypeError,
+ "__type_params__ must be set to a tuple");
+ return -1;
+ }
+ Py_XSETREF(op->func_typeparams, Py_NewRef(value));
+ return 0;
+}
+
PyObject *
_Py_set_function_type_params(PyThreadState *Py_UNUSED(ignored), PyObject *func,
PyObject *type_params)
@@ -687,7 +701,8 @@ static PyGetSetDef func_getsetlist[] = {
{"__dict__", PyObject_GenericGetDict, PyObject_GenericSetDict},
{"__name__", (getter)func_get_name, (setter)func_set_name},
{"__qualname__", (getter)func_get_qualname, (setter)func_set_qualname},
- {"__type_params__", (getter)func_get_type_params, NULL},
+ {"__type_params__", (getter)func_get_type_params,
+ (setter)func_set_type_params},
{NULL} /* Sentinel */
};
More information about the Python-checkins
mailing list