From ff765b41eb005865c12c662a786761961a7ec616 Mon Sep 17 00:00:00 2001 From: Chris Lovering Date: Mon, 26 Jun 2023 09:45:16 +0100 Subject: Update code for new linter rules --- pydis_core/utils/interactions.py | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'pydis_core/utils/interactions.py') 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: -- cgit v1.2.3