forked from goodboy/tractor
Start `.devx.cli` extensions for pop CLI frameworks
Starting of with just a `typer` (and thus transitively `click`) `typer.Typer.callback` hook which allows passthrough of the `--ll <loglevel: str>` and `--pdb <debug_mode: bool>` flags for use when building CLIs that use the runtime Bo Still needs lotsa refinement and obviously better docs but, the doc string for `load_runtime_vars()` shows how to use the underlying `.devx._debug.open_crash_handler()` via a wrapper that can be passed the `--pdb` flag and then enable debug mode throughout the entire actor system.multihomed
parent
fa9a9cfb1d
commit
7bed470f5c
|
@ -27,8 +27,10 @@ from functools import (
|
||||||
partial,
|
partial,
|
||||||
cached_property,
|
cached_property,
|
||||||
)
|
)
|
||||||
from contextlib import asynccontextmanager as acm
|
from contextlib import (
|
||||||
from contextlib import contextmanager as cm
|
asynccontextmanager as acm,
|
||||||
|
contextmanager as cm,
|
||||||
|
)
|
||||||
from typing import (
|
from typing import (
|
||||||
Any,
|
Any,
|
||||||
Callable,
|
Callable,
|
||||||
|
|
|
@ -0,0 +1,149 @@
|
||||||
|
# tractor: structured concurrent "actors".
|
||||||
|
# Copyright 2018-eternity Tyler Goodlet.
|
||||||
|
|
||||||
|
# This program is free software: you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU Affero General Public License as published by
|
||||||
|
# the Free Software Foundation, either version 3 of the License, or
|
||||||
|
# (at your option) any later version.
|
||||||
|
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU Affero General Public License for more details.
|
||||||
|
|
||||||
|
# You should have received a copy of the GNU Affero General Public License
|
||||||
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
"""
|
||||||
|
CLI framework extensions for hacking on the actor runtime.
|
||||||
|
|
||||||
|
Currently popular frameworks supported are:
|
||||||
|
|
||||||
|
- `typer` via the `@callback` API
|
||||||
|
|
||||||
|
"""
|
||||||
|
from __future__ import annotations
|
||||||
|
from contextlib import (
|
||||||
|
# asynccontextmanager as acm,
|
||||||
|
nullcontext,
|
||||||
|
contextmanager as cm,
|
||||||
|
)
|
||||||
|
from typing import (
|
||||||
|
Any,
|
||||||
|
Callable,
|
||||||
|
)
|
||||||
|
from typing_extensions import Annotated
|
||||||
|
|
||||||
|
import typer
|
||||||
|
|
||||||
|
|
||||||
|
from ._debug import open_crash_handler
|
||||||
|
|
||||||
|
|
||||||
|
_runtime_vars: dict[str, Any] = {}
|
||||||
|
|
||||||
|
|
||||||
|
def load_runtime_vars(
|
||||||
|
ctx: typer.Context,
|
||||||
|
callback: Callable,
|
||||||
|
pdb: bool = False, # --pdb
|
||||||
|
ll: Annotated[
|
||||||
|
str,
|
||||||
|
typer.Option(
|
||||||
|
'--loglevel',
|
||||||
|
'-l',
|
||||||
|
help='BigD logging level',
|
||||||
|
),
|
||||||
|
] = 'cancel', # -l info
|
||||||
|
):
|
||||||
|
'''
|
||||||
|
Maybe engage crash handling with `pdbp` when code inside
|
||||||
|
a `typer` CLI endpoint cmd raises.
|
||||||
|
|
||||||
|
To use this callback simply take your `app = typer.Typer()` instance
|
||||||
|
and decorate this function with it like so:
|
||||||
|
|
||||||
|
.. code:: python
|
||||||
|
|
||||||
|
from tractor.devx import cli
|
||||||
|
|
||||||
|
app = typer.Typer()
|
||||||
|
|
||||||
|
# manual decoration to hook into `click`'s context system!
|
||||||
|
cli.load_runtime_vars = app.callback(
|
||||||
|
invoke_without_command=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
And then you can use the now augmented `click` CLI context as so,
|
||||||
|
|
||||||
|
.. code:: python
|
||||||
|
|
||||||
|
@app.command(
|
||||||
|
context_settings={
|
||||||
|
"allow_extra_args": True,
|
||||||
|
"ignore_unknown_options": True,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
def my_cli_cmd(
|
||||||
|
ctx: typer.Context,
|
||||||
|
):
|
||||||
|
rtvars: dict = ctx.runtime_vars
|
||||||
|
pdb: bool = rtvars['pdb']
|
||||||
|
|
||||||
|
with tractor.devx.cli.maybe_open_crash_handler(pdb=pdb):
|
||||||
|
trio.run(
|
||||||
|
partial(
|
||||||
|
my_tractor_main_task_func,
|
||||||
|
debug_mode=pdb,
|
||||||
|
loglevel=rtvars['ll'],
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
which will enable log level and debug mode globally for the entire
|
||||||
|
`tractor` + `trio` runtime thereafter!
|
||||||
|
|
||||||
|
Bo
|
||||||
|
|
||||||
|
'''
|
||||||
|
global _runtime_vars
|
||||||
|
_runtime_vars |= {
|
||||||
|
'pdb': pdb,
|
||||||
|
'll': ll,
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx.runtime_vars: dict[str, Any] = _runtime_vars
|
||||||
|
print(
|
||||||
|
f'`typer` sub-cmd: {ctx.invoked_subcommand}\n'
|
||||||
|
f'`tractor` runtime vars: {_runtime_vars}'
|
||||||
|
)
|
||||||
|
|
||||||
|
# XXX NOTE XXX: hackzone.. if no sub-cmd is specified (the
|
||||||
|
# default if the user just invokes `bigd`) then we simply
|
||||||
|
# invoke the sole `_bigd()` cmd passing in the "parent"
|
||||||
|
# typer.Context directly to that call since we're treating it
|
||||||
|
# as a "non sub-command" or wtv..
|
||||||
|
# TODO: ideally typer would have some kinda built-in way to get
|
||||||
|
# this behaviour without having to construct and manually
|
||||||
|
# invoke our own cmd..
|
||||||
|
if (
|
||||||
|
ctx.invoked_subcommand is None
|
||||||
|
or ctx.invoked_subcommand == callback.__name__
|
||||||
|
):
|
||||||
|
cmd: typer.core.TyperCommand = typer.core.TyperCommand(
|
||||||
|
name='bigd',
|
||||||
|
callback=callback,
|
||||||
|
)
|
||||||
|
ctx.params = {'ctx': ctx}
|
||||||
|
cmd.invoke(ctx)
|
||||||
|
|
||||||
|
|
||||||
|
@cm
|
||||||
|
def maybe_open_crash_handler(pdb: bool = False):
|
||||||
|
# if the --pdb flag is passed always engage
|
||||||
|
# the pdb REPL on any crashes B)
|
||||||
|
rtctx = nullcontext
|
||||||
|
if pdb:
|
||||||
|
rtctx = open_crash_handler
|
||||||
|
|
||||||
|
with rtctx():
|
||||||
|
yield
|
Loading…
Reference in New Issue