diff options
Diffstat (limited to 'code/utils/content_view.py')
| -rw-r--r-- | code/utils/content_view.py | 203 |
1 files changed, 203 insertions, 0 deletions
diff --git a/code/utils/content_view.py b/code/utils/content_view.py new file mode 100644 index 0000000..e8df29c --- /dev/null +++ b/code/utils/content_view.py @@ -0,0 +1,203 @@ +import discord +import sqlite3 + +from utils.content_add import add_content + +""" +View to add the Dropdown menu +""" + + +class AddContentView(discord.ui.View): + def __init__( + self, + content_data: list, + service: str, + host: str, + header: str, + path: str, + profile: int, + *, + timeout=180.0, + ): + super().__init__(timeout=timeout) + # Add the dropdown + self.add_item( + AddContentDropdown( + content_data, service, host, header, path, profile + ) + ) + + +""" +Dropdown containing the top 5 content results +""" + + +class AddContentDropdown(discord.ui.Select): + def __init__( + self, + content_data: list, + service: str, + host: str, + header: str, + path: str, + profile: str, + *, + timeout=180.0, + ): + self.content_data = content_data + self.service = service + self.host = host + self.header = header + self.path = path + self.profile = profile + options = [] + for i in range(len(content_data)): + options.append( + discord.SelectOption( + label=( + f"{content_data[i]['title']} ({content_data[i]['year']})" + ), + description=f"Relevant ID: {content_data[i]['contentId']}", + value=str(i), + ) + ) + + super().__init__( + placeholder="Select from the dropdown", + options=options, + ) + + # Once an option has been selected + async def callback(self, interaction: discord.Interaction): + # Index of selected option + index = int(self.values[0]) + + # Add selected contents info to an embed + embed = discord.Embed( + title="Is the the content you want to add?", + description=( + f"**Title**: {self.content_data[index]['title']} | " + f"**Year**: {self.content_data[index]['year']}\n\n" + f"**Description**: {self.content_data[index]['description']}" + ), + color=0xD01B86, + ) + embed.set_image(url=self.content_data[index]["remotePoster"]) + # Change the view to the Request/Don't Request buttons + view = RequestButtonView( + self.content_data[index], + self.service, + self.host, + self.header, + self.path, + self.profile, + ) + await interaction.response.edit_message(embed=embed, view=view) + + +""" +View containing the "Request" and "Don't Request" buttons +""" + + +class RequestButtonView(discord.ui.View): + def __init__( + self, + content_info: dict, + service: str, + host: str, + headers: str, + path: str, + profile: int, + *, + timeout=180.0, + ): + super().__init__(timeout=timeout) + self.content_info = content_info + self.service = service + self.host = host + self.headers = headers + self.path = path + self.profile = profile + + @discord.ui.button(label="Request", style=discord.ButtonStyle.success) + async def request_button( + self, interaction: discord.Interaction, button: discord.ui.Button + ): + # Add the content to the relevant library + local_id = add_content( + self.content_info, + self.service, + self.host, + self.headers, + self.path, + self.profile, + ) + + # Alert the user that the content has been added + if local_id: + embed = discord.Embed( + title="Content Added", + description=( + f"**{self.content_info['title']}** has been added to the" + f" {self.service} library. Check the status of your" + " requested content with `/status`." + ), + color=0xD01B86, + ) + await interaction.response.send_message(embed=embed) + # Alert the user that the content failed to be added + else: + embed = discord.Embed( + title="Failed to Add Content", + description=( + "An error occured when attempting to add" + f" **{self.content_info['title']}** to the" + f" {self.service} library." + ), + ) + return await interaction.response.send_message(embed=embed) + + # Keep track of the requests for the `/status` command + db = sqlite3.connect("cordarr.db") + cursor = db.cursor() + cursor.execute( + "INSERT INTO requests (title, release_year, local_id, tmdbid," + " tvdbid, user_id) VALUES (?, ?, ?, ?, ?, ?)", + ( + self.content_info["title"], + self.content_info["year"], + local_id, + ( + self.content_info["contentId"] + if self.service == "radarr" + else None + ), + ( + None + if self.service == "radarr" + else self.content_info["contentId"] + ), + interaction.user.id, + ), + ) + db.commit() + db.close() + + @discord.ui.button(label="Don't Request", style=discord.ButtonStyle.danger) + async def dont_request_button( + self, interaction: discord.Interaction, button: discord.ui.Button + ): + embed = discord.Embed( + title="Request Cancelled", + description=( + "Request has been cancelled. If you would like to request a" + " different" + f" {'movie' if self.service == 'radarr' else 'show'}, run the" + " `/request` command again." + ), + color=0xD01B86, + ) + await interaction.response.send_message(embed=embed) |