mirror of
https://github.com/Benexl/FastAnime.git
synced 2026-01-06 17:53:40 -08:00
feat(cli):improve anilist interfaces api
This commit is contained in:
Benex254
404
fastanime/cli/interfaces/anilist_interfaces.py
Normal file
404
fastanime/cli/interfaces/anilist_interfaces.py
Normal file
@@ -0,0 +1,404 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import sys
|
||||
|
||||
from InquirerPy import inquirer
|
||||
from rich import print
|
||||
|
||||
from ...libs.anilist.anilist import AniList
|
||||
from ...libs.anilist.anilist_data_schema import AnilistBaseMediaDataSchema
|
||||
from ...libs.anime_provider.allanime.api import anime_provider
|
||||
from ...libs.anime_provider.allanime.data_types import AllAnimeEpisode, AllAnimeShow
|
||||
from ..config import Config
|
||||
from ..utils.mpv import mpv
|
||||
from ..utils.tools import QueryDict
|
||||
from ..utils.utils import clear, fuzzy_inquirer
|
||||
|
||||
|
||||
# FIXME: BACK brocken
|
||||
def player_controls(config: Config, anilist_config: QueryDict):
|
||||
# user config
|
||||
translation_type: str = config.translation_type
|
||||
|
||||
# internal config
|
||||
_anime: AllAnimeShow = anilist_config._anime
|
||||
current_episode: str = anilist_config.episode_number
|
||||
episodes: list = anilist_config.episodes
|
||||
links: list = anilist_config.current_stream_links
|
||||
current_link: str = anilist_config.current_stream_link
|
||||
anime_title: str = anilist_config.anime_title
|
||||
|
||||
def _back():
|
||||
fetch_streams(config, anilist_config)
|
||||
|
||||
def _replay():
|
||||
print(
|
||||
"[bold magenta]Now Replaying:[/]",
|
||||
anime_title,
|
||||
"[bold magenta] Episode: [/]",
|
||||
current_episode,
|
||||
)
|
||||
|
||||
config.update_watch_history(anime_title, current_episode)
|
||||
|
||||
mpv(current_link)
|
||||
clear()
|
||||
player_controls(config, anilist_config)
|
||||
|
||||
def _next_episode():
|
||||
next_episode = episodes.index(current_episode) + 1
|
||||
if next_episode >= len(episodes):
|
||||
next_episode = len(episodes) - 1
|
||||
episode = anime_provider.get_anime_episode(
|
||||
_anime["_id"], episodes[next_episode], translation_type
|
||||
)
|
||||
|
||||
# update internal config
|
||||
anilist_config.episode = episode
|
||||
anilist_config.episode_number = episodes[next_episode]
|
||||
|
||||
# update user config
|
||||
config.update_watch_history(anime_title, episodes[next_episode])
|
||||
|
||||
# call interface
|
||||
fetch_streams(config, anilist_config)
|
||||
|
||||
def _episodes():
|
||||
# reset watch_history
|
||||
config.update_watch_history(anime_title, None)
|
||||
|
||||
# call interface
|
||||
fetch_episode(config, anilist_config)
|
||||
|
||||
def _previous_episode():
|
||||
prev_episode = episodes.index(current_episode) - 1
|
||||
if prev_episode <= 0:
|
||||
prev_episode = 0
|
||||
episode = anime_provider.get_anime_episode(
|
||||
_anime["_id"], episodes[prev_episode], config.translation_type
|
||||
)
|
||||
|
||||
# update internal config
|
||||
anilist_config.episode = episode
|
||||
# anilist_config.episode_title = episode["title"]
|
||||
anilist_config.episode_number = episodes[prev_episode]
|
||||
|
||||
# update user config
|
||||
config.update_watch_history(anime_title, episodes[prev_episode])
|
||||
|
||||
# call interface
|
||||
fetch_streams(config, anilist_config)
|
||||
|
||||
def _change_quality():
|
||||
# extract the actual link urls
|
||||
options = [link["link"] for link in links]
|
||||
|
||||
# prompt for new quality
|
||||
quality = fuzzy_inquirer("Select Quality:", options)
|
||||
config.quality = options.index(quality) # set quality
|
||||
player_controls(config, anilist_config)
|
||||
|
||||
def _change_translation_type():
|
||||
# prompt for new translation type
|
||||
options = ["sub", "dub"]
|
||||
translation_type = fuzzy_inquirer("Select Translation Type:", options)
|
||||
|
||||
# update internal config
|
||||
config.translation_type = translation_type
|
||||
|
||||
# reload to controls
|
||||
player_controls(config, anilist_config)
|
||||
|
||||
options = {
|
||||
"Replay": _replay,
|
||||
"Next Episode": _next_episode,
|
||||
"Episodes": _episodes,
|
||||
"Previous Episode": _previous_episode,
|
||||
"Change Quality": _change_quality,
|
||||
"Change Translation Type": _change_translation_type,
|
||||
"Back to servers": _back,
|
||||
"Go to Main Menu": lambda: anilist(config, anilist_config),
|
||||
"Go to Anime Options Menu": lambda: anilist_options(config, anilist_config),
|
||||
"Go to Search Results": lambda: select_anime(config, anilist_config),
|
||||
"exit": sys.exit,
|
||||
}
|
||||
|
||||
action = fuzzy_inquirer("Select Action:", options.keys())
|
||||
|
||||
# update_watch_history
|
||||
config.update_watch_history(anime_title, current_episode)
|
||||
options[action]()
|
||||
|
||||
|
||||
def fetch_streams(config: Config, anilist_config: QueryDict):
|
||||
# user config
|
||||
quality: int = config.quality
|
||||
|
||||
# internal config
|
||||
episode: AllAnimeEpisode = anilist_config.episode
|
||||
episode_number: str = anilist_config.episode_number
|
||||
anime_title: str = anilist_config.anime_title
|
||||
|
||||
# get streams for episode from provider
|
||||
episode_streams = anime_provider.get_episode_streams(episode)
|
||||
episode_streams = {
|
||||
episode_stream[0]: episode_stream[1] for episode_stream in episode_streams
|
||||
}
|
||||
|
||||
# prompt for preferred server
|
||||
server = fuzzy_inquirer("Select Server:", [*episode_streams.keys(), "back"])
|
||||
if server == "back":
|
||||
# reset watch_history
|
||||
config.update_watch_history(anime_title, None)
|
||||
|
||||
fetch_episode(config, anilist_config)
|
||||
return
|
||||
selected_server = episode_streams[server]
|
||||
|
||||
links = selected_server["links"]
|
||||
if quality > len(links) - 1:
|
||||
quality = config.quality = len(links) - 1
|
||||
elif quality < 0:
|
||||
quality = config.quality = 0
|
||||
stream_link = links[quality]["link"]
|
||||
|
||||
# update internal config
|
||||
anilist_config.current_stream_links = links
|
||||
anilist_config.current_stream_link = stream_link
|
||||
anilist_config.current_server = selected_server
|
||||
anilist_config.current_server_name = server
|
||||
|
||||
# play video
|
||||
print(
|
||||
"[bold magenta]Now playing:[/]",
|
||||
anime_title,
|
||||
"[bold magenta] Episode: [/]",
|
||||
episode_number,
|
||||
)
|
||||
|
||||
mpv(stream_link)
|
||||
|
||||
# switch to controls
|
||||
clear()
|
||||
player_controls(config, anilist_config)
|
||||
|
||||
|
||||
def fetch_episode(config: Config, anilist_config: QueryDict):
|
||||
# user config
|
||||
translation_type: str = config.translation_type
|
||||
continue_from_history: bool = config.continue_from_history
|
||||
user_watch_history = config.watch_history
|
||||
anime_title = anilist_config.anime_title
|
||||
|
||||
# internal config
|
||||
anime = anilist_config.anime
|
||||
_anime: AllAnimeShow = anilist_config._anime
|
||||
|
||||
# prompt for episode number
|
||||
# TODO: Load episode number from cache
|
||||
episodes = anime["show"]["availableEpisodesDetail"][translation_type]
|
||||
if continue_from_history and user_watch_history.get(anime_title) in episodes:
|
||||
episode_number = user_watch_history[anime_title]
|
||||
print(f"[bold cyan]Continuing from Episode:[/] [bold]{episode_number}[/]")
|
||||
else:
|
||||
episode_number = fuzzy_inquirer(
|
||||
"Select Episode:",
|
||||
[*episodes, "back"],
|
||||
)
|
||||
|
||||
if episode_number == "back":
|
||||
provide_anime(config, anilist_config)
|
||||
return
|
||||
config.update_watch_history(anime_title, episode_number)
|
||||
|
||||
# get the episode info from provider
|
||||
episode = anime_provider.get_anime_episode(
|
||||
_anime["_id"], episode_number, translation_type
|
||||
)
|
||||
|
||||
# update internal config
|
||||
anilist_config.episodes = episodes
|
||||
anilist_config.episode = episode
|
||||
# anilist_config.episode_title = episode["title"]
|
||||
anilist_config.episode_number = episode_number
|
||||
|
||||
# next interface
|
||||
fetch_streams(config, anilist_config)
|
||||
|
||||
|
||||
def fetch_anime_epiosode(config, anilist_config: QueryDict):
|
||||
selected_anime: AllAnimeShow = anilist_config._anime
|
||||
anilist_config.anime = anime_provider.get_anime(selected_anime["_id"])
|
||||
|
||||
fetch_episode(config, anilist_config)
|
||||
|
||||
|
||||
def provide_anime(config: Config, anilist_config: QueryDict):
|
||||
# user config
|
||||
translation_type = config.translation_type
|
||||
|
||||
# internal config
|
||||
selected_anime_title = anilist_config.selected_anime_title
|
||||
|
||||
# search and get the requested title from provider
|
||||
search_results = anime_provider.search_for_anime(
|
||||
selected_anime_title, translation_type
|
||||
)
|
||||
|
||||
search_results = {
|
||||
anime["name"]: anime for anime in search_results["shows"]["edges"]
|
||||
}
|
||||
anime_title = fuzzy_inquirer(
|
||||
"Select Search Result:",
|
||||
[*search_results.keys(), "back"],
|
||||
default=selected_anime_title,
|
||||
)
|
||||
|
||||
if anime_title == "back":
|
||||
anilist_options(config, anilist_config)
|
||||
return
|
||||
anilist_config.anime_title = anime_title
|
||||
anilist_config._anime = search_results[anime_title]
|
||||
fetch_anime_epiosode(config, anilist_config)
|
||||
|
||||
|
||||
def anilist_options(config, anilist_config: QueryDict):
|
||||
selected_anime: AnilistBaseMediaDataSchema = anilist_config.selected_anime_anilist
|
||||
selected_anime_title: str = anilist_config.selected_anime_title
|
||||
|
||||
def _watch_trailer(config, anilist_config):
|
||||
if trailer := selected_anime.get("trailer"):
|
||||
trailer_url = "https://youtube.com/watch?v=" + trailer["id"]
|
||||
print("[bold magenta]Watching Trailer of:[/]", selected_anime_title)
|
||||
mpv(trailer_url)
|
||||
anilist_options(config, anilist_config)
|
||||
|
||||
def _add_to_list(config, anilist_config):
|
||||
pass
|
||||
|
||||
def _remove_from_list():
|
||||
pass
|
||||
|
||||
def _change_translation_type(config, anilist_config):
|
||||
# prompt for new translation type
|
||||
options = ["sub", "dub"]
|
||||
translation_type = fuzzy_inquirer("Select Translation Type:", options)
|
||||
|
||||
# update internal config
|
||||
config.translation_type = translation_type
|
||||
|
||||
anilist_options(config, anilist_config)
|
||||
|
||||
def _view_info(config, anilist_config):
|
||||
from InquirerPy import inquirer
|
||||
from rich.console import Console
|
||||
|
||||
from ...Utility import anilist_data_helper
|
||||
from ...Utility.utils import remove_html_tags
|
||||
from ..utils.print_img import print_img
|
||||
|
||||
clear()
|
||||
console = Console()
|
||||
|
||||
print_img(selected_anime["coverImage"]["medium"])
|
||||
console.print("[bold cyan]Title(jp): ", selected_anime["title"]["romaji"])
|
||||
console.print("[bold cyan]Title(eng): ", selected_anime["title"]["english"])
|
||||
console.print("[bold cyan]Popularity: ", selected_anime["popularity"])
|
||||
console.print("[bold cyan]Favourites: ", selected_anime["favourites"])
|
||||
console.print("[bold cyan]Status: ", selected_anime["status"])
|
||||
console.print(
|
||||
"[bold cyan]Start Date: ",
|
||||
anilist_data_helper.format_anilist_date_object(selected_anime["startDate"]),
|
||||
)
|
||||
console.print(
|
||||
"[bold cyan]End Date: ",
|
||||
anilist_data_helper.format_anilist_date_object(selected_anime["endDate"]),
|
||||
)
|
||||
# console.print("[bold cyan]Season: ", selected_anime["season"])
|
||||
console.print("[bold cyan]Episodes: ", selected_anime["episodes"])
|
||||
console.print(
|
||||
"[bold cyan]Tags: ",
|
||||
anilist_data_helper.format_list_data_with_comma(
|
||||
[tag["name"] for tag in selected_anime["tags"]]
|
||||
),
|
||||
)
|
||||
console.print(
|
||||
"[bold cyan]Genres: ",
|
||||
anilist_data_helper.format_list_data_with_comma(selected_anime["genres"]),
|
||||
)
|
||||
# console.print("[bold cyan]Type: ", selected_anime["st"])
|
||||
if selected_anime["nextAiringEpisode"]:
|
||||
console.print(
|
||||
"[bold cyan]Next Episode: ",
|
||||
anilist_data_helper.extract_next_airing_episode(
|
||||
selected_anime["nextAiringEpisode"]
|
||||
),
|
||||
)
|
||||
console.print(
|
||||
"[bold underline cyan]Description\n[/]",
|
||||
remove_html_tags(str(selected_anime["description"])),
|
||||
)
|
||||
if inquirer.confirm("Enter to continue", default=True).execute():
|
||||
anilist_options(config, anilist_config)
|
||||
return
|
||||
|
||||
options = {
|
||||
"stream": provide_anime,
|
||||
"watch trailer": _watch_trailer,
|
||||
"add to list": _add_to_list,
|
||||
"remove from list": _remove_from_list,
|
||||
"view info": _view_info,
|
||||
"Change Translation Type": _change_translation_type,
|
||||
"back": select_anime,
|
||||
}
|
||||
action = fuzzy_inquirer("Select Action:", options.keys())
|
||||
options[action](config, anilist_config)
|
||||
|
||||
|
||||
def select_anime(config: Config, anilist_config: QueryDict):
|
||||
anime_data = {
|
||||
str(
|
||||
anime["title"][config.preferred_language] or anime["title"]["romaji"]
|
||||
): anime
|
||||
for anime in anilist_config.data["data"]["Page"]["media"]
|
||||
}
|
||||
selected_anime_title = fuzzy_inquirer(
|
||||
"Select Anime:",
|
||||
[
|
||||
*anime_data.keys(),
|
||||
"back",
|
||||
],
|
||||
)
|
||||
if selected_anime_title == "back":
|
||||
anilist(config, anilist_config)
|
||||
return
|
||||
|
||||
selected_anime = anime_data[selected_anime_title]
|
||||
anilist_config.selected_anime_anilist = selected_anime
|
||||
anilist_config.selected_anime_title = selected_anime_title
|
||||
|
||||
anilist_options(config, anilist_config)
|
||||
|
||||
|
||||
def anilist(config: Config, anilist_config: QueryDict):
|
||||
def _anilist_search():
|
||||
search_term = inquirer.text(
|
||||
"Search:", instruction="Enter anime to search for"
|
||||
).execute()
|
||||
|
||||
return AniList.search(query=search_term)
|
||||
|
||||
options = {
|
||||
"trending": AniList.get_trending,
|
||||
"search": _anilist_search,
|
||||
"most popular anime": AniList.get_most_popular,
|
||||
"most favourite anime": AniList.get_most_favourite,
|
||||
"most scored anime": AniList.get_most_scored,
|
||||
"upcoming anime": AniList.get_upcoming_anime,
|
||||
"recently updated anime": AniList.get_most_recently_updated,
|
||||
}
|
||||
action = fuzzy_inquirer("Select Action:", options.keys())
|
||||
anilist_data = options[action]()
|
||||
if anilist_data[0]:
|
||||
anilist_config.data = anilist_data[1]
|
||||
select_anime(config, anilist_config)
|
||||
Reference in New Issue
Block a user