[stubgen] Preserve PEP 604 Unions in generated pyi files

mypy/stubgen.py

@@ -134,6 +134,7 @@
  TypeList,
  TypeStrVisitor,
  UnboundType,
+ UnionType,
  get_proper_type,
 )
 from mypy.visitor import NodeVisitor
@@ -326,6 +327,9 @@ def visit_none_type(self, t: NoneType) -> str:
  def visit_type_list(self, t: TypeList) -> str:
  return f"[{self.list_str(t.items)}]"
 
+ def visit_union_type(self, t: UnionType) -> str:
+ return " | ".join([item.accept(self) for item in t.items])
+
  def args_str(self, args: Iterable[Type]) -> str:
  """Convert an array of arguments to strings and join the results with commas.
 

test-data/unit/stubgen.test

@@ -2783,3 +2783,13 @@ T = TypeVar("T", bound=str | None)
 from typing import TypeVar
 
 T = TypeVar('T', bound=str | None)
+
+
+[case testPEP604UnionType]
+a: str | int
+
+def f(x: str | None) -> None: ...
+[out]
+a: str | int
+
+def f(x: str | None) -> None: ...