130 lines
4.2 KiB
Python
130 lines
4.2 KiB
Python
import logging
|
|
import os
|
|
|
|
import click
|
|
|
|
from anime_downloader import session, util
|
|
from anime_downloader.__version__ import __version__
|
|
from anime_downloader.sites import get_anime_class
|
|
|
|
logger = logging.Logger(__name__)
|
|
|
|
echo = click.echo
|
|
|
|
|
|
# NOTE: Don't put defaults here. Add them to the dict in config
|
|
@click.command()
|
|
@click.argument('anime_url')
|
|
@click.option(
|
|
'--episodes', '-e', 'episode_range', metavar='<int>:<int>',
|
|
help="Range of anime you want to download in the form <start>:<end>")
|
|
@click.option(
|
|
'--url', '-u', type=bool, is_flag=True,
|
|
help="If flag is set, prints the stream url instead of downloading")
|
|
@click.option(
|
|
'--play', 'player', metavar='PLAYER',
|
|
help="Streams in the specified player")
|
|
@click.option(
|
|
'--skip-download', is_flag=True,
|
|
help="Retrieve without downloading")
|
|
@click.option(
|
|
'--download-dir', metavar='PATH',
|
|
help="Specifiy the directory to download to")
|
|
@click.option(
|
|
'--quality', '-q', type=click.Choice(['360p', '480p', '720p', '1080p']),
|
|
help='Specify the quality of episode. Default-720p')
|
|
@click.option(
|
|
'--fallback-qualities', '-fq', cls=util.ClickListOption,
|
|
help='Specifiy the order of fallback qualities as a list.')
|
|
@click.option(
|
|
'--force-download', '-f', is_flag=True,
|
|
help='Force downloads even if file exists')
|
|
@click.option(
|
|
'--file-format', '-ff', default='{anime_title}/{anime_title}_{ep_no}',
|
|
help='Format for how the files to be downloaded be named.',
|
|
metavar='FORMAT STRING'
|
|
)
|
|
@click.option(
|
|
'--provider',
|
|
help='The anime provider (website) for search.',
|
|
type=click.Choice(['9anime', 'kissanime', 'twist.moe',
|
|
'animepahe', 'kisscartoon', 'masterani', 'gogoanime'])
|
|
)
|
|
@click.option(
|
|
'--external-downloader', '-xd',
|
|
help='Use an external downloader command to download. '
|
|
'Use "{aria2}" to use aria2 as downloader. See github wiki.',
|
|
metavar='DOWNLOAD COMMAND'
|
|
)
|
|
@click.option(
|
|
'--chunk-size',
|
|
help='Chunk size for downloading in chunks(in MB). Use this if you '
|
|
'experience throttling.',
|
|
type=int
|
|
)
|
|
@click.option(
|
|
'--disable-ssl',
|
|
is_flag=True,
|
|
help='Disable verifying the SSL certificate, if flag is set'
|
|
)
|
|
@click.pass_context
|
|
def command(ctx, anime_url, episode_range, url, player, skip_download, quality,
|
|
force_download, download_dir, file_format, provider,
|
|
external_downloader, chunk_size, disable_ssl, fallback_qualities):
|
|
""" Download the anime using the url or search for it.
|
|
"""
|
|
util.print_info(__version__)
|
|
|
|
# TODO: Replace by factory
|
|
cls = get_anime_class(anime_url)
|
|
|
|
disable_ssl = cls and cls.__name__ == 'Masterani' or disable_ssl
|
|
session.get_session().verify = not disable_ssl
|
|
|
|
if not cls:
|
|
anime_url = util.search(anime_url, provider)
|
|
cls = get_anime_class(anime_url)
|
|
|
|
anime = cls(anime_url, quality=quality,
|
|
fallback_qualities=fallback_qualities)
|
|
logger.info('Found anime: {}'.format(anime.title))
|
|
|
|
animes = util.parse_ep_str(anime, episode_range)
|
|
|
|
# TODO:
|
|
# Two types of plugins:
|
|
# - Aime plugin: Pass the whole anime
|
|
# - Ep plugin: Pass each episode
|
|
if url or player:
|
|
skip_download = True
|
|
|
|
if download_dir and not skip_download:
|
|
logger.info('Downloading to {}'.format(os.path.abspath(download_dir)))
|
|
|
|
for episode in animes:
|
|
if url:
|
|
util.print_episodeurl(episode)
|
|
|
|
if player:
|
|
util.play_episode(episode, player=player)
|
|
|
|
if not skip_download:
|
|
downloader_session = session.DownloaderSession()
|
|
downloader = 'http'
|
|
if external_downloader:
|
|
logger.info('Downloading episode {} of {}'.format(
|
|
episode.ep_no, anime.title)
|
|
)
|
|
downloader = external_downloader
|
|
if chunk_size is not None:
|
|
chunk_size *= 1e6
|
|
chunk_size = int(chunk_size)
|
|
|
|
downloader_session.get(downloader)
|
|
downloader.download(episode.source(),
|
|
force=force_download,
|
|
path=download_dir,
|
|
format=file_format,
|
|
range_size=chunk_size)
|
|
print()
|