-
Notifications
You must be signed in to change notification settings - Fork 283
Description
I came across this issue while using inheritance to express required keys in a TypedDict, as is recommended by the docs.
It's most easily explained by a minimal example I cooked up. Let's say we have a module foo.py:
from __future__ import annotations
from typing import Optional
from typing_extensions import TypedDict
class Foo(TypedDict):
a: Optional[int]And another module bar.py:
from __future__ import annotations
from typing import get_type_hints
from foo import Foo
class Bar(Foo, total=False):
b: int
print(get_type_hints(Bar))Note that both foo.py and bar.py have adopted postponed evaluation of annotations (PEP 563) by using the __future__ import.
If we execute bar.py, we get the error message NameError: name 'Optional' is not defined.
This is due to the combination of:
get_type_hintsrelies on the MRO to resolve types: https://github.com/python/cpython/blob/3.7/Lib/typing.py#L970TypedDictdoes not preserve the original bases, soFoois not in the MRO forBar:tp_dict = super(_TypedDictMeta, cls).__new__(cls, name, (dict,), ns)
Thus, get_type_hints is unable to resolve the types for annotations that are only imported in foo.py.
I ran this example using typing_extensions 3.7.4.2 (released via #709) and Python 3.7.3, but it seems like this would be an issue using the current main branches of both repositories as well.
I'm wondering what the right approach is to tackling this issue. It is of course solvable by defining Bar in foo.py instead, but it isn't ideal or intuitive to always need to inherit from a TypedDict in the same module.
I was thinking that similarly to __required_keys__ and __optional_keys__, the TypedDict could preserve its original bases in a new dunder attribute, and get_type_hints could work off of that instead of MRO when it is dealing with a TypedDict. I would be willing to contribute the PRs to implement this if the design is acceptable, but am open to other ideas as well.