Added button functionality to Menu

Author: DenverCoder1

README.md

@@ -176,17 +176,19 @@ await pages.start(ctx)
 
 ## Button Component Menus
 
-Here is a button implementation of a basic menu that has a stop button and two reply reactions. Note that the `ButtonMenu` class is used instead of `Menu` in order to make it a `View`.
+Here is a button implementation of a basic menu that has a stop button and two reply reactions.
 
-`ButtonMenu` is a subclass of `Menu` and it therefore has all the same attributes and methods.
+Note that `view=self` is passed with the initial message and `nextcord.ui.button` is used instead of `menus.button`.
 
-Also note that `ButtonMenu.stop()` is a coroutine and needs to be awaited.
+`Menu.disable()` can be used to disable all buttons in the menu.
+
+`Menu.enable()` can be used to enable all buttons in the menu.
 
 ```py
 import nextcord
 from nextcord.ext import menus
 
-class MyButtonMenu(menus.ButtonMenu):
+class MyButtonMenu(menus.Menu):
     async def send_initial_message(self, ctx, channel):
         return await channel.send(f'Hello {ctx.author}', view=self)
 
@@ -200,7 +202,8 @@ class MyButtonMenu(menus.ButtonMenu):
 
     @nextcord.ui.button(emoji="\N{BLACK SQUARE FOR STOP}\ufe0f")
     async def on_stop(self, button, interaction):
-        await self.stop()
+        await self.disable()
+        self.stop()
 ```
 
 Instantiation is the same as above.

nextcord/ext/menus/__init__.py

@@ -281,7 +281,7 @@ def get_buttons(cls):
         return buttons
 
 
-class Menu(metaclass=_MenuMeta):
+class Menu(nextcord.ui.View, metaclass=_MenuMeta):
     r"""An interface that allows handling menus by using reactions as buttons.
 
     Buttons should be marked with the :func:`button` decorator. Please note that
@@ -316,6 +316,7 @@ class Menu(metaclass=_MenuMeta):
     def __init__(self, *, timeout=DEFAULT_TIMEOUT, delete_message_after=False,
                           clear_reactions_after=False, check_embeds=False, message=None):
 
+        super().__init__(timeout=timeout)
         self.timeout = timeout
         self.delete_message_after = delete_message_after
         self.clear_reactions_after = clear_reactions_after
@@ -756,6 +757,34 @@ async def send_initial_message(self, ctx, channel):
         """
         raise NotImplementedError
 
+    async def _set_all_disabled(self, disable: bool):
+        """|coro|
+
+        Enables or disable all :class:`nextcord.ui.Button` components in the menu.
+
+        Parameters
+        ------------
+        disable: :class:`bool`
+            Whether to disable or enable the buttons.
+        """
+        for child in self.children:
+            child.disabled = disable
+        await self.message.edit(view=self)
+
+    async def enable(self):
+        """|coro|
+
+        Enables all :class:`nextcord.ui.Button` components in the menu.
+        """
+        await self._set_all_disabled(False)
+
+    async def disable(self):
+        """|coro|
+
+        Disables all :class:`nextcord.ui.Button` components in the menu.
+        """
+        await self._set_all_disabled(True)
+
     def stop(self):
         """Stops the internal loop."""
         self._running = False
@@ -764,30 +793,6 @@ def stop(self):
         self.__tasks.clear()
 
 
-class ButtonMenu(Menu, nextcord.ui.View):
-    r"""An interface that allows handling menus by using button interaction components.
-
-    Buttons should be marked with the :func:`nextcord.ui.button` decorator. Please note that
-    this expects the methods to have two parameters, the ``button`` and the ``interaction``.
-    The ``button`` is of type :class:`nextcord.ui.Button`.
-    The ``interaction`` is of type :class:`nextcord.Interaction`.
-    """
-    def __init__(self, timeout=DEFAULT_TIMEOUT, *args, **kwargs):
-        Menu.__init__(self, timeout=timeout, *args, **kwargs)
-        nextcord.ui.View.__init__(self, timeout=timeout)
-
-    async def stop(self):
-        """Stops the internal loop and view interactions."""
-        # disable the buttons
-        for child in self.children:
-            child.disabled = True
-        await self.message.edit(view=self)
-        # stop the menu loop
-        Menu.stop(self)
-        # stop view interactions
-        nextcord.ui.View.stop(self)
-
-
 class PageSource:
     """An interface representing a menu page's data source for the actual menu page.