ruff

Author: JaskRendix

docs/add_widgets.py

@@ -68,7 +68,7 @@ def change_background_color(selected_value, color, **kwargs):
     print('Change widget color to', value_tuple[0])  # selected_value format ('Color', surface, color)
     if color == (-1, -1, -1):  # Generate a random color
         color = (randrange(0, 255), randrange(0, 255), randrange(0, 255))
-    widget: 'pygame_menu.widgets.Selector' = kwargs.get('widget')
+    widget: pygame_menu.widgets.Selector = kwargs.get('widget')
     widget.update_font({'selected_color': color})
     widget.get_selection_effect().color = color
 
@@ -252,7 +252,7 @@ def open_link(*args) -> None:
         """
         Opens link.
         """
-        link: 'pygame_menu.widgets.MenuLink' = args[-1]
+        link: pygame_menu.widgets.MenuLink = args[-1]
         link.open()
 
 

pygame_menu/_base.py

@@ -63,7 +63,7 @@ def __repr__(self) -> str:
 
         return sup_repr
 
-    def _update__repr___(self, obj: 'Base') -> None:
+    def _update__repr___(self, obj: Base) -> None:
         """
         Update __repr__ from other Base object.
 
@@ -72,7 +72,7 @@ def _update__repr___(self, obj: 'Base') -> None:
         self._class_id__repr__ = obj._class_id__repr__
         self._id__repr__ = obj._id__repr__
 
-    def set_attribute(self, key: str, value: Any = None) -> 'Base':
+    def set_attribute(self, key: str, value: Any = None) -> Base:
         """
         Set an attribute.
 
@@ -147,7 +147,7 @@ def has_attribute(self, key: str) -> bool:
 
         return self._attributes is not None and key in self._attributes
 
-    def remove_attribute(self, key: str) -> 'Base':
+    def remove_attribute(self, key: str) -> Base:
         """
         Removes the given attribute from the object. Throws ``IndexError`` if
         the given key does not exist.

pygame_menu/_decorator.py

@@ -65,18 +65,18 @@ class Decorator(Base):
         str, tuple[int, int, Union[tuple[Tuple2NumberType, ...], Tuple2NumberType]]]  # centerx, centery, coords
     _cache_last_status: dict[str, tuple[int, int, int, int, int, int]]
     _cache_needs_update: dict[str, bool]
-    _cache_surface: dict[str, Optional['pygame.Surface']]
+    _cache_surface: dict[str, Optional[pygame.Surface]]
     _decor: dict[str, list[tuple[int, str, Any]]]  # type, id, data
     _decor_enabled: dict[str, bool]
     _decor_prev_id: list[str]
-    _obj: Union['pygame_menu.widgets.Widget', 'pygame_menu._scrollarea.ScrollArea', 'pygame_menu.Menu']
+    _obj: Union[pygame_menu.widgets.Widget, pygame_menu._scrollarea.ScrollArea, pygame_menu.Menu]
     _post_enabled: bool
     _prev_enabled: bool
     cache: bool
 
     def __init__(
         self,
-        obj: Union['pygame_menu.widgets.Widget', 'pygame_menu._scrollarea.ScrollArea', 'pygame_menu.Menu'],
+        obj: Union[pygame_menu.widgets.Widget, pygame_menu._scrollarea.ScrollArea, pygame_menu.Menu],
         decorator_id: str = '',
         verbose: bool = True
     ) -> None:
@@ -112,15 +112,15 @@ def __init__(
         self._cache_needs_update = {DECOR_TYPE_PREV: False, DECOR_TYPE_POST: False}
         self._cache_surface = {DECOR_TYPE_PREV: None, DECOR_TYPE_POST: None}
 
-    def __copy__(self) -> 'Decorator':
+    def __copy__(self) -> Decorator:
         """
         Copy method.
 
         :return: Raises copy exception
         """
         raise _DecoratorCopyException('Decorator class cannot be copied')
 
-    def __deepcopy__(self, memodict: dict[int, Any]) -> 'Decorator':
+    def __deepcopy__(self, memodict: dict[int, Any]) -> Decorator:
         """
         Deep-copy method.
 
@@ -180,7 +180,7 @@ def _total_decor(self) -> int:
         """
         return len(self._decor[DECOR_TYPE_PREV]) + len(self._decor[DECOR_TYPE_POST])
 
-    def force_cache_update(self, prev: Optional[bool] = None) -> 'Decorator':
+    def force_cache_update(self, prev: Optional[bool] = None) -> Decorator:
         """
         Forces cache update.
 
@@ -396,7 +396,7 @@ def add_surface(
         self,
         x: NumberType,
         y: NumberType,
-        surface: 'pygame.Surface',
+        surface: pygame.Surface,
         prev: bool = True,
         centered: bool = False,
         **kwargs
@@ -427,7 +427,7 @@ def add_baseimage(
         self,
         x: NumberType,
         y: NumberType,
-        image: 'pygame_menu.BaseImage',
+        image: pygame_menu.BaseImage,
         prev: bool = True,
         centered: bool = False,
         **kwargs
@@ -463,7 +463,7 @@ def add_rect(
         self,
         x: NumberType,
         y: NumberType,
-        rect: 'pygame.Rect',
+        rect: pygame.Rect,
         color: ColorInputType,
         width: int = 0,
         prev: bool = True,
@@ -639,7 +639,7 @@ def add_pixel(
 
     def add_callable(
         self,
-        fun: Union[Callable[['pygame.Surface', Any], Any], CallableNoArgsType],
+        fun: Union[Callable[[pygame.Surface, Any], Any], CallableNoArgsType],
         prev: bool = True,
         pass_args: bool = True
     ) -> str:
@@ -672,7 +672,7 @@ def add_callable(
     def add_textured_polygon(
         self,
         coords: Union[list[Tuple2NumberType], tuple[Tuple2NumberType, ...]],
-        texture: Union['pygame.Surface', 'pygame_menu.BaseImage'],
+        texture: Union[pygame.Surface, pygame_menu.BaseImage],
         tx: int = 0,
         ty: int = 0,
         prev: bool = True,
@@ -811,7 +811,7 @@ def add_vline(
         assert y1 != y2
         return self.add_line((x, y1), (x, y2), color, width, prev, **kwargs)
 
-    def disable(self, decorid: str) -> 'Decorator':
+    def disable(self, decorid: str) -> Decorator:
         """
         Disable a certain decoration from ID. Raises ``IndexError`` if decoration was
         not found.
@@ -825,7 +825,7 @@ def disable(self, decorid: str) -> 'Decorator':
         self.force_cache_update(prev=decorid in self._decor_prev_id)
         return self
 
-    def enable(self, decorid: str) -> 'Decorator':
+    def enable(self, decorid: str) -> Decorator:
         """
         Enable a certain decoration from ID. Raises ``IndexError`` if decoration
         was not found.
@@ -851,7 +851,7 @@ def is_enabled(self, decorid: str) -> bool:
             raise IndexError(f'decoration<"{decorid}"> was not found')
         return self._decor_enabled[decorid]
 
-    def remove(self, decorid: str) -> 'Decorator':
+    def remove(self, decorid: str) -> Decorator:
         """
         Remove a decoration from a given ID. Raises ``IndexError`` if decoration
         was not found.
@@ -873,7 +873,7 @@ def remove(self, decorid: str) -> 'Decorator':
                     return self
         raise IndexError(f'decoration<"{decorid}"> was not found')
 
-    def remove_all(self, prev: Optional[bool] = None) -> 'Decorator':
+    def remove_all(self, prev: Optional[bool] = None) -> Decorator:
         """
         Remove all decorations.
 
@@ -894,7 +894,7 @@ def _draw_assemble_cache(
         self,
         prev: str,
         deco: list[tuple[int, str, Any]],
-        surface: 'pygame.Surface'
+        surface: pygame.Surface
     ) -> None:
         """
         Draw cache, assemble if needed.
@@ -931,7 +931,7 @@ def _draw_assemble_cache(
 
         surface.blit(self._cache_surface[prev], (0, 0))
 
-    def draw_prev(self, surface: 'pygame.Surface') -> 'Decorator':
+    def draw_prev(self, surface: pygame.Surface) -> Decorator:
         """
         Draw prev.
 
@@ -944,7 +944,7 @@ def draw_prev(self, surface: 'pygame.Surface') -> 'Decorator':
             self._draw_assemble_cache(DECOR_TYPE_PREV, self._decor[DECOR_TYPE_PREV], surface)
         return self
 
-    def draw_post(self, surface: 'pygame.Surface') -> 'Decorator':
+    def draw_post(self, surface: pygame.Surface) -> Decorator:
         """
         Draw post.
 
@@ -957,7 +957,7 @@ def draw_post(self, surface: 'pygame.Surface') -> 'Decorator':
             self._draw_assemble_cache(DECOR_TYPE_POST, self._decor[DECOR_TYPE_POST], surface)
         return self
 
-    def _draw(self, deco: list[tuple[int, str, Any]], surface: 'pygame.Surface') -> None:
+    def _draw(self, deco: list[tuple[int, str, Any]], surface: pygame.Surface) -> None:
         """
         Draw.
 
@@ -1055,7 +1055,7 @@ def _draw(self, deco: list[tuple[int, str, Any]], surface: 'pygame.Surface') ->
                 surface.fill(data, rect)
 
             elif dtype == DECORATION_RECT:
-                d_rect: 'pygame.Rect'
+                d_rect: pygame.Rect
                 pos, d_rect, color, width, kwargs = data
                 pos = self._update_pos_list(rect, decoid, pos, **kwargs)[0]
                 d_rect = d_rect.copy()
@@ -1078,7 +1078,7 @@ def _draw(self, deco: list[tuple[int, str, Any]], surface: 'pygame.Surface') ->
 
     def _update_pos_list(
         self,
-        rect: 'pygame.Rect',
+        rect: pygame.Rect,
         decoid: str,
         pos: Union[Tuple2NumberType, tuple[Tuple2NumberType, ...]],  # only (x, y) or ((x1,y1), ...
         use_center_positioning=True