feat: add MarkAsSolution view and integrate acceptance feature in post assist workflow

Author: mystique09

src/ui/views/forum_showcase.py

@@ -99,8 +99,10 @@ async def channel_select(self, interaction: Interaction, selection: ChannelSelec
             self.forum_showcase.target_channel = self.selected_channel
         await interaction.response.defer()
 
+
 class ConfigureInterval(View):
-    def __init__(self,
+    def __init__(
+        self,
         forum_showcase: ForumShowcase,
         forum_showcase_db: ForumShowcaseDB,
         logger: Logger,
@@ -116,7 +118,7 @@ async def cancel_button(self, interaction: Interaction, button: Button):
         await interaction.response.send_message("Skipped!", ephemeral=True)
         self.stop()
 
-    @button(label="Submit", style=ButtonStyle.green, custom_id="submit-interval-select") # type: ignore
+    @button(label="Submit", style=ButtonStyle.green, custom_id="submit-interval-select")  # type: ignore
     async def submit_button(self, interaction: Interaction, button: Button):
         if self.selected_interval is None:
             await interaction.response.send_message(
@@ -159,6 +161,7 @@ async def select_interval(self, interaction: Interaction, selection: Select):
             self.forum_showcase.interval = self.selected_interval
         await interaction.response.defer()
 
+
 class ConfigureWeekday(View):
     def __init__(
         self,
@@ -304,6 +307,8 @@ def parse_schedule(schedule: str) -> datetime:
         second=0,
         microsecond=0,
     )
-    parsed_schedule = (parsed_schedule - timedelta(hours=8)).replace(tzinfo=timezone.utc)
+    parsed_schedule = (parsed_schedule - timedelta(hours=8)).replace(
+        tzinfo=timezone.utc
+    )
 
     return parsed_schedule

src/ui/views/mark_as_solution.py

@@ -0,0 +1,21 @@
+from discord.ui import View, button
+from discord import Message, Thread, ButtonStyle, Interaction, Button
+
+
+class MarkAsSolution(View):
+    def __init__(self, thread: Thread, message: Message):
+        super().__init__(timeout=None)
+        self.thread = thread
+        self.message = message
+        self.confirmed = False
+
+    @button(label="Yes", style=ButtonStyle.green)  # type: ignore
+    async def accept(self, interaction: Interaction, button: Button):
+        self.confirmed = True
+        self.stop()
+
+    @button(label="No", style=ButtonStyle.red)  # type: ignore
+    async def decline(self, interaction: Interaction, button: Button):
+        self.confirmed = False
+        await interaction.response.send_message("Cancelled.", ephemeral=True)
+        self.stop()

src/ui/views/post_assist.py

@@ -14,48 +14,44 @@
 )
 
 
-class PostAssistState():
+class PostAssistState:
     def __init__(
         self,
         forum: int = None,
         tag_message: str = None,
         custom_msg: str = None,
         existing_tags: list[Role] | list[Member] = [],
+        enable_accept_solutions: bool = False,
         finished: bool = False,
     ):
         self.forum: int = forum
         self.tag_list: list[tuple[int, str]] = []
         self.existing_tags: list[Role] | list[Member] = existing_tags or []
         self.tag_message: str = tag_message
         self.custom_msg: str = custom_msg
+        self.enable_accept_solutions: bool = enable_accept_solutions
         self.finished = finished
 
 
 class ConfigurePostAssist(View):
     def __init__(
-        self,
-        forum: int = None,
-        tag_message: str = None,
-        custom_msg: str = None
+        self, forum: int = None, tag_message: str = None, custom_msg: str = None
     ):
         super().__init__(timeout=480)
         self.state = PostAssistState(
             forum=forum,
             tag_message=tag_message,
             custom_msg=custom_msg,
             existing_tags=[],
+            enable_accept_solutions=False,
         )
 
     @select(
         cls=ChannelSelect,
         placeholder="Select forum...",
         channel_types=[ChannelType.forum],
     )
-    async def select_forum(
-        self,
-        interaction: Interaction,
-        selection: ChannelSelect
-    ):
+    async def select_forum(self, interaction: Interaction, selection: ChannelSelect):
         if self.state.forum and self.state.forum != selection.values[0].id:
             return await interaction.response.send_message(
                 f"Please select this forum -> {interaction.guild.get_channel(self.state.forum).mention}.",
@@ -100,13 +96,12 @@ def __init__(self, options: PostAssistState):
         self.state = options
         self.selection = [tag for tag in self.state.existing_tags]
 
-        select_menu = [item for item in
-                       self.children if isinstance(item, MentionableSelect)][0]
+        select_menu = [
+            item for item in self.children if isinstance(item, MentionableSelect)
+        ][0]
         select_menu.default_values = self.state.existing_tags
 
-    @select(cls=MentionableSelect,
-            placeholder="Select member/roles...",
-            max_values=25)
+    @select(cls=MentionableSelect, placeholder="Select member/roles...", max_values=25)
     async def select_entities(
         self, interaction: Interaction, selection: MentionableSelect
     ):
@@ -123,9 +118,7 @@ async def submit(self, interaction: Interaction, button: Button):
             for entity in self.selection
         ]
 
-        modal = PostAssistTagMessage(
-            self.state, required=bool(self.state.tag_list)
-        )
+        modal = PostAssistTagMessage(self.state, required=bool(self.state.tag_list))
         await interaction.response.send_modal(modal)
         await modal.wait()
         self.stop()
@@ -150,7 +143,34 @@ def __init__(self, options: PostAssistState, required: bool):
 
     async def on_submit(self, interaction: Interaction) -> None:
         self.state.tag_message = self.message.value
-        await interaction.response.send_message("Success...", ephemeral=True)
+
+        view = PostAssistMarkAsSolution(self.state)
+        await interaction.response.send_message(
+            "Enable Accept Solution feature?", view=view, ephemeral=True
+        )
+        await view.wait()
+        self.stop()
+
+
+class PostAssistMarkAsSolution(View):
+    def __init__(self, options: PostAssistState):
+        super().__init__(timeout=480)
+        self.state = options
+
+    @button(label="Yes")
+    async def enable(self, interaction: Interaction, button: Button):
+        self.state.enable_accept_solutions = True
+
+        await interaction.response.send_message(
+            "Mark as solution will be enabled.", ephemeral=True
+        )
+        await interaction.followup.send("Success...", ephemeral=True)
+        self.state.finished = True
+        self.stop()
+
+    @button(label="No")
+    async def disable(self, interaction: Interaction, button: Button):
+        self.state.enable_accept_solutions = False
         self.state.finished = True
         self.stop()
 
@@ -172,9 +192,7 @@ async def previous(self, interaction: Interaction, button: Button):
             button.disabled = True
 
         await interaction.response.edit_message(
-            content=format_data(self.data[self.page],
-                                interaction.guild,
-                                self.getter),
+            content=format_data(self.data[self.page], interaction.guild, self.getter),
             view=self,
         )
 
@@ -188,9 +206,7 @@ async def next(self, interaction: Interaction, button: Button):
             button.disabled = True
 
         await interaction.response.edit_message(
-            content=format_data(self.data[self.page],
-                                interaction.guild,
-                                self.getter),
+            content=format_data(self.data[self.page], interaction.guild, self.getter),
             view=self,
         )