Skip to content

Commit

Permalink
bpo-44468: Never skip base classes in typing.get_type_hints(), even…
Browse files Browse the repository at this point in the history
… with invalid `.__module__`. (GH-26862)
  • Loading branch information
will-ca committed Jun 26, 2021
1 parent 521ba88 commit 7569c0f
Show file tree
Hide file tree
Showing 3 changed files with 21 additions and 11 deletions.
25 changes: 18 additions & 7 deletions Lib/test/test_typing.py
Expand Up @@ -2277,13 +2277,6 @@ def test_no_isinstance(self):
with self.assertRaises(TypeError):
issubclass(int, ClassVar)

def test_bad_module(self):
# bpo-41515
class BadModule:
pass
BadModule.__module__ = 'bad' # Something not in sys.modules
self.assertEqual(get_type_hints(BadModule), {})

class FinalTests(BaseTestCase):

def test_basics(self):
Expand Down Expand Up @@ -3033,6 +3026,24 @@ class Foo:
# This previously raised an error under PEP 563.
self.assertEqual(get_type_hints(Foo), {'x': str})

def test_get_type_hints_bad_module(self):
# bpo-41515
class BadModule:
pass
BadModule.__module__ = 'bad' # Something not in sys.modules
self.assertNotIn('bad', sys.modules)
self.assertEqual(get_type_hints(BadModule), {})

def test_get_type_hints_annotated_bad_module(self):
# See https://bugs.python.org/issue44468
class BadBase:
foo: tuple
class BadType(BadBase):
bar: list
BadType.__module__ = BadBase.__module__ = 'bad'
self.assertNotIn('bad', sys.modules)
self.assertEqual(get_type_hints(BadType), {'foo': tuple, 'bar': list})


class GetUtilitiesTestCase(TestCase):
def test_get_origin(self):
Expand Down
5 changes: 1 addition & 4 deletions Lib/typing.py
Expand Up @@ -1701,10 +1701,7 @@ def get_type_hints(obj, globalns=None, localns=None, include_extras=False):
hints = {}
for base in reversed(obj.__mro__):
if globalns is None:
try:
base_globals = sys.modules[base.__module__].__dict__
except KeyError:
continue
base_globals = getattr(sys.modules.get(base.__module__, None), '__dict__', {})
else:
base_globals = globalns
ann = base.__dict__.get('__annotations__', {})
Expand Down
@@ -0,0 +1,2 @@
:func:`typing.get_type_hints` now finds annotations in classes and base classes
with unexpected ``__module__``. Previously, it skipped those MRO elements.

0 comments on commit 7569c0f

Please sign in to comment.