forked from goodboy/tractor
1
0
Fork 0

Create runtime variables

debug_tests
Tyler Goodlet 2020-07-23 13:14:15 -04:00
parent 196cf14211
commit 8c97f7bbb3
1 changed files with 22 additions and 0 deletions

View File

@ -3,10 +3,15 @@ Per process state
""" """
from typing import Optional from typing import Optional
from collections import Mapping from collections import Mapping
import multiprocessing as mp
import trio import trio
_current_actor: Optional['Actor'] = None # type: ignore _current_actor: Optional['Actor'] = None # type: ignore
_runtime_vars = {
'_debug_mode': False,
'_is_root': False,
}
def current_actor() -> 'Actor': # type: ignore def current_actor() -> 'Actor': # type: ignore
@ -36,3 +41,20 @@ class ActorContextInfo(Mapping):
except RuntimeError: except RuntimeError:
# no local actor/task context initialized yet # no local actor/task context initialized yet
return f'no {key} context' return f'no {key} context'
def is_main_process() -> bool:
"""Bool determining if this actor is running in the top-most process.
"""
return mp.current_process().name == 'MainProcess'
def debug_mode() -> bool:
"""Bool determining if "debug mode" is on which enables
remote subactor pdb entry on crashes.
"""
return _runtime_vars['_debug_mode']
def is_root_process() -> bool:
return _runtime_vars['_is_root']