Extract ReadEngine and WriteEngine types for Excel

Author: davetapley

pandas-stubs/io/excel/_base.pyi

@@ -16,7 +16,10 @@ from odf.opendocument import OpenDocument
 from openpyxl.workbook.workbook import Workbook
 from pandas.core.frame import DataFrame
 import pyxlsb.workbook
-from typing_extensions import Self
+from typing_extensions import (
+    Self,
+    TypeAlias,
+)
 from xlrd.book import Book
 
 from pandas._libs.lib import NoDefault
@@ -32,6 +35,8 @@ from pandas._typing import (
     WriteExcelBuffer,
 )
 
+ReadEngine: TypeAlias  = Literal["xlrd", "openpyxl", "odf", "pyxlsb"]
+
 @overload
 def read_excel(
     io: (
@@ -50,7 +55,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 +97,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 +140,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 +182,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 +207,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 +225,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