diff options
Diffstat (limited to '')
| -rw-r--r-- | bot/exts/utils/thread_bumper.py | 40 |
1 files changed, 23 insertions, 17 deletions
diff --git a/bot/exts/utils/thread_bumper.py b/bot/exts/utils/thread_bumper.py index e93301e0c..588ea1274 100644 --- a/bot/exts/utils/thread_bumper.py +++ b/bot/exts/utils/thread_bumper.py @@ -1,5 +1,6 @@ import discord +import sentry_sdk from discord.ext import commands from pydis_core.site_api import ResponseCodeError from pydis_core.utils.channel import get_or_fetch_channel @@ -67,23 +68,28 @@ class ThreadBumper(commands.Cog): await self.bot.wait_until_guild_available() threads_to_maybe_bump = [] - for thread_id in await self.bot.api_client.get(THREAD_BUMP_ENDPOINT): - try: - thread = await get_or_fetch_channel(self.bot, thread_id) - except discord.NotFound: - log.info("Thread %d has been deleted, removing from bumped threads.", thread_id) - await self.bot.api_client.delete(f"{THREAD_BUMP_ENDPOINT}/{thread_id}") - continue - - if not isinstance(thread, discord.Thread): - await self.bot.api_client.delete(f"{THREAD_BUMP_ENDPOINT}/{thread_id}") - continue - - if thread.archived: - threads_to_maybe_bump.append(thread) - - if threads_to_maybe_bump: - await self.unarchive_threads_not_manually_archived(threads_to_maybe_bump) + with sentry_sdk.start_span(description="Fetch threads to bump from site"): + bumped_threads_from_site = await self.bot.api_client.get(THREAD_BUMP_ENDPOINT) + + with sentry_sdk.start_span(description="Sync bumped threads in site with current guild state"): + for thread_id in bumped_threads_from_site: + try: + thread = await get_or_fetch_channel(self.bot, thread_id) + except discord.NotFound: + log.info("Thread %d has been deleted, removing from bumped threads.", thread_id) + await self.bot.api_client.delete(f"{THREAD_BUMP_ENDPOINT}/{thread_id}") + continue + + if not isinstance(thread, discord.Thread): + await self.bot.api_client.delete(f"{THREAD_BUMP_ENDPOINT}/{thread_id}") + continue + + if thread.archived: + threads_to_maybe_bump.append(thread) + + with sentry_sdk.start_span(description="Unarchive threads that should be bumped"): + if threads_to_maybe_bump: + await self.unarchive_threads_not_manually_archived(threads_to_maybe_bump) @commands.group(name="bump") async def thread_bump_group(self, ctx: commands.Context) -> None: |