Unwrap TypedDict item types before storing (#17640)

ilevkivskyi · hauntsaninja · commit 7d805b364ee8 · 2024-08-24T13:28:52.000-07:00
Fixes #17604 Fixes #17608 Fix is trivial, rectify an obvious omission in my original PR. (cherry picked from commit b56f357)
diff --git a/mypy/semanal_typeddict.py b/mypy/semanal_typeddict.py
@@ -323,7 +323,9 @@ def analyze_typeddict_classdef_fields(
                         return None, [], [], set()  # Need to defer
                     types.append(analyzed)
                     if not has_placeholder(analyzed):
-                        stmt.type = analyzed
+                        stmt.type = (
+                            analyzed.item if isinstance(analyzed, RequiredType) else analyzed
+                        )
                 # ...despite possible minor failures that allow further analysis.
                 if stmt.type is None or hasattr(stmt, "new_syntax") and not stmt.new_syntax:
                     self.fail(TPDICT_CLASS_ERROR, stmt)
diff --git a/test-data/unit/check-typeddict.test b/test-data/unit/check-typeddict.test
@@ -2382,6 +2382,14 @@ class ForceDeferredEval: pass
 [builtins fixtures/dict.pyi]
 [typing fixtures/typing-typeddict.pyi]
 
+[case testTypedDictRequiredUnimportedAny]
+# flags: --disallow-any-unimported
+from typing import NotRequired, TypedDict
+from nonexistent import Foo  # type: ignore[import-not-found]
+class Bar(TypedDict):
+    foo: NotRequired[Foo]  # E: Type of variable becomes "Any" due to an unfollowed import
+[typing fixtures/typing-typeddict.pyi]
+
 -- Required[]
 
 [case testDoesRecognizeRequiredInTypedDictWithClass]
