corbz/PYRSS-Bot
corbz/PYRSS-Bot
1
0
Fork 0
Code Issues 14 Pull Requests Actions Releases 2 Activity

task group and exception checker

Browse Source
This commit is contained in:
Corban-Lee Jones 2024-09-10 10:44:50 +01:00
parent e43d7aacf9
commit 1d78a05807
1 changed files with 20 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/extensions/tasks.py
View File

@ -65,13 +65,24 @@ class TaskCog(commands.Cog):
""" """
self.subscription_task.cancel() self.subscription_task.cancel()
@app_commands.command(name="debug-trigger-task") group = app_commands.Group(
async def debug_trigger_task(self, inter): name="task",
description="Commands for tasks",
guild_only=True
)
@group.command(name="trigger")
async def cmd_trigger_task(self, inter):
await inter.response.defer() await inter.response.defer()
start_time = perf_counter() start_time = perf_counter()
await self.subscription_task()
end_time = perf_counter() try:
await inter.followup.send(f"completed in {end_time - start_time:.4f} seconds") await self.subscription_task()
except Exception as error:
await inter.followup.send(str(error))
finally:
end_time = perf_counter()
await inter.followup.send(f"completed in {end_time - start_time:.4f} seconds")
@tasks.loop(time=subscription_task_times) @tasks.loop(time=subscription_task_times)
async def subscription_task(self): async def subscription_task(self):
@ -199,7 +210,7 @@ class TaskCog(commands.Cog):
continue continue
blocked = any(self.filter_item(_filter, item) for _filter in filters) blocked = any(self.filter_item(_filter, item) for _filter in filters)
mutated_item = item.create_mutated_copy(sub.mutators) mutated_item = item.create_mutated_copy(sub.mutators) if sub.mutators else None
for channel in channels: for channel in channels:
await self.track_and_send(sub, feed, item, mutated_item, channel, blocked) await self.track_and_send(sub, feed, item, mutated_item, channel, blocked)
@ -225,7 +236,7 @@ class TaskCog(commands.Cog):
log.debug("filter match found? '%s'", match_found) log.debug("filter match found? '%s'", match_found)
return match_found return match_found
async def track_and_send(self, sub: Subscription, feed: RSSFeed, item: RSSItem, mutated_item: RSSItem, channel: TextChannel, blocked: bool): async def track_and_send(self, sub: Subscription, feed: RSSFeed, item: RSSItem, mutated_item: RSSItem | None, channel: TextChannel, blocked: bool):
message_id = -1 message_id = -1
log.debug("track and send func %s, %s", item.guid, item.title) log.debug("track and send func %s, %s", item.guid, item.title)
@ -243,7 +254,8 @@ class TaskCog(commands.Cog):
if not blocked: if not blocked:
try: try:
log.debug("sending '%s', exists '%s'", item.guid, result[1]) log.debug("sending '%s', exists '%s'", item.guid, result[1])
message = await channel.send(embed=await mutated_item.to_embed(sub, feed, self.api.session)) sendable_item = mutated_item or item
message = await channel.send(embed=await sendable_item.to_embed(sub, feed, self.api.session))
message_id = message.id message_id = message.id
except Forbidden: except Forbidden:
log.error(f"Forbidden to send to channel {channel.id}") log.error(f"Forbidden to send to channel {channel.id}")