Compare commits
13 Commits
2c1c89b2f4
...
0bb66f5755
Author | SHA1 | Date |
---|---|---|
|
0bb66f5755 | |
|
ee1596da4d | |
|
9262beb0c4 | |
|
d830899a9b | |
|
06d47d9376 | |
|
11e1c8d14f | |
|
0f3d714d38 | |
|
ce89cabe1f | |
|
3b8119ff81 | |
|
039de4dce7 | |
|
8dc804279c | |
|
54b9a37430 | |
|
bc9a9a66bc |
|
@ -4,15 +4,9 @@ import trio
|
||||||
|
|
||||||
async def breakpoint_forever():
|
async def breakpoint_forever():
|
||||||
"Indefinitely re-enter debugger in child actor."
|
"Indefinitely re-enter debugger in child actor."
|
||||||
try:
|
while True:
|
||||||
while True:
|
yield 'yo'
|
||||||
yield 'yo'
|
await tractor.breakpoint()
|
||||||
await tractor.breakpoint()
|
|
||||||
except BaseException:
|
|
||||||
tractor.log.get_console_log().exception(
|
|
||||||
'Cancelled while trying to enter pause point!'
|
|
||||||
)
|
|
||||||
raise
|
|
||||||
|
|
||||||
|
|
||||||
async def name_error():
|
async def name_error():
|
||||||
|
@ -25,7 +19,7 @@ async def main():
|
||||||
"""
|
"""
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
debug_mode=True,
|
debug_mode=True,
|
||||||
loglevel='cancel',
|
loglevel='error',
|
||||||
) as n:
|
) as n:
|
||||||
|
|
||||||
p0 = await n.start_actor('bp_forever', enable_modules=[__name__])
|
p0 = await n.start_actor('bp_forever', enable_modules=[__name__])
|
||||||
|
|
|
@ -45,7 +45,6 @@ async def spawn_until(depth=0):
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
# TODO: notes on the new boxed-relayed errors through proxy actors
|
|
||||||
async def main():
|
async def main():
|
||||||
"""The main ``tractor`` routine.
|
"""The main ``tractor`` routine.
|
||||||
|
|
||||||
|
|
|
@ -38,7 +38,6 @@ async def main():
|
||||||
"""
|
"""
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
debug_mode=True,
|
debug_mode=True,
|
||||||
# loglevel='runtime',
|
|
||||||
) as n:
|
) as n:
|
||||||
|
|
||||||
# Spawn both actors, don't bother with collecting results
|
# Spawn both actors, don't bother with collecting results
|
||||||
|
|
|
@ -23,6 +23,5 @@ async def main():
|
||||||
n.start_soon(debug_actor.run, die)
|
n.start_soon(debug_actor.run, die)
|
||||||
n.start_soon(crash_boi.run, die)
|
n.start_soon(crash_boi.run, die)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
trio.run(main)
|
trio.run(main)
|
||||||
|
|
|
@ -2,13 +2,10 @@ import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
async def main(
|
async def main():
|
||||||
registry_addrs: tuple[str, int]|None = None
|
|
||||||
):
|
|
||||||
|
|
||||||
async with tractor.open_root_actor(
|
async with tractor.open_root_actor(
|
||||||
debug_mode=True,
|
debug_mode=True,
|
||||||
# loglevel='runtime',
|
|
||||||
):
|
):
|
||||||
while True:
|
while True:
|
||||||
await tractor.breakpoint()
|
await tractor.breakpoint()
|
||||||
|
|
|
@ -3,20 +3,17 @@ import tractor
|
||||||
|
|
||||||
|
|
||||||
async def breakpoint_forever():
|
async def breakpoint_forever():
|
||||||
'''
|
"""Indefinitely re-enter debugger in child actor.
|
||||||
Indefinitely re-enter debugger in child actor.
|
"""
|
||||||
|
|
||||||
'''
|
|
||||||
while True:
|
while True:
|
||||||
await trio.sleep(0.1)
|
await trio.sleep(0.1)
|
||||||
await tractor.pause()
|
await tractor.breakpoint()
|
||||||
|
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
|
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
debug_mode=True,
|
debug_mode=True,
|
||||||
loglevel='cancel',
|
|
||||||
) as n:
|
) as n:
|
||||||
|
|
||||||
portal = await n.run_in_actor(
|
portal = await n.run_in_actor(
|
||||||
|
|
|
@ -3,26 +3,16 @@ import tractor
|
||||||
|
|
||||||
|
|
||||||
async def name_error():
|
async def name_error():
|
||||||
getattr(doggypants) # noqa (on purpose)
|
getattr(doggypants)
|
||||||
|
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
debug_mode=True,
|
debug_mode=True,
|
||||||
# loglevel='transport',
|
) as n:
|
||||||
) as an:
|
|
||||||
|
|
||||||
# TODO: ideally the REPL arrives at this frame in the parent,
|
portal = await n.run_in_actor(name_error)
|
||||||
# ABOVE the @api_frame of `Portal.run_in_actor()` (which
|
await portal.result()
|
||||||
# should eventually not even be a portal method ... XD)
|
|
||||||
# await tractor.pause()
|
|
||||||
p: tractor.Portal = await an.run_in_actor(name_error)
|
|
||||||
|
|
||||||
# with this style, should raise on this line
|
|
||||||
await p.result()
|
|
||||||
|
|
||||||
# with this alt style should raise at `open_nusery()`
|
|
||||||
# return await p.result()
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|
|
@ -7,7 +7,7 @@ def sync_pause(
|
||||||
error: bool = False,
|
error: bool = False,
|
||||||
):
|
):
|
||||||
if use_builtin:
|
if use_builtin:
|
||||||
breakpoint(hide_tb=False)
|
breakpoint()
|
||||||
|
|
||||||
else:
|
else:
|
||||||
tractor.pause_from_sync()
|
tractor.pause_from_sync()
|
||||||
|
@ -20,20 +20,18 @@ def sync_pause(
|
||||||
async def start_n_sync_pause(
|
async def start_n_sync_pause(
|
||||||
ctx: tractor.Context,
|
ctx: tractor.Context,
|
||||||
):
|
):
|
||||||
actor: tractor.Actor = tractor.current_actor()
|
# sync to requesting peer
|
||||||
|
|
||||||
# sync to parent-side task
|
|
||||||
await ctx.started()
|
await ctx.started()
|
||||||
|
|
||||||
|
actor: tractor.Actor = tractor.current_actor()
|
||||||
print(f'entering SYNC PAUSE in {actor.uid}')
|
print(f'entering SYNC PAUSE in {actor.uid}')
|
||||||
sync_pause()
|
sync_pause()
|
||||||
print(f'back from SYNC PAUSE in {actor.uid}')
|
print(f'back from SYNC PAUSE in {actor.uid}')
|
||||||
|
|
||||||
|
|
||||||
async def main() -> None:
|
async def main() -> None:
|
||||||
|
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
# NOTE: required for pausing from sync funcs
|
|
||||||
maybe_enable_greenback=True,
|
|
||||||
debug_mode=True,
|
debug_mode=True,
|
||||||
) as an:
|
) as an:
|
||||||
|
|
||||||
|
|
|
@ -36,7 +36,6 @@ def parse_ipaddr(arg):
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
__tracebackhide__: bool = True
|
|
||||||
|
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
parser.add_argument("--uid", type=parse_uid)
|
parser.add_argument("--uid", type=parse_uid)
|
||||||
|
|
|
@ -106,7 +106,6 @@ def _trio_main(
|
||||||
Entry point for a `trio_run_in_process` subactor.
|
Entry point for a `trio_run_in_process` subactor.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
__tracebackhide__: bool = True
|
|
||||||
_state._current_actor = actor
|
_state._current_actor = actor
|
||||||
trio_main = partial(
|
trio_main = partial(
|
||||||
async_main,
|
async_main,
|
||||||
|
|
|
@ -22,10 +22,9 @@ from contextlib import asynccontextmanager
|
||||||
from functools import partial
|
from functools import partial
|
||||||
import importlib
|
import importlib
|
||||||
import logging
|
import logging
|
||||||
import os
|
|
||||||
import signal
|
import signal
|
||||||
import sys
|
import sys
|
||||||
from typing import Callable
|
import os
|
||||||
import warnings
|
import warnings
|
||||||
|
|
||||||
|
|
||||||
|
@ -79,8 +78,6 @@ 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
|
|
||||||
enable_stack_on_sig: bool = False,
|
|
||||||
|
|
||||||
# internal logging
|
# internal logging
|
||||||
loglevel: str|None = None,
|
loglevel: str|None = None,
|
||||||
|
@ -102,36 +99,19 @@ async def open_root_actor(
|
||||||
# Override the global debugger hook to make it play nice with
|
# Override the global debugger hook to make it play nice with
|
||||||
# ``trio``, see much discussion in:
|
# ``trio``, see much discussion in:
|
||||||
# https://github.com/python-trio/trio/issues/1155#issuecomment-742964018
|
# https://github.com/python-trio/trio/issues/1155#issuecomment-742964018
|
||||||
builtin_bp_handler: Callable = sys.breakpointhook
|
|
||||||
orig_bp_path: str|None = os.environ.get(
|
|
||||||
'PYTHONBREAKPOINT',
|
|
||||||
None,
|
|
||||||
)
|
|
||||||
if (
|
if (
|
||||||
debug_mode
|
await _debug.maybe_init_greenback(
|
||||||
and maybe_enable_greenback
|
|
||||||
and await _debug.maybe_init_greenback(
|
|
||||||
raise_not_found=False,
|
raise_not_found=False,
|
||||||
)
|
)
|
||||||
):
|
):
|
||||||
|
builtin_bp_handler = sys.breakpointhook
|
||||||
|
orig_bp_path: str|None = os.environ.get(
|
||||||
|
'PYTHONBREAKPOINT',
|
||||||
|
None,
|
||||||
|
)
|
||||||
os.environ['PYTHONBREAKPOINT'] = (
|
os.environ['PYTHONBREAKPOINT'] = (
|
||||||
'tractor.devx._debug.pause_from_sync'
|
'tractor.devx._debug.pause_from_sync'
|
||||||
)
|
)
|
||||||
else:
|
|
||||||
# TODO: disable `breakpoint()` by default (without
|
|
||||||
# `greenback`) since it will break any multi-actor
|
|
||||||
# usage by a clobbered TTY's stdstreams!
|
|
||||||
def block_bps(*args, **kwargs):
|
|
||||||
raise RuntimeError(
|
|
||||||
'Trying to use `breakpoint()` eh?\n'
|
|
||||||
'Welp, `tractor` blocks `breakpoint()` built-in calls by default!\n'
|
|
||||||
'If you need to use it please install `greenback` and set '
|
|
||||||
'`debug_mode=True` when opening the runtime '
|
|
||||||
'(either via `.open_nursery()` or `open_root_actor()`)\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
sys.breakpointhook = block_bps
|
|
||||||
# os.environ['PYTHONBREAKPOINT'] = None
|
|
||||||
|
|
||||||
# attempt to retreive ``trio``'s sigint handler and stash it
|
# attempt to retreive ``trio``'s sigint handler and stash it
|
||||||
# on our debugger lock state.
|
# on our debugger lock state.
|
||||||
|
@ -211,11 +191,7 @@ async def open_root_actor(
|
||||||
assert _log
|
assert _log
|
||||||
|
|
||||||
# TODO: factor this into `.devx._stackscope`!!
|
# TODO: factor this into `.devx._stackscope`!!
|
||||||
if (
|
if debug_mode:
|
||||||
debug_mode
|
|
||||||
and
|
|
||||||
enable_stack_on_sig
|
|
||||||
):
|
|
||||||
try:
|
try:
|
||||||
logger.info('Enabling `stackscope` traces on SIGUSR1')
|
logger.info('Enabling `stackscope` traces on SIGUSR1')
|
||||||
from .devx import enable_stack_on_sig
|
from .devx import enable_stack_on_sig
|
||||||
|
@ -392,14 +368,12 @@ async def open_root_actor(
|
||||||
_state._last_actor_terminated = actor
|
_state._last_actor_terminated = actor
|
||||||
|
|
||||||
# restore built-in `breakpoint()` hook state
|
# restore built-in `breakpoint()` hook state
|
||||||
if debug_mode:
|
sys.breakpointhook = builtin_bp_handler
|
||||||
if builtin_bp_handler is not None:
|
if orig_bp_path is not None:
|
||||||
sys.breakpointhook = builtin_bp_handler
|
os.environ['PYTHONBREAKPOINT'] = orig_bp_path
|
||||||
if orig_bp_path is not None:
|
else:
|
||||||
os.environ['PYTHONBREAKPOINT'] = orig_bp_path
|
# clear env back to having no entry
|
||||||
else:
|
os.environ.pop('PYTHONBREAKPOINT')
|
||||||
# clear env back to having no entry
|
|
||||||
os.environ.pop('PYTHONBREAKPOINT')
|
|
||||||
|
|
||||||
logger.runtime("Root actor terminated")
|
logger.runtime("Root actor terminated")
|
||||||
|
|
||||||
|
|
|
@ -503,7 +503,7 @@ async def trio_proc(
|
||||||
})
|
})
|
||||||
|
|
||||||
# track subactor in current nursery
|
# track subactor in current nursery
|
||||||
curr_actor: Actor = current_actor()
|
curr_actor = current_actor()
|
||||||
curr_actor._actoruid2nursery[subactor.uid] = actor_nursery
|
curr_actor._actoruid2nursery[subactor.uid] = actor_nursery
|
||||||
|
|
||||||
# resume caller at next checkpoint now that child is up
|
# resume caller at next checkpoint now that child is up
|
||||||
|
|
|
@ -30,16 +30,11 @@ if TYPE_CHECKING:
|
||||||
|
|
||||||
_current_actor: Actor|None = None # type: ignore # noqa
|
_current_actor: Actor|None = None # type: ignore # noqa
|
||||||
_last_actor_terminated: Actor|None = None
|
_last_actor_terminated: Actor|None = None
|
||||||
|
|
||||||
# TODO: mk this a `msgspec.Struct`!
|
|
||||||
_runtime_vars: dict[str, Any] = {
|
_runtime_vars: dict[str, Any] = {
|
||||||
'_debug_mode': False,
|
'_debug_mode': False,
|
||||||
'_is_root': False,
|
'_is_root': False,
|
||||||
'_root_mailbox': (None, None),
|
'_root_mailbox': (None, None),
|
||||||
'_registry_addrs': [],
|
'_registry_addrs': [],
|
||||||
|
|
||||||
# for `breakpoint()` support
|
|
||||||
'use_greenback': False,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -119,11 +119,11 @@ class ActorNursery:
|
||||||
name: str,
|
name: str,
|
||||||
*,
|
*,
|
||||||
bind_addrs: list[tuple[str, int]] = [_default_bind_addr],
|
bind_addrs: list[tuple[str, int]] = [_default_bind_addr],
|
||||||
rpc_module_paths: list[str]|None = None,
|
rpc_module_paths: list[str] | None = None,
|
||||||
enable_modules: list[str]|None = None,
|
enable_modules: list[str] | None = None,
|
||||||
loglevel: str|None = None, # set log level per subactor
|
loglevel: str | None = None, # set log level per subactor
|
||||||
nursery: trio.Nursery|None = None,
|
nursery: trio.Nursery | None = None,
|
||||||
debug_mode: bool|None = None,
|
debug_mode: bool | None = None,
|
||||||
infect_asyncio: bool = False,
|
infect_asyncio: bool = False,
|
||||||
) -> Portal:
|
) -> Portal:
|
||||||
'''
|
'''
|
||||||
|
|
|
@ -1,177 +0,0 @@
|
||||||
# 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/>.
|
|
||||||
|
|
||||||
'''
|
|
||||||
Tools for code-object annotation, introspection and mutation
|
|
||||||
as it pertains to improving the grok-ability of our runtime!
|
|
||||||
|
|
||||||
'''
|
|
||||||
from __future__ import annotations
|
|
||||||
import inspect
|
|
||||||
# import msgspec
|
|
||||||
# from pprint import pformat
|
|
||||||
from types import (
|
|
||||||
FrameType,
|
|
||||||
FunctionType,
|
|
||||||
MethodType,
|
|
||||||
# CodeType,
|
|
||||||
)
|
|
||||||
from typing import (
|
|
||||||
# Any,
|
|
||||||
Callable,
|
|
||||||
# TYPE_CHECKING,
|
|
||||||
Type,
|
|
||||||
)
|
|
||||||
|
|
||||||
from tractor.msg import (
|
|
||||||
pretty_struct,
|
|
||||||
NamespacePath,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
# TODO: yeah, i don't love this and we should prolly just
|
|
||||||
# write a decorator that actually keeps a stupid ref to the func
|
|
||||||
# obj..
|
|
||||||
def get_class_from_frame(fr: FrameType) -> (
|
|
||||||
FunctionType
|
|
||||||
|MethodType
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Attempt to get the function (or method) reference
|
|
||||||
from a given `FrameType`.
|
|
||||||
|
|
||||||
Verbatim from an SO:
|
|
||||||
https://stackoverflow.com/a/2220759
|
|
||||||
|
|
||||||
'''
|
|
||||||
args, _, _, value_dict = inspect.getargvalues(fr)
|
|
||||||
|
|
||||||
# we check the first parameter for the frame function is
|
|
||||||
# named 'self'
|
|
||||||
if (
|
|
||||||
len(args)
|
|
||||||
and
|
|
||||||
# TODO: other cases for `@classmethod` etc..?)
|
|
||||||
args[0] == 'self'
|
|
||||||
):
|
|
||||||
# in that case, 'self' will be referenced in value_dict
|
|
||||||
instance: object = value_dict.get('self')
|
|
||||||
if instance:
|
|
||||||
# return its class
|
|
||||||
return getattr(
|
|
||||||
instance,
|
|
||||||
'__class__',
|
|
||||||
None,
|
|
||||||
)
|
|
||||||
|
|
||||||
# return None otherwise
|
|
||||||
return None
|
|
||||||
|
|
||||||
|
|
||||||
def func_ref_from_frame(
|
|
||||||
frame: FrameType,
|
|
||||||
) -> Callable:
|
|
||||||
func_name: str = frame.f_code.co_name
|
|
||||||
try:
|
|
||||||
return frame.f_globals[func_name]
|
|
||||||
except KeyError:
|
|
||||||
cls: Type|None = get_class_from_frame(frame)
|
|
||||||
if cls:
|
|
||||||
return getattr(
|
|
||||||
cls,
|
|
||||||
func_name,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
# TODO: move all this into new `.devx._code`!
|
|
||||||
# -[ ] prolly create a `@runtime_api` dec?
|
|
||||||
# -[ ] ^- make it capture and/or accept buncha optional
|
|
||||||
# meta-data like a fancier version of `@pdbp.hideframe`.
|
|
||||||
#
|
|
||||||
class CallerInfo(pretty_struct.Struct):
|
|
||||||
rt_fi: inspect.FrameInfo
|
|
||||||
call_frame: FrameType
|
|
||||||
|
|
||||||
@property
|
|
||||||
def api_func_ref(self) -> Callable|None:
|
|
||||||
return func_ref_from_frame(self.rt_fi.frame)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def api_nsp(self) -> NamespacePath|None:
|
|
||||||
func: FunctionType = self.api_func_ref
|
|
||||||
if func:
|
|
||||||
return NamespacePath.from_ref(func)
|
|
||||||
|
|
||||||
return '<unknown>'
|
|
||||||
|
|
||||||
@property
|
|
||||||
def caller_func_ref(self) -> Callable|None:
|
|
||||||
return func_ref_from_frame(self.call_frame)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def caller_nsp(self) -> NamespacePath|None:
|
|
||||||
func: FunctionType = self.caller_func_ref
|
|
||||||
if func:
|
|
||||||
return NamespacePath.from_ref(func)
|
|
||||||
|
|
||||||
return '<unknown>'
|
|
||||||
|
|
||||||
|
|
||||||
def find_caller_info(
|
|
||||||
dunder_var: str = '__runtimeframe__',
|
|
||||||
iframes:int = 1,
|
|
||||||
check_frame_depth: bool = True,
|
|
||||||
|
|
||||||
) -> CallerInfo|None:
|
|
||||||
'''
|
|
||||||
Scan up the callstack for a frame with a `dunder_var: str` variable
|
|
||||||
and return the `iframes` frames above it.
|
|
||||||
|
|
||||||
By default we scan for a `__runtimeframe__` scope var which
|
|
||||||
denotes a `tractor` API above which (one frame up) is "user
|
|
||||||
app code" which "called into" the `tractor` method or func.
|
|
||||||
|
|
||||||
TODO: ex with `Portal.open_context()`
|
|
||||||
|
|
||||||
'''
|
|
||||||
# TODO: use this instead?
|
|
||||||
# https://docs.python.org/3/library/inspect.html#inspect.getouterframes
|
|
||||||
frames: list[inspect.FrameInfo] = inspect.stack()
|
|
||||||
for fi in frames:
|
|
||||||
assert (
|
|
||||||
fi.function
|
|
||||||
==
|
|
||||||
fi.frame.f_code.co_name
|
|
||||||
)
|
|
||||||
this_frame: FrameType = fi.frame
|
|
||||||
dunder_val: int|None = this_frame.f_locals.get(dunder_var)
|
|
||||||
if dunder_val:
|
|
||||||
go_up_iframes: int = (
|
|
||||||
dunder_val # could be 0 or `True` i guess?
|
|
||||||
or
|
|
||||||
iframes
|
|
||||||
)
|
|
||||||
rt_frame: FrameType = fi.frame
|
|
||||||
call_frame = rt_frame
|
|
||||||
for i in range(go_up_iframes):
|
|
||||||
call_frame = call_frame.f_back
|
|
||||||
|
|
||||||
return CallerInfo(
|
|
||||||
rt_fi=fi,
|
|
||||||
call_frame=call_frame,
|
|
||||||
)
|
|
||||||
|
|
||||||
return None
|
|
|
@ -190,14 +190,11 @@ class Lock:
|
||||||
is_trio_main = (
|
is_trio_main = (
|
||||||
# TODO: since this is private, @oremanj says
|
# TODO: since this is private, @oremanj says
|
||||||
# we should just copy the impl for now..
|
# we should just copy the impl for now..
|
||||||
(is_main_thread := trio._util.is_main_thread())
|
trio._util.is_main_thread()
|
||||||
and
|
and
|
||||||
(async_lib := sniffio.current_async_library()) == 'trio'
|
(async_lib := sniffio.current_async_library()) == 'trio'
|
||||||
)
|
)
|
||||||
if (
|
if not is_trio_main:
|
||||||
not is_trio_main
|
|
||||||
and is_main_thread
|
|
||||||
):
|
|
||||||
log.warning(
|
log.warning(
|
||||||
f'Current async-lib detected by `sniffio`: {async_lib}\n'
|
f'Current async-lib detected by `sniffio`: {async_lib}\n'
|
||||||
)
|
)
|
||||||
|
|
|
@ -23,31 +23,12 @@ into each ``trio.Nursery`` except it links the lifetimes of memory space
|
||||||
disjoint, parallel executing tasks in separate actors.
|
disjoint, parallel executing tasks in separate actors.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
from __future__ import annotations
|
|
||||||
import multiprocessing as mp
|
|
||||||
from signal import (
|
from signal import (
|
||||||
signal,
|
signal,
|
||||||
SIGUSR1,
|
SIGUSR1,
|
||||||
)
|
)
|
||||||
import traceback
|
|
||||||
from typing import TYPE_CHECKING
|
|
||||||
|
|
||||||
import trio
|
import trio
|
||||||
from tractor import (
|
|
||||||
_state,
|
|
||||||
log as logmod,
|
|
||||||
)
|
|
||||||
|
|
||||||
log = logmod.get_logger(__name__)
|
|
||||||
|
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
|
||||||
from tractor._spawn import ProcessType
|
|
||||||
from tractor import (
|
|
||||||
Actor,
|
|
||||||
ActorNursery,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@trio.lowlevel.disable_ki_protection
|
@trio.lowlevel.disable_ki_protection
|
||||||
def dump_task_tree() -> None:
|
def dump_task_tree() -> None:
|
||||||
|
@ -60,15 +41,9 @@ def dump_task_tree() -> None:
|
||||||
recurse_child_tasks=True
|
recurse_child_tasks=True
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
log = get_console_log(
|
log = get_console_log('cancel')
|
||||||
name=__name__,
|
log.pdb(
|
||||||
level='cancel',
|
f'Dumping `stackscope` tree:\n\n'
|
||||||
)
|
|
||||||
actor: Actor = _state.current_actor()
|
|
||||||
log.devx(
|
|
||||||
f'Dumping `stackscope` tree for actor\n'
|
|
||||||
f'{actor.name}: {actor}\n'
|
|
||||||
f' |_{mp.current_process()}\n\n'
|
|
||||||
f'{tree_str}\n'
|
f'{tree_str}\n'
|
||||||
)
|
)
|
||||||
# import logging
|
# import logging
|
||||||
|
@ -81,13 +56,8 @@ def dump_task_tree() -> None:
|
||||||
# ).exception("Error printing task tree")
|
# ).exception("Error printing task tree")
|
||||||
|
|
||||||
|
|
||||||
def signal_handler(
|
def signal_handler(sig: int, frame: object) -> None:
|
||||||
sig: int,
|
import traceback
|
||||||
frame: object,
|
|
||||||
|
|
||||||
relay_to_subs: bool = True,
|
|
||||||
|
|
||||||
) -> None:
|
|
||||||
try:
|
try:
|
||||||
trio.lowlevel.current_trio_token(
|
trio.lowlevel.current_trio_token(
|
||||||
).run_sync_soon(dump_task_tree)
|
).run_sync_soon(dump_task_tree)
|
||||||
|
@ -95,26 +65,6 @@ def signal_handler(
|
||||||
# not in async context -- print a normal traceback
|
# not in async context -- print a normal traceback
|
||||||
traceback.print_stack()
|
traceback.print_stack()
|
||||||
|
|
||||||
if not relay_to_subs:
|
|
||||||
return
|
|
||||||
|
|
||||||
an: ActorNursery
|
|
||||||
for an in _state.current_actor()._actoruid2nursery.values():
|
|
||||||
|
|
||||||
subproc: ProcessType
|
|
||||||
subactor: Actor
|
|
||||||
for subactor, subproc, _ in an._children.values():
|
|
||||||
log.devx(
|
|
||||||
f'Relaying `SIGUSR1`[{sig}] to sub-actor\n'
|
|
||||||
f'{subactor}\n'
|
|
||||||
f' |_{subproc}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
if isinstance(subproc, trio.Process):
|
|
||||||
subproc.send_signal(sig)
|
|
||||||
|
|
||||||
elif isinstance(subproc, mp.Process):
|
|
||||||
subproc._send_signal(sig)
|
|
||||||
|
|
||||||
|
|
||||||
def enable_stack_on_sig(
|
def enable_stack_on_sig(
|
||||||
|
@ -132,6 +82,3 @@ def enable_stack_on_sig(
|
||||||
# NOTE: not the above can be triggered from
|
# NOTE: not the above can be triggered from
|
||||||
# a (xonsh) shell using:
|
# a (xonsh) shell using:
|
||||||
# kill -SIGUSR1 @$(pgrep -f '<cmd>')
|
# kill -SIGUSR1 @$(pgrep -f '<cmd>')
|
||||||
#
|
|
||||||
# for example if you were looking to trace a `pytest` run
|
|
||||||
# kill -SIGUSR1 @$(pgrep -f 'pytest')
|
|
||||||
|
|
|
@ -21,11 +21,6 @@ Log like a forester!
|
||||||
from collections.abc import Mapping
|
from collections.abc import Mapping
|
||||||
import sys
|
import sys
|
||||||
import logging
|
import logging
|
||||||
from logging import (
|
|
||||||
LoggerAdapter,
|
|
||||||
Logger,
|
|
||||||
StreamHandler,
|
|
||||||
)
|
|
||||||
import colorlog # type: ignore
|
import colorlog # type: ignore
|
||||||
|
|
||||||
import trio
|
import trio
|
||||||
|
@ -53,19 +48,20 @@ LOG_FORMAT = (
|
||||||
|
|
||||||
DATE_FORMAT = '%b %d %H:%M:%S'
|
DATE_FORMAT = '%b %d %H:%M:%S'
|
||||||
|
|
||||||
# FYI, ERROR is 40
|
LEVELS: dict[str, int] = {
|
||||||
CUSTOM_LEVELS: dict[str, int] = {
|
|
||||||
'TRANSPORT': 5,
|
'TRANSPORT': 5,
|
||||||
'RUNTIME': 15,
|
'RUNTIME': 15,
|
||||||
'DEVX': 17,
|
'CANCEL': 16,
|
||||||
'CANCEL': 18,
|
|
||||||
'PDB': 500,
|
'PDB': 500,
|
||||||
}
|
}
|
||||||
|
# _custom_levels: set[str] = {
|
||||||
|
# lvlname.lower for lvlname in LEVELS.keys()
|
||||||
|
# }
|
||||||
|
|
||||||
STD_PALETTE = {
|
STD_PALETTE = {
|
||||||
'CRITICAL': 'red',
|
'CRITICAL': 'red',
|
||||||
'ERROR': 'red',
|
'ERROR': 'red',
|
||||||
'PDB': 'white',
|
'PDB': 'white',
|
||||||
'DEVX': 'cyan',
|
|
||||||
'WARNING': 'yellow',
|
'WARNING': 'yellow',
|
||||||
'INFO': 'green',
|
'INFO': 'green',
|
||||||
'CANCEL': 'yellow',
|
'CANCEL': 'yellow',
|
||||||
|
@ -82,7 +78,7 @@ BOLD_PALETTE = {
|
||||||
|
|
||||||
# TODO: this isn't showing the correct '{filename}'
|
# TODO: this isn't showing the correct '{filename}'
|
||||||
# as it did before..
|
# as it did before..
|
||||||
class StackLevelAdapter(LoggerAdapter):
|
class StackLevelAdapter(logging.LoggerAdapter):
|
||||||
|
|
||||||
def transport(
|
def transport(
|
||||||
self,
|
self,
|
||||||
|
@ -90,8 +86,7 @@ class StackLevelAdapter(LoggerAdapter):
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
'''
|
'''
|
||||||
IPC transport level msg IO; generally anything below
|
IPC level msg-ing.
|
||||||
`._ipc.Channel` and friends.
|
|
||||||
|
|
||||||
'''
|
'''
|
||||||
return self.log(5, msg)
|
return self.log(5, msg)
|
||||||
|
@ -107,11 +102,11 @@ class StackLevelAdapter(LoggerAdapter):
|
||||||
msg: str,
|
msg: str,
|
||||||
) -> None:
|
) -> None:
|
||||||
'''
|
'''
|
||||||
Cancellation sequencing, mostly for runtime reporting.
|
Cancellation logging, mostly for runtime reporting.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
return self.log(
|
return self.log(
|
||||||
level=22,
|
level=16,
|
||||||
msg=msg,
|
msg=msg,
|
||||||
# stacklevel=4,
|
# stacklevel=4,
|
||||||
)
|
)
|
||||||
|
@ -121,21 +116,11 @@ class StackLevelAdapter(LoggerAdapter):
|
||||||
msg: str,
|
msg: str,
|
||||||
) -> None:
|
) -> None:
|
||||||
'''
|
'''
|
||||||
`pdb`-REPL (debugger) related statuses.
|
Debugger logging.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
return self.log(500, msg)
|
return self.log(500, msg)
|
||||||
|
|
||||||
def devx(
|
|
||||||
self,
|
|
||||||
msg: str,
|
|
||||||
) -> None:
|
|
||||||
'''
|
|
||||||
"Developer experience" sub-sys statuses.
|
|
||||||
|
|
||||||
'''
|
|
||||||
return self.log(17, msg)
|
|
||||||
|
|
||||||
def log(
|
def log(
|
||||||
self,
|
self,
|
||||||
level,
|
level,
|
||||||
|
@ -151,7 +136,8 @@ class StackLevelAdapter(LoggerAdapter):
|
||||||
if self.isEnabledFor(level):
|
if self.isEnabledFor(level):
|
||||||
stacklevel: int = 3
|
stacklevel: int = 3
|
||||||
if (
|
if (
|
||||||
level in CUSTOM_LEVELS.values()
|
level in LEVELS.values()
|
||||||
|
# or level in _custom_levels
|
||||||
):
|
):
|
||||||
stacklevel: int = 4
|
stacklevel: int = 4
|
||||||
|
|
||||||
|
@ -198,30 +184,8 @@ class StackLevelAdapter(LoggerAdapter):
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
# TODO IDEAs:
|
|
||||||
# -[ ] move to `.devx.pformat`?
|
|
||||||
# -[ ] do per task-name and actor-name color coding
|
|
||||||
# -[ ] unique color per task-id and actor-uuid
|
|
||||||
def pformat_task_uid(
|
|
||||||
id_part: str = 'tail'
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Return `str`-ified unique for a `trio.Task` via a combo of its
|
|
||||||
`.name: str` and `id()` truncated output.
|
|
||||||
|
|
||||||
'''
|
|
||||||
task: trio.Task = trio.lowlevel.current_task()
|
|
||||||
tid: str = str(id(task))
|
|
||||||
if id_part == 'tail':
|
|
||||||
tid_part: str = tid[-6:]
|
|
||||||
else:
|
|
||||||
tid_part: str = tid[:6]
|
|
||||||
|
|
||||||
return f'{task.name}[{tid_part}]'
|
|
||||||
|
|
||||||
|
|
||||||
_conc_name_getters = {
|
_conc_name_getters = {
|
||||||
'task': pformat_task_uid,
|
'task': lambda: trio.lowlevel.current_task().name,
|
||||||
'actor': lambda: current_actor(),
|
'actor': lambda: current_actor(),
|
||||||
'actor_name': lambda: current_actor().name,
|
'actor_name': lambda: current_actor().name,
|
||||||
'actor_uid': lambda: current_actor().uid[1][:6],
|
'actor_uid': lambda: current_actor().uid[1][:6],
|
||||||
|
@ -229,10 +193,7 @@ _conc_name_getters = {
|
||||||
|
|
||||||
|
|
||||||
class ActorContextInfo(Mapping):
|
class ActorContextInfo(Mapping):
|
||||||
'''
|
"Dyanmic lookup for local actor and task names"
|
||||||
Dyanmic lookup for local actor and task names.
|
|
||||||
|
|
||||||
'''
|
|
||||||
_context_keys = (
|
_context_keys = (
|
||||||
'task',
|
'task',
|
||||||
'actor',
|
'actor',
|
||||||
|
@ -263,7 +224,6 @@ def get_logger(
|
||||||
'''Return the package log or a sub-logger for ``name`` if provided.
|
'''Return the package log or a sub-logger for ``name`` if provided.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
log: Logger
|
|
||||||
log = rlog = logging.getLogger(_root_name)
|
log = rlog = logging.getLogger(_root_name)
|
||||||
|
|
||||||
if (
|
if (
|
||||||
|
@ -306,7 +266,7 @@ def get_logger(
|
||||||
logger = StackLevelAdapter(log, ActorContextInfo())
|
logger = StackLevelAdapter(log, ActorContextInfo())
|
||||||
|
|
||||||
# additional levels
|
# additional levels
|
||||||
for name, val in CUSTOM_LEVELS.items():
|
for name, val in LEVELS.items():
|
||||||
logging.addLevelName(val, name)
|
logging.addLevelName(val, name)
|
||||||
|
|
||||||
# ensure customs levels exist as methods
|
# ensure customs levels exist as methods
|
||||||
|
@ -318,7 +278,7 @@ def get_logger(
|
||||||
def get_console_log(
|
def get_console_log(
|
||||||
level: str | None = None,
|
level: str | None = None,
|
||||||
**kwargs,
|
**kwargs,
|
||||||
) -> LoggerAdapter:
|
) -> logging.LoggerAdapter:
|
||||||
'''Get the package logger and enable a handler which writes to stderr.
|
'''Get the package logger and enable a handler which writes to stderr.
|
||||||
|
|
||||||
Yeah yeah, i know we can use ``DictConfig``. You do it.
|
Yeah yeah, i know we can use ``DictConfig``. You do it.
|
||||||
|
@ -343,7 +303,7 @@ def get_console_log(
|
||||||
None,
|
None,
|
||||||
)
|
)
|
||||||
):
|
):
|
||||||
handler = StreamHandler()
|
handler = logging.StreamHandler()
|
||||||
formatter = colorlog.ColoredFormatter(
|
formatter = colorlog.ColoredFormatter(
|
||||||
LOG_FORMAT,
|
LOG_FORMAT,
|
||||||
datefmt=DATE_FORMAT,
|
datefmt=DATE_FORMAT,
|
||||||
|
@ -363,19 +323,3 @@ def get_loglevel() -> str:
|
||||||
|
|
||||||
# global module logger for tractor itself
|
# global module logger for tractor itself
|
||||||
log = get_logger('tractor')
|
log = get_logger('tractor')
|
||||||
|
|
||||||
|
|
||||||
def at_least_level(
|
|
||||||
log: Logger|LoggerAdapter,
|
|
||||||
level: int|str,
|
|
||||||
) -> bool:
|
|
||||||
'''
|
|
||||||
Predicate to test if a given level is active.
|
|
||||||
|
|
||||||
'''
|
|
||||||
if isinstance(level, str):
|
|
||||||
level: int = CUSTOM_LEVELS[level.upper()]
|
|
||||||
|
|
||||||
if log.getEffectiveLevel() <= level:
|
|
||||||
return True
|
|
||||||
return False
|
|
||||||
|
|
Loading…
Reference in New Issue