2021-11-08 11:52:41 -05:00
|
|
|
from functools import update_wrapper
|
|
|
|
import typing as t
|
|
|
|
|
2021-11-05 13:36:33 -04:00
|
|
|
import click
|
|
|
|
|
2021-11-08 11:52:41 -05:00
|
|
|
from aprsd import config as aprsd_config
|
2022-07-07 11:02:43 -04:00
|
|
|
from aprsd.logging import log
|
2022-07-20 08:43:57 -04:00
|
|
|
from aprsd.utils import trace
|
2021-11-08 11:52:41 -05:00
|
|
|
|
|
|
|
|
|
|
|
F = t.TypeVar("F", bound=t.Callable[..., t.Any])
|
|
|
|
|
|
|
|
common_options = [
|
|
|
|
click.option(
|
|
|
|
"--loglevel",
|
|
|
|
default="INFO",
|
|
|
|
show_default=True,
|
|
|
|
type=click.Choice(
|
|
|
|
["CRITICAL", "ERROR", "WARNING", "INFO", "DEBUG"],
|
|
|
|
case_sensitive=False,
|
|
|
|
),
|
|
|
|
show_choices=True,
|
|
|
|
help="The log level to use for aprsd.log",
|
|
|
|
),
|
|
|
|
click.option(
|
|
|
|
"-c",
|
|
|
|
"--config",
|
|
|
|
"config_file",
|
|
|
|
show_default=True,
|
|
|
|
default=aprsd_config.DEFAULT_CONFIG_FILE,
|
|
|
|
help="The aprsd config file to use for options.",
|
|
|
|
),
|
|
|
|
click.option(
|
|
|
|
"--quiet",
|
|
|
|
is_flag=True,
|
|
|
|
default=False,
|
|
|
|
help="Don't log to stdout",
|
|
|
|
),
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
|
|
def add_options(options):
|
|
|
|
def _add_options(func):
|
|
|
|
for option in reversed(options):
|
|
|
|
func = option(func)
|
|
|
|
return func
|
|
|
|
return _add_options
|
|
|
|
|
|
|
|
|
|
|
|
def process_standard_options(f: F) -> F:
|
|
|
|
def new_func(*args, **kwargs):
|
|
|
|
ctx = args[0]
|
|
|
|
ctx.ensure_object(dict)
|
|
|
|
ctx.obj["loglevel"] = kwargs["loglevel"]
|
|
|
|
ctx.obj["config_file"] = kwargs["config_file"]
|
|
|
|
ctx.obj["quiet"] = kwargs["quiet"]
|
|
|
|
ctx.obj["config"] = aprsd_config.parse_config(kwargs["config_file"])
|
|
|
|
log.setup_logging(
|
2022-02-21 16:04:33 -05:00
|
|
|
ctx.obj["config"],
|
|
|
|
ctx.obj["loglevel"],
|
2021-11-08 11:52:41 -05:00
|
|
|
ctx.obj["quiet"],
|
|
|
|
)
|
2022-07-20 08:43:57 -04:00
|
|
|
if ctx.obj["config"]["aprsd"].get("trace", False):
|
|
|
|
trace.setup_tracing(["method", "api"])
|
2021-11-08 11:52:41 -05:00
|
|
|
|
|
|
|
del kwargs["loglevel"]
|
|
|
|
del kwargs["config_file"]
|
|
|
|
del kwargs["quiet"]
|
|
|
|
return f(*args, **kwargs)
|
|
|
|
|
|
|
|
return update_wrapper(t.cast(F, new_func), f)
|
2021-11-10 11:52:51 -05:00
|
|
|
|
|
|
|
|
|
|
|
def process_standard_options_no_config(f: F) -> F:
|
|
|
|
"""Use this as a decorator when config isn't needed."""
|
|
|
|
def new_func(*args, **kwargs):
|
|
|
|
ctx = args[0]
|
|
|
|
ctx.ensure_object(dict)
|
|
|
|
ctx.obj["loglevel"] = kwargs["loglevel"]
|
|
|
|
ctx.obj["config_file"] = kwargs["config_file"]
|
|
|
|
ctx.obj["quiet"] = kwargs["quiet"]
|
|
|
|
log.setup_logging_no_config(
|
|
|
|
ctx.obj["loglevel"],
|
|
|
|
ctx.obj["quiet"],
|
|
|
|
)
|
|
|
|
|
|
|
|
del kwargs["loglevel"]
|
|
|
|
del kwargs["config_file"]
|
|
|
|
del kwargs["quiet"]
|
|
|
|
return f(*args, **kwargs)
|
|
|
|
|
|
|
|
return update_wrapper(t.cast(F, new_func), f)
|