mirror of
https://github.com/Benexl/FastAnime.git
synced 2025-12-27 13:13:25 -08:00
405 lines
13 KiB
Python
405 lines
13 KiB
Python
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)
|