Create events type and update plt.connect and mpl_connect

Author: ZPyrolink

lib/matplotlib/backend_bases.pyi

@@ -21,7 +21,19 @@ from matplotlib.transforms import Bbox, BboxBase, Transform, TransformedPath
 from collections.abc import Callable, Iterable, Sequence
 from typing import Any, IO, Literal, NamedTuple, TypeVar, overload
 from numpy.typing import ArrayLike
-from .typing import ColorType, LineStyleType, CapStyleType, JoinStyleType
+from .typing import (
+    ColorType,
+    LineStyleType,
+    CapStyleType,
+    JoinStyleType,
+    MouseEventType,
+    KeyEventType,
+    DrawEventType,
+    PickEventType,
+    ResizeEventType,
+    CloseEventType,
+    EventType
+)
 
 def register_backend(
     format: str, backend: str | type[FigureCanvasBase], description: str | None = ...
@@ -354,37 +366,31 @@ class FigureCanvasBase:
     @overload
     def mpl_connect(
         self,
-        s: Literal[
-            "button_press_event",
-            "motion_notify_event",
-            "scroll_event",
-            "figure_enter_event",
-            "figure_leave_event",
-            "axes_enter_event",
-            "axes_leave_event",
-            "button_release_event",
-        ],
+        s: MouseEventType,
         func: Callable[[MouseEvent], Any],
     ) -> int: ...
 
     @overload
     def mpl_connect(
         self,
-        s: Literal["key_press_event", "key_release_event"],
+        s: KeyEventType,
         func: Callable[[KeyEvent], Any],
     ) -> int: ...
 
     @overload
-    def mpl_connect(self, s: Literal["pick_event"], func: Callable[[PickEvent], Any]) -> int: ...
+    def mpl_connect(self, s: PickEventType, func: Callable[[PickEvent], Any]) -> int: ...
+
+    @overload
+    def mpl_connect(self, s: ResizeEventType, func: Callable[[ResizeEvent], Any]) -> int: ...
 
     @overload
-    def mpl_connect(self, s: Literal["resize_event"], func: Callable[[ResizeEvent], Any]) -> int: ...
+    def mpl_connect(self, s: CloseEventType, func: Callable[[CloseEvent], Any]) -> int: ...
 
     @overload
-    def mpl_connect(self, s: Literal["close_event"], func: Callable[[CloseEvent], Any]) -> int: ...
+    def mpl_connect(self, s: DrawEventType, func: Callable[[DrawEvent], Any]) -> int: ...
 
     @overload
-    def mpl_connect(self, s: str, func: Callable[[Event], Any]) -> int: ...
+    def mpl_connect(self, s: EventType, func: Callable[[Event], Any]) -> int: ...
     def mpl_disconnect(self, cid: int) -> None: ...
     def new_timer(
         self,

lib/matplotlib/pyplot.py

@@ -131,6 +131,7 @@
         HashableList,
         LineStyleType,
         MarkerType,
+        EventType
     )
     from matplotlib.widgets import SubplotTool
 
@@ -1176,7 +1177,7 @@ def get_current_fig_manager() -> FigureManagerBase | None:
 
 
 @_copy_docstring_and_deprecators(FigureCanvasBase.mpl_connect)
-def connect(s: str, func: Callable[[Event], Any]) -> int:
+def connect(s: EventType, func: Callable[[Event], Any]) -> int:
     return gcf().canvas.mpl_connect(s, func)
 
 

lib/matplotlib/typing.py

@@ -107,3 +107,33 @@
 _HT = TypeVar("_HT", bound=Hashable)
 HashableList: TypeAlias = list[_HT | "HashableList[_HT]"]
 """A nested list of Hashable values."""
+
+MouseEventType: TypeAlias = Literal[
+    "button_press_event",
+    "button_release_event",
+    "motion_notify_event",
+    "scroll_event",
+    "figure_enter_event",
+    "figure_leave_event",
+    "axes_enter_event",
+    "axes_leave_event",
+]
+
+KeyEventType: TypeAlias = Literal[
+    "key_press_event",
+    "key_release_event"
+]
+
+DrawEventType: TypeAlias = Literal["draw_event"]
+PickEventType: TypeAlias = Literal["pick_event"]
+ResizeEventType: TypeAlias = Literal["resize_event"]
+CloseEventType: TypeAlias = Literal["close_event"]
+
+EventType: TypeAlias = Literal[
+    MouseEventType,
+    KeyEventType,
+    DrawEventType,
+    PickEventType,
+    ResizeEventType,
+    CloseEventType,
+]