From: Ken Jin <28750310+Fidget-Spinner@users.noreply.github.com> Date: Mon, 12 Apr 2021 17:23:12 +0000 (+0800) Subject: bpo-42904: Fix get_type_hints for class local namespaces (GH-24201) X-Git-Tag: v3.10.0b1~314 X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=852150ddfe68bc2696fc880175aeb855a0c16ae6;p=thirdparty%2FPython%2Fcpython.git bpo-42904: Fix get_type_hints for class local namespaces (GH-24201) --- diff --git a/Lib/test/test_typing.py b/Lib/test/test_typing.py index 82c517a4e600..062163ccf4b6 100644 --- a/Lib/test/test_typing.py +++ b/Lib/test/test_typing.py @@ -2874,7 +2874,7 @@ class GetTypeHintTests(BaseTestCase): {'x': int, 'y': int}) self.assertEqual(gth(mod_generics_cache.B), {'my_inner_a1': mod_generics_cache.B.A, - 'my_inner_a2': mod_generics_cache.A, + 'my_inner_a2': mod_generics_cache.B.A, 'my_outer_a': mod_generics_cache.A}) def test_respect_no_type_check(self): @@ -3010,6 +3010,13 @@ class GetTypeHintTests(BaseTestCase): {'other': MySet[T], 'return': MySet[T]} ) + def test_get_type_hints_classes(self): + class Foo: + y = str + x: y + # This previously raised an error under PEP 563. + self.assertEqual(get_type_hints(Foo), {'x': str}) + class GetUtilitiesTestCase(TestCase): def test_get_origin(self): diff --git a/Lib/typing.py b/Lib/typing.py index a24c01f0e3b9..583438e4740a 100644 --- a/Lib/typing.py +++ b/Lib/typing.py @@ -1632,12 +1632,13 @@ def get_type_hints(obj, globalns=None, localns=None, include_extras=False): else: base_globals = globalns ann = base.__dict__.get('__annotations__', {}) + base_locals = dict(vars(base)) if localns is None else localns for name, value in ann.items(): if value is None: value = type(None) if isinstance(value, str): value = ForwardRef(value, is_argument=False) - value = _eval_type(value, base_globals, localns) + value = _eval_type(value, base_globals, base_locals) hints[name] = value return hints if include_extras else {k: _strip_annotations(t) for k, t in hints.items()} diff --git a/Misc/NEWS.d/next/Library/2021-01-12-23-17-02.bpo-42904.-4qkTD.rst b/Misc/NEWS.d/next/Library/2021-01-12-23-17-02.bpo-42904.-4qkTD.rst new file mode 100644 index 000000000000..ef4b4e56fbeb --- /dev/null +++ b/Misc/NEWS.d/next/Library/2021-01-12-23-17-02.bpo-42904.-4qkTD.rst @@ -0,0 +1,2 @@ +:func:`typing.get_type_hints` now checks the local namespace of a class when +evaluating :pep:`563` annotations inside said class.