mirror of
https://github.com/jeertmans/manim-slides.git
synced 2025-05-18 03:05:21 +08:00
2f0453c9a6
* chore(lib): add missing type hints and mypy is happy! Closes #34 * fix(ci): add missing dep for mypy
83 lines
2.1 KiB
Python
83 lines
2.1 KiB
Python
from typing import Any, Callable
|
|
|
|
import click
|
|
from click import Context, Parameter
|
|
|
|
from .defaults import CONFIG_PATH, FOLDER_PATH
|
|
from .manim import logger
|
|
|
|
F = Callable[..., Any]
|
|
Wrapper = Callable[[F], F]
|
|
|
|
|
|
def config_path_option(function: F) -> F:
|
|
"""Wraps a function to add configuration path option."""
|
|
wrapper: Wrapper = click.option(
|
|
"-c",
|
|
"--config",
|
|
"config_path",
|
|
metavar="FILE",
|
|
default=CONFIG_PATH,
|
|
type=click.Path(dir_okay=False),
|
|
help="Set path to configuration file.",
|
|
show_default=True,
|
|
)
|
|
return wrapper(function)
|
|
|
|
|
|
def config_options(function: F) -> F:
|
|
"""Wraps a function to add configuration options."""
|
|
function = config_path_option(function)
|
|
function = click.option(
|
|
"-f", "--force", is_flag=True, help="Overwrite any existing configuration file."
|
|
)(function)
|
|
function = click.option(
|
|
"-m",
|
|
"--merge",
|
|
is_flag=True,
|
|
help="Merge any existing configuration file with the new configuration.",
|
|
)(function)
|
|
return function
|
|
|
|
|
|
def verbosity_option(function: F) -> F:
|
|
"""Wraps a function to add verbosity option."""
|
|
|
|
def callback(ctx: Context, param: Parameter, value: bool) -> None:
|
|
|
|
if not value or ctx.resilient_parsing:
|
|
return
|
|
|
|
logger.setLevel(value)
|
|
|
|
wrapper: Wrapper = click.option(
|
|
"-v",
|
|
"--verbosity",
|
|
type=click.Choice(
|
|
["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
|
|
case_sensitive=False,
|
|
),
|
|
help="Verbosity of CLI output",
|
|
default=None,
|
|
expose_value=False,
|
|
envvar="MANIM_SLIDES_VERBOSITY",
|
|
show_envvar=True,
|
|
callback=callback,
|
|
)
|
|
|
|
return wrapper(function)
|
|
|
|
|
|
def folder_path_option(function: F) -> F:
|
|
"""Wraps a function to add folder path option."""
|
|
wrapper: Wrapper = click.option(
|
|
"--folder",
|
|
metavar="DIRECTORY",
|
|
default=FOLDER_PATH,
|
|
type=click.Path(exists=True, file_okay=False),
|
|
help="Set slides folder.",
|
|
show_default=True,
|
|
)
|
|
|
|
return wrapper(function)
|