feat!: Add view.parent and store view.message on receiving Interaction for a component (#2036)

Signed-off-by: Om <[email protected]>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: OmLanke

CHANGELOG.md

@@ -46,6 +46,9 @@ These changes are available on the `master` branch, but have not yet been releas
   ([#2030](https://github.com/Pycord-Development/pycord/pull/2030))
 - Added `Interaction.respond` and `Interaction.edit` as shortcut responses.
   ([#2026](https://github.com/Pycord-Development/pycord/pull/2026))
+- Added `view.parent` which is set when the view was sent by
+  `interaction.response.send_message`.
+  ([#2036](https://github.com/Pycord-Development/pycord/pull/2036))
 
 ### Changed
 
@@ -57,11 +60,16 @@ These changes are available on the `master` branch, but have not yet been releas
   `GroupChannel`. ([#2025](https://github.com/Pycord-Development/pycord/pull/2025))
 - `DMChannel.recipients` can now be `None`
   ([#2025](https://github.com/Pycord-Development/pycord/pull/2025))
+- Store `view.message` on receiving Interaction for a component.
+  ([#2036](https://github.com/Pycord-Development/pycord/pull/2036))
 
 ### Removed
 
 - Removed `@client.once()` in favour of `@client.listen(once=True)`.
   ([#1957](https://github.com/Pycord-Development/pycord/pull/1957))
+- Removed `view.message` being set when the view was sent by
+  `interaction.response.send_message`.
+  ([#2036](https://github.com/Pycord-Development/pycord/pull/2036))
 
 ### Fixed
 

discord/interactions.py

@@ -913,7 +913,7 @@ async def send_message(
             if ephemeral and view.timeout is None:
                 view.timeout = 15 * 60.0
 
-            view.message = await self._parent.original_response()
+            view.parent = self._parent
             self._parent._state.store_view(view)
 
         self._responded = True

discord/ui/view.py

@@ -141,6 +141,9 @@ class View:
     message: Optional[:class:`.Message`]
         The message that this view is attached to.
         If ``None`` then the view has not been sent with a message.
+    parent: Optional[:class:`.Interaction`]
+        The parent interaction which this view was sent from.
+        If ``None`` then the view was not sent using :meth:`InteractionResponse.send_message`.
     """
 
     __discord_ui_view__: ClassVar[bool] = True
@@ -187,6 +190,7 @@ def __init__(
         self.__timeout_task: asyncio.Task[None] | None = None
         self.__stopped: asyncio.Future[bool] = loop.create_future()
         self._message: Message | InteractionMessage | None = None
+        self.parent: Interaction | None = None
 
     def __repr__(self) -> str:
         return f"<{self.__class__.__name__} timeout={self.timeout} children={len(self.children)}>"
@@ -363,9 +367,12 @@ async def on_timeout(self) -> None:
         A callback that is called when a view's timeout elapses without being explicitly stopped.
         """
         if self.disable_on_timeout:
-            if self._message:
-                self.disable_all_items()
-                await self._message.edit(view=self)
+            self.disable_all_items()
+            message = self._message or self.parent
+            if message:
+                m = await message.edit(view=self)
+                if m:
+                    self._message = m
 
     async def on_check_failure(self, interaction: Interaction) -> None:
         """|coro|
@@ -438,6 +445,9 @@ def _dispatch_item(self, item: Item, interaction: Interaction):
         if self.__stopped.done():
             return
 
+        if interaction.message:
+            self.message = interaction.message
+
         asyncio.create_task(
             self._scheduled_task(item, interaction),
             name=f"discord-ui-view-dispatch-{self.id}",