added error handling to views

SylteA · Nov 30, 2023 · b92ce53 · b92ce53
1 parent 517dd27
commit b92ce53
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git 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
@@ -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")