Croc/croccybot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

(reminders): Add ratelimit bucket.

Browse Source
This commit is contained in:
Conatum
2023-08-18 23:06:03 +03:00
parent 47d58bea37
commit 10084089f1
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/modules/reminders/cog.py
View File

@@ -35,6 +35,7 @@ from utils.lib import parse_duration, utc_now, strfdur, error_embed
from utils.monitor import TaskMonitor from utils.monitor import TaskMonitor
from utils.transformers import DurationTransformer from utils.transformers import DurationTransformer
from utils.ui import LeoUI, AButton, AsComponents from utils.ui import LeoUI, AButton, AsComponents
from utils.ratelimits import Bucket
from . import babel, logger from . import babel, logger
@@ -178,7 +179,10 @@ class Reminders(LionCog):
self.executor_name = appname_from_shard(0) self.executor_name = appname_from_shard(0)
if self.executor: if self.executor:
self.monitor: Optional[ReminderMonitor] = ReminderMonitor(executor=self.execute_reminder) self.monitor: Optional[ReminderMonitor] = ReminderMonitor(
executor=self.execute_reminder,
bucket=Bucket(1, 1)
)
else: else:
self.monitor = None self.monitor = None
@@ -203,7 +207,7 @@ class Reminders(LionCog):
self.monitor._monitor_task.cancel() self.monitor._monitor_task.cancel()
# Attach and populate the reminder monitor # Attach and populate the reminder monitor
self.monitor = ReminderMonitor(executor=self.execute_reminder) self.monitor = ReminderMonitor(executor=self.execute_reminder, bucket=Bucket(1, 1))
await self.reload_reminders() await self.reload_reminders()
# Start firing reminders # Start firing reminders