Compare commits

...

6 Commits

Author SHA1 Message Date
Gud Boi 3727ce9e6f Bump lockfile for `xonsh>=0.23.8` release 2026-06-24 16:20:50 -04:00
Gud Boi 90c316f90c Code-style, couple newline/ws tweaks
(cherry picked from commit 8526985c97)
(cherry picked from commit 0952b33a9e)
2026-06-24 16:20:50 -04:00
Gud Boi 6db796ed83 Pin to latest `xonsh` release
(cherry picked from commit c4cad921b9)
(cherry picked from commit ade15b4204)
2026-06-24 16:20:50 -04:00
Gud Boi eb7ef3f0bf Hoist proc-title prefix to `_def_prefix` const
Make the sub-actor proc-title prefix a single
authoritative constant (`_proctitle._def_prefix`) so
the reap-recognition markers and `xontrib` banner pick
it up automatically — one place to flip the prefix
shape going fwd.

Deats,
- `_proctitle._def_prefix: str = '_subactor'`. New
  module-level const consumed by everything that needs
  to know the prefix.
- `set_actor_proctitle(actor, prefix=_def_prefix)`:
  takes an explicit `prefix` arg (default = the const)
  so callers can override per-spawn if they want.
- Default proc-title format:
  `'tractor[<reprol>]'` → `f'{prefix}[<reprol>]'`
  i.e. `_subactor[<reprol>]` by default.
- `_testing/_reap.py`: cmdline + comm markers source
  the prefix from `_proctitle._def_prefix` instead of
  the hardcoded `'tractor['`. So
  `_is_tractor_subactor()` tracks the const
  automatically.
- `xontrib/tractor_diag.xsh`: `acli.reap` orphan-mode
  banner now interpolates the
  `_TRACTOR_PROC_CMDLINE_MARKERS` tuple directly so
  the human-readable mode line stays in sync if the
  prefix shape changes again.

(this commit msg was generated in some part by [`claude-code`][claude-code-gh])
[claude-code-gh]: https://github.com/anthropics/claude-code

(cherry picked from commit 3a45dbd503)
(cherry picked from commit fd8d39c0ce)
2026-06-24 16:20:50 -04:00
Gud Boi bb0b2ffa3c Add `add_log_level()` factory + register `IO`=21
Follow-up to f595acc7 (`supervise_run_process`) which
called `log.io(...)` for std-stream relay assuming an
`IO=21` level existed. Add the registration via a new
factory + tests covering both the factory and the new
level.

`add_log_level()` factory,
- One call wires the four (otherwise hand-synced) pieces:
  - `CUSTOM_LEVELS[NAME]` — drives the `stacklevel` bump
    in `StackLevelAdapter.log()` + `get_logger()`'s
    per-level audit.
  - `logging.addLevelName()` — stdlib name registration.
  - `STD_PALETTE[NAME]` + `BOLD_PALETTE['bold'][NAME]` —
    color entries consumed by `get_console_log()`'s
    `ColoredFormatter` build.
  - Same-named (lowercase) emit method bound on
    `StackLevelAdapter` so `log.<name>('msg')` works +
    `get_logger()`'s per-level method audit passes.
- Idempotent: re-registering an existing name is a
  no-op-ish refresh that won't clobber an already-bound
  method.
- Method binding uses a default-arg `_level=value` so
  the level int is captured (not late-bound across
  multiple registrations).

`IO=21` level (first user),
- Purple. Used by `tractor.trionics._subproc`'s
  std-stream relay (see f595acc7).
- Value 21 picked to sit just ABOVE stdlib `INFO`=20 so
  it's SHOWN BY DEFAULT at usual `info`/`devx` console
  levels — a `runtime`=15 relay would be silently
  filtered (footgun for daemon supervisors whose whole
  point is visibility). Still distinctly labeled +
  filterable.

Tests (`tests/test_log_sys.py`),
- `test_io_custom_level_registered`: validates the IO
  level is fully wired (`CUSTOM_LEVELS`, `addLevelName`,
  both palettes, `StackLevelAdapter.io()` callable);
  emits a record + sanity-asserts `21 >= INFO(20)`.
- `test_add_log_level_pluggable`: registers a fresh
  `XLVL=19` (cyan) via `add_log_level()`, asserts all
  four wires + the bound `xlog.xlvl()` emit, then
  try/finally cleans up the module-global mutations so
  later `get_logger()` audits don't trip on a
  half-removed level.

