| import traceback |
| from contextlib import contextmanager |
| from typing import List, Any, Dict |
| from ._compatibility import compatibility |
| |
| __all__ = ['preserve_node_meta', 'has_preserved_node_meta', |
| 'set_stack_trace', 'format_stack', |
| 'set_current_meta', 'get_current_meta'] |
| |
| current_meta: Dict[str, Any] = {} |
| should_preserve_node_meta = False |
| |
| |
| @compatibility(is_backward_compatible=False) |
| @contextmanager |
| def preserve_node_meta(): |
| global should_preserve_node_meta |
| |
| saved_should_preserve_node_meta = should_preserve_node_meta |
| try: |
| should_preserve_node_meta = True |
| yield |
| finally: |
| should_preserve_node_meta = saved_should_preserve_node_meta |
| |
| |
| @compatibility(is_backward_compatible=False) |
| def set_stack_trace(stack : List[str]): |
| global current_meta |
| |
| if should_preserve_node_meta and stack: |
| current_meta["stack_trace"] = "".join(stack) |
| |
| |
| @compatibility(is_backward_compatible=False) |
| def format_stack() -> List[str]: |
| if should_preserve_node_meta: |
| return [current_meta.get("stack_trace", "")] |
| else: |
| # fallback to traceback.format_stack() |
| return traceback.format_list(traceback.extract_stack()[:-1]) |
| |
| |
| @compatibility(is_backward_compatible=False) |
| def has_preserved_node_meta() -> bool: |
| return should_preserve_node_meta |
| |
| |
| @compatibility(is_backward_compatible=False) |
| @contextmanager |
| def set_current_meta(meta : Dict[str, Any]): |
| global current_meta |
| |
| if should_preserve_node_meta and meta: |
| saved_meta = current_meta |
| try: |
| current_meta = meta |
| yield |
| finally: |
| current_meta = saved_meta |
| else: |
| yield |
| |
| |
| @compatibility(is_backward_compatible=False) |
| def get_current_meta() -> Dict[str, Any]: |
| return current_meta.copy() |