diff options
author | 2023-07-14 13:21:53 +0100 | |
---|---|---|
committer | 2023-07-14 13:21:53 +0100 | |
commit | 29af8a99dea22358ba0774f6b4eb9fbfbaeae353 (patch) | |
tree | f2a31e3b2cbf9803f7d89429a33ed2e5c1a10999 /pydis_core/utils/interactions.py | |
parent | Update get_or_fetch_channel's return type (#187) (diff) | |
parent | Update code for new linter rules (diff) |
Merge pull request #184 from python-discord/bump-d.py
Bump d.py & update view with user and role check
Diffstat (limited to 'pydis_core/utils/interactions.py')
-rw-r--r-- | pydis_core/utils/interactions.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/pydis_core/utils/interactions.py b/pydis_core/utils/interactions.py index cb5f26b4..d3432c3a 100644 --- a/pydis_core/utils/interactions.py +++ b/pydis_core/utils/interactions.py @@ -4,6 +4,7 @@ from typing import Literal from discord import ButtonStyle, HTTPException, Interaction, Message, NotFound, ui from pydis_core.utils.logging import get_logger +from pydis_core.utils.scheduling import create_task log = get_logger(__name__) @@ -80,6 +81,12 @@ class ViewWithUserAndRoleCheck(ui.View): await interaction.response.send_message("This is not your button to click!", ephemeral=True) return False + def stop(self) -> None: + """Stop listening for interactions, and remove the view from ``self.message`` if set.""" + super().stop() + if self.message: + create_task(_handle_modify_message(self.message, "edit")) + async def on_timeout(self) -> None: """Remove the view from ``self.message`` if set.""" if self.message: |