mirror of
https://github.com/python/cpython.git
synced 2025-11-29 14:31:30 +00:00
bpo-46539: Pass status of special typeforms to forward references (GH-30926)
Previously this didn't matter because there weren't any valid code paths that could trigger a type check with a special form, but after the bug fix for `Annotated` wrapping special forms it's now possible to annotate something like `Annotated['ClassVar[int]', (3, 4)]`. This change would also be needed for proposed future changes, such as allowing `ClassVar` and `Final` to nest each other in dataclasses.
This commit is contained in:
parent
6b491b9dc0
commit
ced50051bb
3 changed files with 18 additions and 3 deletions
|
|
@ -2870,6 +2870,20 @@ class ForwardRefTests(BaseTestCase):
|
||||||
self.assertEqual(get_type_hints(foo, globals(), locals()),
|
self.assertEqual(get_type_hints(foo, globals(), locals()),
|
||||||
{'a': Callable[..., T]})
|
{'a': Callable[..., T]})
|
||||||
|
|
||||||
|
def test_special_forms_forward(self):
|
||||||
|
|
||||||
|
class C:
|
||||||
|
a: Annotated['ClassVar[int]', (3, 5)] = 4
|
||||||
|
b: Annotated['Final[int]', "const"] = 4
|
||||||
|
|
||||||
|
class CF:
|
||||||
|
b: List['Final[int]'] = 4
|
||||||
|
|
||||||
|
self.assertEqual(get_type_hints(C, globals())['a'], ClassVar[int])
|
||||||
|
self.assertEqual(get_type_hints(C, globals())['b'], Final[int])
|
||||||
|
with self.assertRaises(TypeError):
|
||||||
|
get_type_hints(CF, globals()),
|
||||||
|
|
||||||
def test_syntax_error(self):
|
def test_syntax_error(self):
|
||||||
|
|
||||||
with self.assertRaises(SyntaxError):
|
with self.assertRaises(SyntaxError):
|
||||||
|
|
|
||||||
|
|
@ -142,12 +142,12 @@ __all__ = [
|
||||||
# legitimate imports of those modules.
|
# legitimate imports of those modules.
|
||||||
|
|
||||||
|
|
||||||
def _type_convert(arg, module=None):
|
def _type_convert(arg, module=None, *, allow_special_forms=False):
|
||||||
"""For converting None to type(None), and strings to ForwardRef."""
|
"""For converting None to type(None), and strings to ForwardRef."""
|
||||||
if arg is None:
|
if arg is None:
|
||||||
return type(None)
|
return type(None)
|
||||||
if isinstance(arg, str):
|
if isinstance(arg, str):
|
||||||
return ForwardRef(arg, module=module)
|
return ForwardRef(arg, module=module, is_class=allow_special_forms)
|
||||||
return arg
|
return arg
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -169,7 +169,7 @@ def _type_check(arg, msg, is_argument=True, module=None, *, allow_special_forms=
|
||||||
if is_argument:
|
if is_argument:
|
||||||
invalid_generic_forms += (Final,)
|
invalid_generic_forms += (Final,)
|
||||||
|
|
||||||
arg = _type_convert(arg, module=module)
|
arg = _type_convert(arg, module=module, allow_special_forms=allow_special_forms)
|
||||||
if (isinstance(arg, _GenericAlias) and
|
if (isinstance(arg, _GenericAlias) and
|
||||||
arg.__origin__ in invalid_generic_forms):
|
arg.__origin__ in invalid_generic_forms):
|
||||||
raise TypeError(f"{arg} is not valid as type argument")
|
raise TypeError(f"{arg} is not valid as type argument")
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1 @@
|
||||||
|
In :func:`typing.get_type_hints`, support evaluating stringified ``ClassVar`` and ``Final`` annotations inside ``Annotated``. Patch by Gregory Beauregard.
|
||||||
Loading…
Add table
Add a link
Reference in a new issue