diff options
author | Arjun Satarkar <me@arjunsatarkar.net> | 2023-12-21 12:14:24 +0000 |
---|---|---|
committer | Arjun Satarkar <me@arjunsatarkar.net> | 2023-12-21 12:14:24 +0000 |
commit | 4d905d103cbaf568fecc097f13f66f843d5c7282 (patch) | |
tree | 26e53a469d37245b135fcc851f1023880687f05c /question_of_the_day | |
parent | ea8bd90461334e52488b06b3e3c1a783c511727c (diff) | |
download | aps-cogs-4d905d103cbaf568fecc097f13f66f843d5c7282.tar aps-cogs-4d905d103cbaf568fecc097f13f66f843d5c7282.tar.gz aps-cogs-4d905d103cbaf568fecc097f13f66f843d5c7282.zip |
question_of_the_day: separate the qotd-posting logic out
Diffstat (limited to 'question_of_the_day')
-rw-r--r-- | question_of_the_day/question_of_the_day.py | 65 |
1 files changed, 45 insertions, 20 deletions
diff --git a/question_of_the_day/question_of_the_day.py b/question_of_the_day/question_of_the_day.py index bcc7b14..fe48e28 100644 --- a/question_of_the_day/question_of_the_day.py +++ b/question_of_the_day/question_of_the_day.py @@ -51,26 +51,8 @@ class QuestionOfTheDay(commands.Cog): self.logger.info( f"QOTD was due for guild {guild.name} ({guild_id}) but no channel was set, so it was not posted." ) - async with self.config.guild(guild).questions() as questions: - channel = await guild.fetch_channel(channel_id) - questions_len = len(questions) - if not questions_len: - await channel.send( - "# Question of the Day\n**No questions left!**" - ) - continue - question_index = random.randrange(0, questions_len) - question = questions[question_index] - await channel.send( - f"# Question of the Day\n" - f"{question['question']}\n{redbot.core.utils.chat_formatting.italics((await guild.fetch_member(question['asked_by'])).name)}" - f" ({question['asked_by']})" - ) - del questions[question_index] - self.logger.info( - f"Posted QOTD for guild {guild.name} ({guild_id})." - ) - + channel = await guild.fetch_channel(channel_id) + await self.post_question(channel) current_time = time.time() current_datetime = datetime.datetime.fromtimestamp( @@ -194,12 +176,18 @@ class QuestionOfTheDay(commands.Cog): @qotd.command() @checks.admin_or_permissions(manage_server=True) async def post_here(self, ctx): + """ + Set the current channel as where QOTDs should be posted. + """ await self.config.guild(ctx.guild).post_in_channel.set(ctx.channel.id) await ctx.reply("Questions of the day will be posted in this channel.") @qotd.command() @checks.admin_or_permissions(manage_server=True) async def toggle(self, ctx): + """ + Turn questions of the day on or off for this server. + """ should_be_enabled = not await self.config.guild(ctx.guild).enabled() await self.config.guild(ctx.guild).enabled.set(should_be_enabled) post_at = await self.config.guild(ctx.guild).post_at() @@ -218,6 +206,9 @@ class QuestionOfTheDay(commands.Cog): @qotd.command() async def suggest(self, ctx, *, question: str): + """ + Add a question to the suggestion queue (it can be approved or denied by moderators). + """ if not await self.check_and_handle_question_length(ctx, question): return async with self.config.guild( @@ -235,6 +226,9 @@ class QuestionOfTheDay(commands.Cog): @qotd.command() async def suggestions(self, ctx): + """ + View all questions in the suggestion queue. + """ pages = await self.paginate_questions( ctx, await self.config.guild(ctx.guild).suggested_questions() ) @@ -246,6 +240,11 @@ class QuestionOfTheDay(commands.Cog): @qotd.command() @checks.admin_or_permissions(manage_server=True) async def approve(self, ctx, suggestion_id: int): + """ + Approve a suggestion using its id (see `qotd suggestions`). + + This adds the suggestion to the main queue. + """ async with self.config.guild( ctx.guild ).suggested_questions() as suggested_questions: @@ -273,6 +272,11 @@ class QuestionOfTheDay(commands.Cog): @qotd.command() @checks.admin_or_permissions(manage_server=True) async def deny(self, ctx, suggestion_id: int): + """ + Decline a suggestion and remove it from the suggestion queue. + + For the suggestion's id, see `qotd suggestions`. + """ async with self.config.guild( ctx.guild ).suggested_questions() as suggested_questions: @@ -287,6 +291,27 @@ class QuestionOfTheDay(commands.Cog): except IndexError: await ctx.reply(f"Error: no suggestion with id {suggestion_id}.") + async def post_question (self, channel): + guild = channel.guild + async with self.config.guild(guild).questions() as questions: + questions_len = len(questions) + if not questions_len: + await channel.send( + "# Question of the Day\n**No questions left!**" + ) + else: + question_index = random.randrange(0, questions_len) + question = questions[question_index] + await channel.send( + f"# Question of the Day\n" + f"{question['question']}\n{redbot.core.utils.chat_formatting.italics((await guild.fetch_member(question['asked_by'])).name)}" + f" ({question['asked_by']})" + ) + del questions[question_index] + self.logger.info( + f"Posted QOTD for guild {guild.name} ({guild.id})." + ) + async def paginate_questions(self, ctx, questions: list): return [ x |