(this patch was generated in some part by [`claude-code`][claude-code-gh])
[claude-code-gh]: https://github.com/anthropics/claude-code

(cherry picked from commit 7bd7dd50c7)
(cherry picked from commit 93558fe3c9)
2026-06-24 16:20:50 -04:00
Bd fd41f88b2b
Merge pull request #464 from goodboy/trionics_start_or_cancel
Add `start_or_cancel()` to `trionics._taskc`
2026-06-24 15:08:20 -04:00
10 changed files with 168 additions and 32 deletions

View File

@ -104,7 +104,7 @@ testing = [
repl = [ repl = [
"pyperclip>=1.9.0", "pyperclip>=1.9.0",
"prompt-toolkit>=3.0.50", "prompt-toolkit>=3.0.50",
"xonsh>=0.23.0", "xonsh>=0.23.8",
"psutil>=7.0.0", "psutil>=7.0.0",
] ]
lint = [ lint = [

View File

@ -8,9 +8,9 @@ after `Actor` construction, so any spawned sub-actor process
should: should:
- have `argv[0]` (== `/proc/<pid>/cmdline`) start with - have `argv[0]` (== `/proc/<pid>/cmdline`) start with
`tractor[<aid.reprol()>]` `<_def_prefix>[<aid.reprol()>]` (currently `_subactor[]`)
- have `/proc/<pid>/comm` start with `tractor[` (kernel - have `/proc/<pid>/comm` start with `<_def_prefix>[`
truncates to ~15 bytes) (kernel truncates to ~15 bytes)
- be detected as a tractor sub-actor by - be detected as a tractor sub-actor by
`_is_tractor_subactor(pid)` via the cmdline marker. `_is_tractor_subactor(pid)` via the cmdline marker.
@ -27,7 +27,10 @@ import trio
import tractor import tractor
from tractor.runtime._runtime import Actor from tractor.runtime._runtime import Actor
from tractor.devx._proctitle import set_actor_proctitle from tractor.devx._proctitle import (
set_actor_proctitle,
_def_prefix,
)
from tractor._testing._reap import ( from tractor._testing._reap import (
_is_tractor_subactor, _is_tractor_subactor,
_read_cmdline, _read_cmdline,
@ -41,8 +44,9 @@ _non_linux: bool = platform.system() != 'Linux'
def test_set_actor_proctitle_format(): def test_set_actor_proctitle_format():
''' '''
`set_actor_proctitle()` returns the canonical `set_actor_proctitle()` returns the canonical
`tractor[<aid.reprol()>]` form and actually mutates `<_def_prefix>[<aid.reprol()>]` form (currently
the running proc's title. `_subactor[]`) and actually mutates the running
proc's title.
''' '''
pytest.importorskip( pytest.importorskip(
@ -60,12 +64,14 @@ def test_set_actor_proctitle_format():
) )
title: str = set_actor_proctitle(actor) title: str = set_actor_proctitle(actor)
# canonical wrapping: `tractor[<aid.reprol()>]`. We # canonical wrapping: `<_def_prefix>[<aid.reprol()>]`.
# compare against the runtime-computed `reprol()` # We source BOTH the prefix (`_def_prefix`) and the
# rather than a hard-coded value so the test stays # runtime-computed `reprol()` rather than hard-coding,
# decoupled from `Aid.reprol()`'s internal format # so the test stays decoupled from the prefix shape
# (currently `<name>@<pid>`, but could evolve). # (flipped to `_subactor` in `3a45dbd5`) AND from
expected: str = f'tractor[{actor.aid.reprol()}]' # `Aid.reprol()`'s internal format (currently
# `<name>@<pid>`, but could evolve).
expected: str = f'{_def_prefix}[{actor.aid.reprol()}]'
assert title == expected assert title == expected
# sanity: the actor's name must be in the title # sanity: the actor's name must be in the title
# somewhere (so a future `reprol()` change that # somewhere (so a future `reprol()` change that
@ -140,15 +146,17 @@ def test_subactor_proctitle_visible_via_proc():
) )
pid, info = matched[0] pid, info = matched[0]
# canonical proctitle prefix in cmdline (full form) # canonical proctitle prefix in cmdline (full form);
assert info['cmdline'].startswith('tractor[proctitle_boi@'), ( # prefix sourced from `_def_prefix` so it tracks the
f'cmdline missing `tractor[proctitle_boi@…]` prefix: ' # `3a45dbd5` flip (`tractor[` -> `_subactor[`).
assert info['cmdline'].startswith(f'{_def_prefix}[proctitle_boi@'), (
f'cmdline missing `{_def_prefix}[proctitle_boi@…]` prefix: '
f'{info["cmdline"]!r}' f'{info["cmdline"]!r}'
) )
# comm is kernel-truncated to ~15 bytes — just check the # comm is kernel-truncated to ~15 bytes — just check the
# `tractor[` prefix made it. # `<_def_prefix>[` prefix made it.
assert info['comm'].startswith('tractor['), ( assert info['comm'].startswith(f'{_def_prefix}['), (
f'comm missing `tractor[` prefix: {info["comm"]!r}' f'comm missing `{_def_prefix}[` prefix: {info["comm"]!r}'
) )
# intrinsic-signal detector should match. # intrinsic-signal detector should match.
assert info['is_tractor'] is True assert info['is_tractor'] is True

View File

@ -162,6 +162,66 @@ def test_implicit_mod_name_applied_for_child(
assert submod.log.logger in sub_logs assert submod.log.logger in sub_logs
def test_io_custom_level_registered():
'''
The `IO`(21) level (registered via `add_log_level()` at
import, for `tractor.trionics._subproc`'s std-stream relay)
is fully wired and SHOWN BY DEFAULT at `info`-level consoles
since `21 >= INFO(20)`.
'''
import logging
assert log.CUSTOM_LEVELS.get('IO') == 21
assert logging.getLevelName(21) == 'IO'
assert log.STD_PALETTE.get('IO')
assert log.BOLD_PALETTE['bold'].get('IO')
iolog = log.get_logger('io_lvl_test')
assert callable(getattr(iolog, 'io', None))
# emit must not raise
iolog.io('hello from the IO level')
# 21 >= INFO(20) -> shown when console set to `info`
assert 21 >= logging.INFO
def test_add_log_level_pluggable():
'''
`add_log_level()` is the single pluggable entry-point: one
call wires `CUSTOM_LEVELS` + `addLevelName` + both palettes +
a same-named `StackLevelAdapter` emit method (so
`get_logger()`'s per-level audit passes).
'''
import logging
name: str = 'XLVL'
val: int = 19
try:
log.add_log_level(name, val, 'cyan')
assert log.CUSTOM_LEVELS[name] == val
assert logging.getLevelName(val) == name
assert log.STD_PALETTE[name] == 'cyan'
assert log.BOLD_PALETTE['bold'][name] == 'bold_cyan'
# the audit in `get_logger()` (asserts a method per
# `CUSTOM_LEVELS` entry) must still pass.
xlog = log.get_logger('xlvl_test')
emit = getattr(xlog, name.lower(), None)
assert callable(emit)
emit('hello from a plugged-in level')
finally:
# best-effort cleanup of our module-global mutations so
# later `get_logger()` audits don't see a half-removed
# level.
log.CUSTOM_LEVELS.pop(name, None)
log.STD_PALETTE.pop(name, None)
log.BOLD_PALETTE['bold'].pop(name, None)
if hasattr(log.StackLevelAdapter, name.lower()):
delattr(log.StackLevelAdapter, name.lower())
# TODO, moar tests against existing feats: # TODO, moar tests against existing feats:
# ------ - ------ # ------ - ------
# - [ ] color settings? # - [ ] color settings?

View File

@ -155,7 +155,6 @@ async def maybe_block_bp(
os.environ.pop('PYTHONBREAKPOINT', None) os.environ.pop('PYTHONBREAKPOINT', None)
@acm @acm
async def open_root_actor( async def open_root_actor(
*, *,
@ -186,6 +185,7 @@ async def open_root_actor(
# enables the multi-process debugger support # enables the multi-process debugger support
debug_mode: bool = False, debug_mode: bool = False,
maybe_enable_greenback: bool = False, # `.pause_from_sync()/breakpoint()` support maybe_enable_greenback: bool = False, # `.pause_from_sync()/breakpoint()` support
# ^XXX NOTE^ the perf implications of use, # ^XXX NOTE^ the perf implications of use,
# https://greenback.readthedocs.io/en/latest/principle.html#performance # https://greenback.readthedocs.io/en/latest/principle.html#performance
enable_stack_on_sig: bool = False, enable_stack_on_sig: bool = False,

View File

@ -90,7 +90,6 @@ keys are caller-defined).
''' '''
from __future__ import annotations from __future__ import annotations
import os import os
import pathlib import pathlib
import re import re
@ -99,6 +98,9 @@ import stat
import sys import sys
import time import time
from tractor.devx import _proctitle
# `/dev/shm` is the POSIX-shm filesystem on Linux + FreeBSD. # `/dev/shm` is the POSIX-shm filesystem on Linux + FreeBSD.
# macOS uses `shm_open` syscalls without a fs-visible path, # macOS uses `shm_open` syscalls without a fs-visible path,
# so the shm helpers refuse to run there. # so the shm helpers refuse to run there.
@ -230,9 +232,9 @@ def _read_comm(pid: int) -> str:
# while `cmdline` for zombies often reads as empty. # while `cmdline` for zombies often reads as empty.
_TRACTOR_PROC_CMDLINE_MARKERS: tuple[str, ...] = ( _TRACTOR_PROC_CMDLINE_MARKERS: tuple[str, ...] = (
'tractor._child', 'tractor._child',
'tractor[', _proctitle._def_prefix,
) )
_TRACTOR_PROC_COMM_MARKER: str = 'tractor[' _TRACTOR_PROC_COMM_MARKER: str = _proctitle._def_prefix
def _is_tractor_subactor(pid: int) -> bool: def _is_tractor_subactor(pid: int) -> bool:

View File

@ -24,7 +24,10 @@ which" at a glance without needing to read full
`/proc/<pid>/cmdline`. `/proc/<pid>/cmdline`.
Format: Format:
``tractor[<aid.reprol()>]`` e.g. ``tractor[doggy@1027301b]`` ``<_def_prefix>[<aid.reprol()>]`` e.g. ``_subactor[doggy@1027301b]``
(prefix from the `_def_prefix` const, flipped `tractor` ->
`_subactor` so sub-actor procs are visually distinct from the
root in `ps`/`htop` and the reap-recognition markers.)
Uses the canonical `Aid.reprol()` form Uses the canonical `Aid.reprol()` form
(``<name>@<uuid_short>``) so the proc-title matches the (``<name>@<uuid_short>``) so the proc-title matches the
@ -52,7 +55,13 @@ except ImportError:
_stp = None _stp = None
def set_actor_proctitle(actor: 'Actor') -> str | None: _def_prefix: str = '_subactor'
def set_actor_proctitle(
actor: 'Actor',
prefix: str = _def_prefix,
) -> str | None:
''' '''
Set the calling process's proc-title to identify it as a Set the calling process's proc-title to identify it as a
tractor sub-actor. tractor sub-actor.
@ -69,6 +78,6 @@ def set_actor_proctitle(actor: 'Actor') -> str | None:
if _stp is None: if _stp is None:
return None return None
title: str = f'tractor[{actor.aid.reprol()}]' title: str = f'{prefix}[{actor.aid.reprol()}]'
_stp.setproctitle(title) _stp.setproctitle(title)
return title return title

View File

@ -262,6 +262,63 @@ class StackLevelAdapter(LoggerAdapter):
) )
def add_log_level(
name: str,
value: int,
color: str = 'white',
) -> None:
'''
Register a new custom log level with `tractor`'s logging
machinery in ONE call the single pluggable entry-point that
keeps the (otherwise hand-synced) pieces consistent:
- `CUSTOM_LEVELS[name]` (drives the `stacklevel` bump in
`StackLevelAdapter.log()` + the `get_logger()` audit).
- `logging.addLevelName()` registration.
- `STD_PALETTE`/`BOLD_PALETTE` color entries (consumed when
`get_console_log()` builds its `ColoredFormatter`).
- a same-named (lowercase) emit method bound on
`StackLevelAdapter` so `log.<name>('msg')` works (and so
`get_logger()`'s per-level method audit passes).
Idempotent: re-registering an existing name is a no-op-ish
refresh (won't clobber an already-bound method).
'''
name_up: str = name.upper()
name_lo: str = name.lower()
CUSTOM_LEVELS[name_up] = value
logging.addLevelName(value, name_up)
STD_PALETTE[name_up] = color
BOLD_PALETTE['bold'][name_up] = f'bold_{color}'
if not hasattr(StackLevelAdapter, name_lo):
# bind via default-arg so `value` is captured (not
# late-bound); delegates to `.log()` exactly like the
# hand-written level methods above.
def _emit(
self,
msg: str,
*,
_level: int = value,
) -> None:
return self.log(_level, msg)
_emit.__name__ = name_lo
_emit.__qualname__ = f'StackLevelAdapter.{name_lo}'
setattr(StackLevelAdapter, name_lo, _emit)
# `IO`: child-subproc std-stream relay (see
# `tractor.trionics._subproc`). Value 21 sits just ABOVE
# `INFO`(20) so it's SHOWN BY DEFAULT at the usual `info`/`devx`
# console levels (a `runtime`(15) relay would be silently
# filtered) yet still distinctly labelled/colored + separately
# filterable.
add_log_level('IO', 21, 'purple')
# TODO IDEAs: # TODO IDEAs:
# -[ ] move to `.devx.pformat`? # -[ ] move to `.devx.pformat`?
# -[ ] do per task-name and actor-name color coding # -[ ] do per task-name and actor-name color coding

View File

@ -139,7 +139,7 @@ _RUNTIME_VARS_DEFAULTS: dict[str, Any] = {
# `debug_mode: bool` settings # `debug_mode: bool` settings
'_debug_mode': False, # bool '_debug_mode': False, # bool
'repl_fixture': False, # |AbstractContextManager[bool] 'repl_fixture': False, # |AbstractContextManager[bool]
'use_greenback': False, # `.pause_from_sync()`/`breakpoint()` 'use_greenback': False, # `.pause_from_sync()`/`breakpoint()`
'use_stackscope': False, # trio-task-stack dumps on SIGUSR1 'use_stackscope': False, # trio-task-stack dumps on SIGUSR1

View File

@ -797,7 +797,7 @@ dev = [
{ name = "pytest-timeout", specifier = ">=2.3" }, { name = "pytest-timeout", specifier = ">=2.3" },
{ name = "stackscope", specifier = ">=0.2.2,<0.3" }, { name = "stackscope", specifier = ">=0.2.2,<0.3" },
{ name = "typing-extensions", specifier = ">=4.14.1" }, { name = "typing-extensions", specifier = ">=4.14.1" },
{ name = "xonsh", specifier = ">=0.23.0" }, { name = "xonsh", specifier = ">=0.23.8" },
] ]
devx = [ devx = [
{ name = "stackscope", specifier = ">=0.2.2,<0.3" }, { name = "stackscope", specifier = ">=0.2.2,<0.3" },
@ -809,7 +809,7 @@ repl = [
{ name = "prompt-toolkit", specifier = ">=3.0.50" }, { name = "prompt-toolkit", specifier = ">=3.0.50" },
{ name = "psutil", specifier = ">=7.0.0" }, { name = "psutil", specifier = ">=7.0.0" },
{ name = "pyperclip", specifier = ">=1.9.0" }, { name = "pyperclip", specifier = ">=1.9.0" },
{ name = "xonsh", specifier = ">=0.23.0" }, { name = "xonsh", specifier = ">=0.23.8" },
] ]
subints = [{ name = "msgspec", marker = "python_full_version >= '3.14'", specifier = ">=0.21.0" }] subints = [{ name = "msgspec", marker = "python_full_version >= '3.14'", specifier = ">=0.21.0" }]
sync-pause = [{ name = "greenback", marker = "python_full_version == '3.13.*'", specifier = ">=1.2.1,<2" }] sync-pause = [{ name = "greenback", marker = "python_full_version == '3.13.*'", specifier = ">=1.2.1,<2" }]

View File

@ -488,6 +488,7 @@ def _tractor_reap(args):
reap, reap,
reap_shm, reap_shm,
reap_uds, reap_uds,
_TRACTOR_PROC_CMDLINE_MARKERS,
) )
rc: int = 0 rc: int = 0
@ -500,9 +501,8 @@ def _tractor_reap(args):
else: else:
pids = find_orphans() pids = find_orphans()
mode = ( mode = (
'orphans (PPid==1, intrinsic ' f'orphans (PPid==1, intrinsic '
'cmdline/comm match — `tractor[…]` or ' f'cmdline/comm match — {_TRACTOR_PROC_CMDLINE_MARKERS}'
'`tractor._child`)'
) )
if not pids: if not pids: