Menus: Add register_close_callback decorator

Signed-off-by: BackRaw <backraw@gmx.de>

Author: BackRaw

addons/source-python/packages/source-python/menus/base.py

@@ -41,7 +41,7 @@ class _BaseMenu(AutoUnload, list):
 
     _instances = {}
 
-    def __init__(self, data=None, select_callback=None, build_callback=None):
+    def __init__(self, data=None, select_callback=None, build_callback=None, close_callback=None):
         """Initialize the menu.
 
         :param iterable|None data: Data that should be added to the menu.
@@ -56,6 +56,13 @@ def __init__(self, data=None, select_callback=None, build_callback=None):
         :param callable|None build_callback: A function that gets called
             before a menu is displayed.
 
+            The callback will receive 2 parameters:
+                1. The instance of this menu.
+                2. The index of the player who will receive this menu.
+
+        :param callable|None close_callback: A function that gets called
+            when a menu is closed by a player.
+
             The callback will receive 2 parameters:
                 1. The instance of this menu.
                 2. The index of the player who will receive this menu.
@@ -64,6 +71,7 @@ def __init__(self, data=None, select_callback=None, build_callback=None):
 
         self.select_callback = select_callback
         self.build_callback = build_callback
+        self.close_callback = close_callback
         self._player_pages = defaultdict(_PlayerPage)
         self._instances[id(self)] = self
 
@@ -113,6 +121,15 @@ def _select(self, player_index, choice_index):
         if self.select_callback is not None:
             return self.select_callback(self, player_index, choice_index)
 
+    def _select_close(self, player_index):
+        """Handle the close menu selection.
+
+        :param int player_index: The index of the player who made the
+            selection.
+        """
+        if self.close_callback is not None:
+            return self.close_callback(self, player_index)
+
     def send(self, *ply_indexes, **tokens):
         """Send the menu to the given player indexes.
 
@@ -260,6 +277,21 @@ def register_build_callback(self, callback):
         self.build_callback = callback
         return callback
 
+    def register_close_callback(self, callback):
+        """Register a close callback for the menu.
+
+        Can and should be used as a decorator.
+
+        :param callable callback: A function that gets called
+            when a menu is closed by a player.
+
+            The callback will receive 2 parameters:
+                1. The instance of this menu.
+                2. The index of the player who will receive this menu.
+        """
+        self.close_callback = callback
+        return callback
+
 
 class _MenuData(object):
     """Base class for menu data.

addons/source-python/packages/source-python/menus/esc.py

@@ -49,7 +49,7 @@ class SimpleESCMenu(_BaseMenu):
     """This class creates basic ESC menus."""
 
     def __init__(
-            self, data=None, select_callback=None, build_callback=None,
+            self, data=None, select_callback=None, build_callback=None, close_callback=None,
             description=None, title=None, title_color=WHITE):
         """Initialize the object.
 
@@ -64,7 +64,7 @@ def __init__(
             menu.
         :param Color title_color: The color of the title.
         """
-        super().__init__(data, select_callback, build_callback)
+        super().__init__(data, select_callback, build_callback, close_callback)
         self.description = description
         self.title = title
         self.title_color = title_color
@@ -109,7 +109,7 @@ def _get_menu_data(self, player_index):
     def _select(self, player_index, choice_index):
         """See :meth:`menus.base._BaseMenu._select`."""
         if choice_index == 0:
-            return None
+            return self._select_close(player_index)
 
         option = self._player_pages[player_index].options[choice_index]
         if not option.selectable:
@@ -165,7 +165,7 @@ class PagedESCMenu(SimpleESCMenu, _PagedMenuBase):
     """
 
     def __init__(
-            self, data=None, select_callback=None, build_callback=None,
+            self, data=None, select_callback=None, build_callback=None, close_callback=None,
             description=None, title=None, title_color=WHITE, fill=True,
             parent_menu=None):
         """Initialize the object.
@@ -184,7 +184,7 @@ def __init__(
             hitting 'Back' on the first page.
         """
         super().__init__(
-            data, select_callback, build_callback,
+            data, select_callback, build_callback, close_callback,
             description, title, title_color)
         self.fill = fill
         self.parent_menu = parent_menu
@@ -280,10 +280,9 @@ def _get_menu_data(self, player_index):
 
     def _select(self, player_index, choice_index):
         """See :meth:`menus.base._BaseMenu._select`."""
-        # Do nothing if the menu is being closed
         if choice_index == 0:
             del self._player_pages[player_index]
-            return None
+            return self._select_close(player_index)
 
         # Get the player's current page
         page_index = self.get_player_page(player_index)

addons/source-python/packages/source-python/menus/radio.py

@@ -106,7 +106,7 @@ def _slots_to_bin(slots):
     def _select(self, player_index, choice_index):
         """See :meth:`menus.base._BaseMenu._select`."""
         if choice_index == BUTTON_CLOSE:
-            return None
+            return self._select_close(player_index)
 
         return super()._select(
             player_index,
@@ -139,7 +139,7 @@ class PagedRadioMenu(SimpleRadioMenu, _PagedMenuBase):
 
     def __init__(
             self, data=None, select_callback=None,
-            build_callback=None, description=None,
+            build_callback=None, close_callback=None, description=None,
             title=None, top_separator='-' * 30, bottom_separator='-' * 30,
             fill=True, parent_menu=None):
         """Initialize the object.
@@ -162,7 +162,7 @@ def __init__(
         :param _BaseMenu parent_menu: A menu that will be displayed when
             hitting 'Back' on the first page.
         """
-        super().__init__(data, select_callback, build_callback)
+        super().__init__(data, select_callback, build_callback, close_callback)
 
         self.title = title
         self.description = description
@@ -290,10 +290,9 @@ def _get_menu_data(self, player_index):
 
     def _select(self, player_index, choice_index):
         """See :meth:`menus.base._BaseMenu._select`."""
-        # Do nothing if the menu is being closed
         if choice_index == BUTTON_CLOSE:
             del self._player_pages[player_index]
-            return None
+            return self._select_close(player_index)
 
         # Get the player's current page
         page = self._player_pages[player_index]
@@ -321,7 +320,7 @@ class ListRadioMenu(PagedRadioMenu):
     Navigation options are added automatically."""
 
     def __init__(
-            self, data=None, select_callback=None, build_callback=None,
+            self, data=None, select_callback=None, build_callback=None, close_callback=None,
             description=None, title=None, top_separator='-' * 30,
             bottom_separator='-' * 30, fill=True, parent_menu=None,
             items_per_page=10):
@@ -341,7 +340,7 @@ def __init__(
         :param int items_per_page: Number of options that should be displayed
             on a single page.
         """
-        super().__init__(data, select_callback, build_callback, description,
+        super().__init__(data, select_callback, build_callback, close_callback, description,
             title, top_separator, bottom_separator, fill, parent_menu)
         self.items_per_page = items_per_page