diff options
author | 2023-06-26 09:45:16 +0100 | |
---|---|---|
committer | 2023-07-14 13:20:00 +0100 | |
commit | ff765b41eb005865c12c662a786761961a7ec616 (patch) | |
tree | f2a31e3b2cbf9803f7d89429a33ed2e5c1a10999 /pydis_core/utils | |
parent | Bump dev deps (diff) |
Update code for new linter rules
Diffstat (limited to 'pydis_core/utils')
-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: |