Chewing_Bever/didier
0
0
Fork 0
mirror of https://github.com/stijndcl/didier.git synced 2026-04-10 17:15:47 +02:00
Code Issues Projects Releases Wiki Activity

Create task to purge old announcements

Browse source
This commit is contained in:
stijndcl 2022-06-19 00:51:24 +02:00
parent d7262595c6
commit d75831f848
2 changed files with 28 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

16
didier/cogs/tasks.py
View file

@ -3,22 +3,23 @@ import traceback
from discord.ext import commands, tasks
import settings
from database.crud.ufora_announcements import remove_old_announcements
from didier import Didier
from didier.data.embeds.ufora.announcements import fetch_ufora_announcements
# TODO task to clean up old announcements? (> 1 week)
class Tasks(commands.Cog):
"""Task loops that run periodically"""
client: Didier
def __init__(self, client: Didier):
def __init__(self, client: Didier): # pylint: disable=no-member
self.client = client
# Only pull announcements if a token was provided
if settings.UFORA_RSS_TOKEN is not None and settings.UFORA_ANNOUNCEMENTS_CHANNEL is not None:
self.pull_ufora_announcements.start() # pylint: disable=no-member
self.pull_ufora_announcements.start()
self.remove_old_ufora_announcements.start()
@tasks.loop(minutes=10)
async def pull_ufora_announcements(self):
@ -43,6 +44,15 @@ class Tasks(commands.Cog):
"""Error handler for the Ufora Announcements task"""
print("".join(traceback.format_exception(type(error), error, error.__traceback__)))
@tasks.loop(hours=24)
async def remove_old_ufora_announcements(self):
"""Remove all announcements that are over 1 week old, once per day"""
await remove_old_announcements(self.client.db_session)
@remove_old_ufora_announcements.before_loop
async def _before_remove_old_ufora_announcements(self):
await self.client.wait_until_ready()
async def setup(client: Didier):
"""Load the cog"""