Compare commits

...

2 Commits

Author SHA1 Message Date
bobloy
393d7cb350 cog_unload, not __unload, whoops. 2020-10-23 11:41:46 -04:00
bobloy
70f4b0b716 Check for unload and cancel announce task properly. 2020-10-23 11:36:51 -04:00
2 changed files with 9 additions and 1 deletions

View File

@ -6,4 +6,4 @@ from .announcedaily import AnnounceDaily
def setup(bot: Red): def setup(bot: Red):
daily = AnnounceDaily(bot) daily = AnnounceDaily(bot)
bot.add_cog(daily) bot.add_cog(daily)
bot.loop.create_task(daily.check_day()) daily.announce_task = bot.loop.create_task(daily.check_day())

View File

@ -1,6 +1,8 @@
import asyncio import asyncio
import random import random
from asyncio import Task
from datetime import datetime, timedelta from datetime import datetime, timedelta
from typing import Optional
import discord import discord
from redbot.core import Config, checks, commands from redbot.core import Config, checks, commands
@ -38,10 +40,16 @@ class AnnounceDaily(Cog):
self.config.register_global(**default_global) self.config.register_global(**default_global)
self.config.register_guild(**default_guild) self.config.register_guild(**default_guild)
self.announce_task = None
async def red_delete_data_for_user(self, **kwargs): async def red_delete_data_for_user(self, **kwargs):
"""Nothing to delete""" """Nothing to delete"""
return return
def cog_unload(self):
if self.announce_task is not None:
self.announce_task.cancel()
async def _get_msgs(self): async def _get_msgs(self):
return DEFAULT_MESSAGES + await self.config.messages() return DEFAULT_MESSAGES + await self.config.messages()