From 84b6acbea9f02751c3bb5f470df746eed4bc0af5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Joakim=20Hells=C3=A9n?= Date: Wed, 26 Oct 2022 01:57:33 +0200 Subject: [PATCH] Stop adding pages when we have 25 --- discord_reminder_bot/create_pages.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/discord_reminder_bot/create_pages.py b/discord_reminder_bot/create_pages.py index 1968e7a..5f0ba77 100644 --- a/discord_reminder_bot/create_pages.py +++ b/discord_reminder_bot/create_pages.py @@ -4,14 +4,14 @@ from typing import List import interactions from apscheduler.job import Job from apscheduler.triggers.date import DateTrigger -from interactions import ActionRow, ComponentContext +from interactions import ActionRow, CommandContext, ComponentContext from interactions.ext.paginator import Page, Paginator, RowPosition from discord_reminder_bot.countdown import calculate from discord_reminder_bot.settings import scheduler -def create_pages(ctx) -> list[Page]: +def create_pages(ctx: CommandContext) -> list[Page]: """Create pages for the paginator. Args: @@ -107,6 +107,11 @@ def create_pages(ctx) -> list[Page]: pause_or_unpause_button = pause_button components.insert(1, pause_or_unpause_button) + # Only allow 25 pages + if len(pages) == 25: + ctx.channel.send("I haven't added support for more than 25 reminders. Pull requests welcome 🙃") + return pages + # Add a page to pages list title = f"{message[:87]}..." if len(message) > 90 else message pages.append(