Allow typing.TypeGuard to be used in type stubs.

For #916.

PiperOrigin-RevId: 373679585

pytype/pyi/definitions.py

@@ -25,10 +25,12 @@
 _TYPING_SETS = ("typing.Intersection", "typing.Optional", "typing.Union")
 
 # Aliases for some typing.X types
-_TUPLE_TYPES = ("tuple", "builtins.tuple", "typing.Tuple")
+_ANNOTATED_TYPES = ("typing.Annotated", "typing_extensions.Annotated")
 _CALLABLE_TYPES = ("typing.Callable", "collections.abc.Callable")
+_CONCATENATE_TYPES = ("typing.Concatenate", "typing_extensions.Concatenate")
 _LITERAL_TYPES = ("typing.Literal", "typing_extensions.Literal")
-_ANNOTATED_TYPES = ("typing.Annotated", "typing_extensions.Annotated")
+_TUPLE_TYPES = ("tuple", "builtins.tuple", "typing.Tuple")
+_TYPEGUARD_TYPES = ("typing.TypeGuard", "typing_extensions.TypeGuard")
 
 
 def _split_definitions(defs: List[Any]):
@@ -432,6 +434,10 @@ def _parameterized_type(self, base_type, parameters):
       return types.pytd_literal(parameters)
     elif self._matches_named_type(base_type, _ANNOTATED_TYPES):
       return types.pytd_annotated(parameters)
+    elif self._matches_named_type(base_type, _TYPEGUARD_TYPES):
+      # We do not yet support PEP 647, User-Defined Type Guards. To avoid
+      # blocking typeshed, convert type guards to plain bools.
+      return pytd.NamedType("bool")
     elif any(isinstance(p, types.Constant) for p in parameters):
       parameters = ", ".join(
           p.repr_str() if isinstance(p, types.Constant) else "_"
@@ -458,8 +464,8 @@ def _parameterized_type(self, base_type, parameters):
           # To avoid blocking typeshed from adopting this PEP, we convert new
           # features to Any.
           if p in self.param_specs or (
-              isinstance(p, pytd.GenericType) and self._matches_full_name(
-                  p, ("typing.Concatenate", "typing_extensions.Concatenate"))):
+              isinstance(p, pytd.GenericType) and
+              self._matches_full_name(p, _CONCATENATE_TYPES)):
             callable_parameters.append(pytd.AnythingType())
           else:
             callable_parameters.append(p)

pytype/pyi/parser_test.py

@@ -2925,5 +2925,34 @@ def f(x: Callable[..., R]) -> Callable[..., R]: ...
     """)
 
 
+class TypeGuardTest(_ParserTestBase):
+
+  def test_typing_extensions(self):
+    self.check("""
+      from typing import List
+
+      from typing_extensions import TypeGuard
+
+      def f(x: List[object]) -> TypeGuard[List[str]]: ...
+    """, """
+      from typing import List
+
+      from typing_extensions import TypeGuard
+
+      def f(x: List[object]) -> bool: ...
+  """)
+
+  def test_typing(self):
+    self.check("""
+      from typing import List, TypeGuard
+
+      def f(x: List[object]) -> TypeGuard[List[str]]: ...
+    """, """
+      from typing import List
+
+      def f(x: List[object]) -> bool: ...
+    """)
+
+
 if __name__ == "__main__":
   unittest.main()