mirror of
https://github.com/Benexl/FastAnime.git
synced 2025-12-25 04:15:19 -08:00
102 lines
3.2 KiB
Python
102 lines
3.2 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
import click
|
|
from click.core import ParameterSource
|
|
|
|
from ..core.config import AppConfig
|
|
from ..core.constants import PROJECT_NAME, USER_CONFIG_PATH, __version__
|
|
from .config import ConfigLoader
|
|
from .options import options_from_model
|
|
from .utils.exceptions import setup_exceptions_handler
|
|
from .utils.lazyloader import LazyGroup
|
|
from .utils.logging import setup_logging
|
|
|
|
if TYPE_CHECKING:
|
|
from typing import TypedDict
|
|
|
|
from typing_extensions import Unpack
|
|
|
|
class Options(TypedDict):
|
|
no_config: bool | None
|
|
trace: bool | None
|
|
log_to_file: bool | None
|
|
dev: bool | None
|
|
log: bool | None
|
|
rich_traceback: bool | None
|
|
rich_traceback_theme: str
|
|
|
|
|
|
commands = {
|
|
"config": ".config",
|
|
"search": ".search",
|
|
"download": ".download",
|
|
"anilist": ".anilist",
|
|
}
|
|
|
|
|
|
@click.group(
|
|
cls=LazyGroup,
|
|
root="fastanime.cli.commands",
|
|
lazy_subcommands=commands,
|
|
context_settings=dict(auto_envvar_prefix=PROJECT_NAME),
|
|
)
|
|
@click.version_option(__version__, "--version")
|
|
@click.option("--no-config", is_flag=True, help="Don't load the user config file.")
|
|
@click.option(
|
|
"--trace", is_flag=True, help="Controls Whether to display tracebacks or not"
|
|
)
|
|
@click.option("--dev", is_flag=True, help="Controls Whether the app is in dev mode")
|
|
@click.option("--log", is_flag=True, help="Controls Whether to log")
|
|
@click.option("--log-to-file", is_flag=True, help="Controls Whether to log to a file")
|
|
@click.option(
|
|
"--rich-traceback",
|
|
is_flag=True,
|
|
help="Controls Whether to display a rich traceback",
|
|
)
|
|
@click.option(
|
|
"--rich-traceback-theme",
|
|
default="github-dark",
|
|
help="Controls Whether to display a rich traceback",
|
|
)
|
|
@options_from_model(AppConfig)
|
|
@click.pass_context
|
|
def cli(ctx: click.Context, **options: "Unpack[Options]"):
|
|
"""
|
|
The main entry point for the FastAnime CLI.
|
|
"""
|
|
setup_logging(options["log"], options["log_to_file"])
|
|
setup_exceptions_handler(
|
|
options["trace"],
|
|
options["dev"],
|
|
options["rich_traceback"],
|
|
options["rich_traceback_theme"],
|
|
)
|
|
|
|
loader = ConfigLoader(config_path=USER_CONFIG_PATH)
|
|
config = AppConfig.model_validate({}) if options["no_config"] else loader.load()
|
|
|
|
# update app config with command line parameters
|
|
for param_name, param_value in ctx.params.items():
|
|
source = ctx.get_parameter_source(param_name)
|
|
if (
|
|
source == ParameterSource.ENVIRONMENT
|
|
or source == ParameterSource.COMMANDLINE
|
|
):
|
|
parameter = None
|
|
for param in ctx.command.params:
|
|
if param.name == param_name:
|
|
parameter = param
|
|
break
|
|
if (
|
|
parameter
|
|
and hasattr(parameter, "model_name")
|
|
and hasattr(parameter, "field_name")
|
|
):
|
|
model_name = getattr(parameter, "model_name")
|
|
field_name = getattr(parameter, "field_name")
|
|
if hasattr(config, model_name):
|
|
model_instance = getattr(config, model_name)
|
|
if hasattr(model_instance, field_name):
|
|
setattr(model_instance, field_name, param_value)
|
|
ctx.obj = config
|