feat(cli): add bing mode to search subcommand using episode ranges

This commit is contained in:
Benex254
2024-08-05 09:47:02 +03:00
parent ab9efb7d3f
commit 7a1236c677
2 changed files with 46 additions and 11 deletions

View File

@@ -9,7 +9,9 @@ from ...utils.tools import QueryDict
help="Search for anime using anilists api and get top ~50 results",
short_help="Search for anime",
)
@click.option("--title", prompt="Enter anime title")
@click.argument(
"title",
)
@click.pass_obj
def search(config, title):
success, search_results = AniList.search(title)

View File

@@ -7,25 +7,33 @@ from ...libs.anime_provider.allanime.api import anime_provider
from ...libs.anime_provider.types import Anime
from ...libs.fzf import fzf
from ..utils.mpv import mpv
from ..utils.tools import exit_app
from ..utils.utils import clear
@click.command(
help="This subcommand directly interacts with the provider to enable basic streaming and perhaps automation if required",
short_help="Directly stream anime with provider",
help="This subcommand directly interacts with the provider to enable basic streaming. Useful for binging anime.",
short_help="Binge anime",
)
@click.option(
"--episode-range",
"-r",
help="A range of episodes to binge",
)
@click.argument("anime_title", required=True, type=str)
@click.pass_obj
def search(
config: Config,
anime_title: str,
):
def search(config: Config, anime_title: str, episode_range: str):
search_results = anime_provider.search_for_anime(
anime_title, config.translation_type
)
search_results = search_results["results"]
if not search_results:
print("Anime not found :cry:")
exit_app()
search_results_ = {
search_result["title"]: search_result for search_result in search_results
}
if config.auto_select:
search_result = max(
search_results_.keys(), key=lambda title: fuzz.ratio(title, anime_title)
@@ -39,10 +47,32 @@ def search(
anime: Anime = anime_provider.get_anime(search_results_[search_result]["id"])
def stream_anime():
episodes = anime["availableEpisodesDetail"][config.translation_type]
episode_range_ = None
episodes = anime["availableEpisodesDetail"][config.translation_type]
if episode_range:
episodes_start, episodes_end = episode_range.split("-")
if episodes_start and episodes_end:
episode_range_ = iter(
range(round(float(episodes_start)), round(float(episodes_end)) + 1)
)
else:
episode_range_ = iter(sorted(episodes, key=float))
episode = fzf.run(episodes, "Select an episode: ", header=search_result)
def stream_anime():
clear()
episode = None
if episode_range_:
try:
episode = str(next(episode_range_))
print(
f"[cyan]Auto selecting:[/] {search_result} [cyan]Episode:[/] {episode}"
)
except StopIteration:
print("[green]Completed binge sequence[/]:smile:")
if not episode or episode not in episodes:
episode = fzf.run(episodes, "Select an episode: ", header=search_result)
streams = anime_provider.get_episode_streams(
anime, episode, config.translation_type
)
@@ -50,7 +80,10 @@ def search(
print("Failed to get streams")
return
links = [link["link"] for server in streams for link in server["links"]]
link = fzf.run(links, "Select stream", "Streams")
# TODO: Come up with way to know quality and better server interface
link = links[config.quality]
# link = fzf.run(links, "Select stream", "Streams")
mpv(link, search_result)
stream_anime()