152 lines
5.6 KiB
Python
152 lines
5.6 KiB
Python
from manga_api import Manga, Chapter
|
|
import discord
|
|
import os.path
|
|
|
|
import requests
|
|
from util import parallel_downloads
|
|
|
|
prev_label = "Prev"
|
|
next_label = "Next"
|
|
add_label = "Add"
|
|
remove_label = "Remove"
|
|
|
|
|
|
class ListManga(discord.ui.View):
|
|
def __init__(self, manga_list: list[Manga], user_manga_list: list[Manga]):
|
|
super().__init__()
|
|
self.user_manga_list = user_manga_list
|
|
self.ret = []
|
|
self.index = 0
|
|
self.manga_list = manga_list
|
|
self.thumbnail_files: list[str] = gen_manga_files(self.manga_list)
|
|
self.comp = self.children
|
|
self.prev_button: discord.Button = None
|
|
self.next_button: discord.Button = None
|
|
self.add_button: discord.Button = None
|
|
self.remove_button: discord.Button = None
|
|
|
|
for comp in self.comp:
|
|
if comp.type.name == "button":
|
|
button: discord.Button = comp
|
|
if button.label == prev_label:
|
|
self.prev_button = button
|
|
if button.label == next_label:
|
|
self.next_button = button
|
|
if button.label == add_label:
|
|
self.add_button = button
|
|
if button.label == remove_label:
|
|
self.remove_button = button
|
|
self.msg: discord.Message = None
|
|
print(self.manga_list)
|
|
|
|
def set_msg(self, msg):
|
|
self.msg = msg
|
|
|
|
async def update_buttons(self):
|
|
self.prev_button.disabled = self.index == 0
|
|
self.next_button.disabled = self.index == len(self.manga_list) - 1
|
|
self.add_button.disabled = self.manga_list[self.index] in self.user_manga_list
|
|
self.remove.disabled = self.manga_list[self.index] not in self.user_manga_list
|
|
|
|
async def force_reload(self):
|
|
await self.update_buttons()
|
|
|
|
self.msg = await self.msg.edit(
|
|
embed=manga_embed(self.manga_list, self.index),
|
|
attachments=[
|
|
parallel_downloads.discord_file_from_filename(self.thumbnail_files[self.index])
|
|
],
|
|
view=self
|
|
)
|
|
|
|
async def print_manga(self, interaction: discord.Interaction):
|
|
|
|
await interaction.response.defer()
|
|
|
|
await self.update_buttons()
|
|
|
|
await self.force_reload()
|
|
|
|
@discord.ui.button(label=prev_label, style=discord.ButtonStyle.grey)
|
|
async def previous(self, interaction: discord.Interaction, button: discord.ui.Button):
|
|
self.prev_button = button
|
|
if self.index > 0:
|
|
self.index -= 1
|
|
print("cant go anymore back")
|
|
|
|
await self.print_manga(interaction)
|
|
|
|
@discord.ui.button(label=next_label, style=discord.ButtonStyle.grey)
|
|
async def next(self, interaction: discord.Interaction, button: discord.ui.Button):
|
|
self.next_button = button
|
|
if self.index < len(self.manga_list) - 1:
|
|
self.index += 1
|
|
|
|
await self.print_manga(interaction)
|
|
|
|
@discord.ui.button(label=add_label, style=discord.ButtonStyle.green)
|
|
async def add(self, interaction: discord.Interaction, button: discord.ui.Button):
|
|
self.ret.append({"manga": self.manga_list[self.index].id, "action": 1})
|
|
if self.manga_list[self.index] not in self.user_manga_list:
|
|
self.user_manga_list.append(self.manga_list[self.index])
|
|
|
|
await self.print_manga(interaction)
|
|
|
|
@discord.ui.button(label=remove_label, style=discord.ButtonStyle.red)
|
|
async def remove(self, interaction: discord.Interaction, button: discord.ui.Button):
|
|
self.ret.append({"manga": self.manga_list[self.index].id, "action": -1})
|
|
if self.manga_list[self.index] in self.user_manga_list:
|
|
self.user_manga_list.remove(self.manga_list[self.index])
|
|
|
|
await self.print_manga(interaction)
|
|
|
|
@discord.ui.button(label='Done', style=discord.ButtonStyle.blurple)
|
|
async def exit(self, interaction: discord.Interaction, button: discord.ui.Button):
|
|
if len(self.ret) > 0:
|
|
await interaction.response.send_message("Your list has been altered successfully", ephemeral=True)
|
|
else:
|
|
await interaction.response.send_message("You didn't make any changes", ephemeral=True)
|
|
self.stop()
|
|
|
|
|
|
def gen_manga_files(manga_list: list[Manga]):
|
|
return parallel_downloads.parallel_download(manga_list)
|
|
|
|
|
|
def manga_embed(manga_list: list[Manga], index: int):
|
|
manga = manga_list[index]
|
|
e = discord.Embed(title=f"({index + 1}\\{len(manga_list)}) {manga.get_title()}",
|
|
description=manga.get_description(), url=manga.get_url())
|
|
extension = manga.get_cover_art_extension()
|
|
e.set_thumbnail(url=f"attachment://{manga.id}.{extension}")
|
|
|
|
return e
|
|
|
|
def manga_list_embed(manga_list: list[Manga]):
|
|
e = discord.Embed(
|
|
title=f"Here's The Manga That Match Your Query ({len(manga_list)} Items)",
|
|
description="".join(
|
|
[f"{index+1} - {manga_list[index].get_title()}\n" for index in range(len(manga_list))]
|
|
),
|
|
color=discord.Color.blurple()
|
|
)
|
|
|
|
return e
|
|
|
|
|
|
def get_chapter_filenames(manga: Manga):
|
|
return parallel_downloads.parallel_download([manga])
|
|
|
|
|
|
def chapter_embed(manga: Manga, chapter: Chapter):
|
|
volume_info = f"Volume {chapter.get_volume()}" if chapter.get_volume() else ""
|
|
chapter_title = f"{chapter.get_title()}" if chapter.get_title() else ""
|
|
e = discord.Embed(
|
|
title=f'"{manga.get_title()}" Chapter {chapter_title} Released!',
|
|
description=f"{volume_info} Chapter {chapter.get_number()} of {manga.get_title()} Released."
|
|
f"\nGo read it now!",
|
|
url=chapter.get_url()
|
|
)
|
|
e.set_thumbnail(url=f"attachment://{manga.id}.{manga.get_cover_art_extension()}")
|
|
return e
|