From b92ce5315cc7cd29f2be4164d3d037130458614f Mon Sep 17 00:00:00 2001 From: FirePlank <44502537+FirePlank@users.noreply.github.com> Date: Thu, 30 Nov 2023 19:42:42 +0200 Subject: [PATCH] added error handling to views --- bot/extensions/adventofcode/views.py | 7 ++++++- bot/extensions/polls/views.py | 5 ++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/bot/extensions/adventofcode/views.py b/bot/extensions/adventofcode/views.py index 163b86a0..71e84e2e 100644 --- a/bot/extensions/adventofcode/views.py +++ b/bot/extensions/adventofcode/views.py @@ -1,10 +1,12 @@ import re from datetime import datetime +from typing import Any from zoneinfo import ZoneInfo import discord from bs4 import BeautifulSoup -from discord import ui +from discord import Interaction, ui +from discord.ui import Item from bot import core from bot.extensions.adventofcode.utils import LEADERBOARD_ID, YEAR, Member, fetch_leaderboard, home_embed, ordinal @@ -109,3 +111,6 @@ async def global_leaderboard(self, interaction: core.InteractionType, _button: u embed.description = s_desc await interaction.response.edit_message(embed=embed, view=self) + + async def on_error(self, interaction: Interaction, _error: Exception, _item: Item[Any], /) -> None: + await interaction.client.on_error("adventofcode_view") diff --git a/bot/extensions/polls/views.py b/bot/extensions/polls/views.py index c0041342..00008238 100644 --- a/bot/extensions/polls/views.py +++ b/bot/extensions/polls/views.py @@ -1,7 +1,7 @@ import typing as t import discord -from discord import ui +from discord import Interaction, ui from bot import core from bot.extensions.polls.utils import emojis @@ -118,3 +118,6 @@ async def create_poll(self, interaction: core.InteractionType, _button: ui.Butto for i in range(0, len(embed.fields)): await message.add_reaction(emojis[i]) + + async def on_error(self, interaction: Interaction, _error: Exception, _item: ui.Item[t.Any], /) -> None: + await interaction.client.on_error("poll_view")