Extract ReadEngine and WriteEngine types for Excel

Author: davetapley

pandas-stubs/io/excel/_base.pyi

@@ -9,6 +9,7 @@ from types import TracebackType
 from typing import (
     Any,
     Literal,
+    TypeAlias,
     overload,
 )
 
@@ -32,6 +33,8 @@ from pandas._typing import (
     WriteExcelBuffer,
 )
 
+ReadEngine: TypeAlias  = Literal["xlrd", "openpyxl", "odf", "pyxlsb"]
+
 @overload
 def read_excel(
     io: (
@@ -50,7 +53,7 @@ def read_excel(
     index_col: int | Sequence[int] | None = ...,
     usecols: str | UsecolsArgType = ...,
     dtype: str | Dtype | Mapping[str, str | Dtype] | None = ...,
-    engine: Literal["xlrd", "openpyxl", "odf", "pyxlsb"] | None = ...,
+    engine: ReadEngine | None = ...,
     converters: Mapping[int | str, Callable[[object], object]] | None = ...,
     true_values: Iterable[Hashable] | None = ...,
     false_values: Iterable[Hashable] | None = ...,
@@ -92,7 +95,7 @@ def read_excel(
     index_col: int | Sequence[int] | None = ...,
     usecols: str | UsecolsArgType = ...,
     dtype: str | Dtype | Mapping[str, str | Dtype] | None = ...,
-    engine: Literal["xlrd", "openpyxl", "odf", "pyxlsb"] | None = ...,
+    engine: ReadEngine | None = ...,
     converters: Mapping[int | str, Callable[[object], object]] | None = ...,
     true_values: Iterable[Hashable] | None = ...,
     false_values: Iterable[Hashable] | None = ...,
@@ -135,7 +138,7 @@ def read_excel(  # type: ignore[misc]
     index_col: int | Sequence[int] | None = ...,
     usecols: str | UsecolsArgType = ...,
     dtype: str | Dtype | Mapping[str, str | Dtype] | None = ...,
-    engine: Literal["xlrd", "openpyxl", "odf", "pyxlsb"] | None = ...,
+    engine: ReadEngine | None = ...,
     converters: Mapping[int | str, Callable[[object], object]] | None = ...,
     true_values: Iterable[Hashable] | None = ...,
     false_values: Iterable[Hashable] | None = ...,
@@ -177,7 +180,7 @@ def read_excel(
     index_col: int | Sequence[int] | None = ...,
     usecols: str | UsecolsArgType = ...,
     dtype: str | Dtype | Mapping[str, str | Dtype] | None = ...,
-    engine: Literal["xlrd", "openpyxl", "odf", "pyxlsb"] | None = ...,
+    engine: ReadEngine | None = ...,
     converters: Mapping[int | str, Callable[[object], object]] | None = ...,
     true_values: Iterable[Hashable] | None = ...,
     false_values: Iterable[Hashable] | None = ...,
@@ -202,11 +205,14 @@ def read_excel(
     dtype_backend: DtypeBackend | NoDefault = ...,
 ) -> DataFrame: ...
 
+
+WriteEngine: TypeAlias  = Literal["openpyxl", "odf", "xlsxwriter"]
+
 class ExcelWriter:
     def __init__(
         self,
         path: FilePath | WriteExcelBuffer | ExcelWriter,
-        engine: Literal["auto", "openpyxl", "odf", "xlsxwriter"] | None = ...,
+        engine: WriteEngine | Literal["auto"] | None = ...,
         date_format: str | None = ...,
         datetime_format: str | None = ...,
         mode: Literal["w", "a"] = ...,
@@ -217,7 +223,7 @@ class ExcelWriter:
     @property
     def supported_extensions(self) -> tuple[str, ...]: ...
     @property
-    def engine(self) -> Literal["openpyxl", "odf", "xlsxwriter"]: ...
+    def engine(self) -> WriteEngine: ...
     @property
     def sheets(self) -> dict[str, Any]: ...
     @property