From 06c7d2613a3eebac6a09befcad2b2fae6374d262 Mon Sep 17 00:00:00 2001
From: Ali Hamdan <ali.hamdan.dev@gmail.com>
Date: Mon, 17 Jun 2024 15:22:14 +0200
Subject: [PATCH] stubgen: Gracefully handle invalid `Optional` and recognize
 aliases to PEP 604 unions (#17386)

This Fixes 2 issues with invalid `Optional` (inspired by an error
reported in #17197):
- do not crash on empty `Optional`
- treat `Optional` with more than one index as an unknown type instead
of choosing the first type.

It also fixes PEP 604 unions not being recognized as type aliases.
---
 mypy/stubgen.py             |  6 ++++++
 mypy/stubutil.py            |  4 +++-
 test-data/unit/stubgen.test | 30 ++++++++++++++++++++++++++++++
 3 files changed, 39 insertions(+), 1 deletion(-)

diff --git a/mypy/stubgen.py b/mypy/stubgen.py
index 22028694ad6b..8478bd2135e4 100755
--- a/mypy/stubgen.py
+++ b/mypy/stubgen.py
@@ -314,6 +314,8 @@ def visit_index_expr(self, node: IndexExpr) -> str:
                 return " | ".join([item.accept(self) for item in node.index.items])
             return node.index.accept(self)
         if base_fullname == "typing.Optional":
+            if isinstance(node.index, TupleExpr):
+                return self.stubgen.add_name("_typeshed.Incomplete")
             return f"{node.index.accept(self)} | None"
         base = node.base.accept(self)
         index = node.index.accept(self)
@@ -1060,6 +1062,10 @@ def is_alias_expression(self, expr: Expression, top_level: bool = True) -> bool:
                 else:
                     return False
             return all(self.is_alias_expression(i, top_level=False) for i in indices)
+        elif isinstance(expr, OpExpr) and expr.op == "|":
+            return self.is_alias_expression(
+                expr.left, top_level=False
+            ) and self.is_alias_expression(expr.right, top_level=False)
         else:
             return False
 
diff --git a/mypy/stubutil.py b/mypy/stubutil.py
index 8e41d6862531..2f2db0dbbe53 100644
--- a/mypy/stubutil.py
+++ b/mypy/stubutil.py
@@ -257,7 +257,9 @@ def visit_unbound_type(self, t: UnboundType) -> str:
         if fullname == "typing.Union":
             return " | ".join([item.accept(self) for item in t.args])
         if fullname == "typing.Optional":
-            return f"{t.args[0].accept(self)} | None"
+            if len(t.args) == 1:
+                return f"{t.args[0].accept(self)} | None"
+            return self.stubgen.add_name("_typeshed.Incomplete")
         if fullname in TYPING_BUILTIN_REPLACEMENTS:
             s = self.stubgen.add_name(TYPING_BUILTIN_REPLACEMENTS[fullname], require=True)
         if self.known_modules is not None and "." in s:
diff --git a/test-data/unit/stubgen.test b/test-data/unit/stubgen.test
index 916e2e3a8e17..5dcb0706a8cb 100644
--- a/test-data/unit/stubgen.test
+++ b/test-data/unit/stubgen.test
@@ -4366,3 +4366,33 @@ class Foo(Enum):
 class Bar(Enum):
     A = ...
     B = ...
+
+[case testGracefullyHandleInvalidOptionalUsage]
+from typing import Optional
+
+x: Optional  # invalid
+y: Optional[int]  # valid
+z: Optional[int, str]  # invalid
+w: Optional[int | str]  # valid
+r: Optional[type[int | str]]
+
+X = Optional
+Y = Optional[int]
+Z = Optional[int, str]
+W = Optional[int | str]
+R = Optional[type[int | str]]
+
+[out]
+from _typeshed import Incomplete
+from typing import Optional
+
+x: Incomplete
+y: int | None
+z: Incomplete
+w: int | str | None
+r: type[int | str] | None
+X = Optional
+Y = int | None
+Z = Incomplete
+W = int | str | None
+R = type[int | str] | None