frame.py
94 lines
| 3.0 KiB
| text/x-python
|
PythonLexer
Brian Granger
|
r2498 | # encoding: utf-8 | ||
""" | ||||
Utilities for working with stack frames. | ||||
""" | ||||
#----------------------------------------------------------------------------- | ||||
Matthias BUSSONNIER
|
r5390 | # Copyright (C) 2008-2011 The IPython Development Team | ||
Brian Granger
|
r2498 | # | ||
# Distributed under the terms of the BSD License. The full license is in | ||||
# the file COPYING, distributed as part of this software. | ||||
#----------------------------------------------------------------------------- | ||||
#----------------------------------------------------------------------------- | ||||
# Imports | ||||
#----------------------------------------------------------------------------- | ||||
import sys | ||||
M Bussonnier
|
r28899 | from typing import Any | ||
Brian Granger
|
r2498 | |||
#----------------------------------------------------------------------------- | ||||
# Code | ||||
#----------------------------------------------------------------------------- | ||||
def extract_vars(*names,**kw): | ||||
"""Extract a set of variables by name from another frame. | ||||
Thomas Kluyver
|
r13587 | Parameters | ||
---------- | ||||
*names : str | ||||
Brian Granger
|
r2498 | One or more variable names which will be extracted from the caller's | ||
Thomas Kluyver
|
r13587 | frame. | ||
Matthias Bussonnier
|
r26419 | **kw : integer, optional | ||
Brian Granger
|
r2498 | How many frames in the stack to walk when looking for your variables. | ||
Thomas Kluyver
|
r13587 | The default is 0, which will use the frame where the call was made. | ||
Brian Granger
|
r2498 | |||
Thomas Kluyver
|
r13587 | Examples | ||
-------- | ||||
:: | ||||
Brian Granger
|
r2498 | |||
In [2]: def func(x): | ||||
...: y = 1 | ||||
Thomas Kluyver
|
r13373 | ...: print(sorted(extract_vars('x','y').items())) | ||
Brian Granger
|
r2498 | ...: | ||
In [3]: func('hello') | ||||
Thomas Kluyver
|
r7014 | [('x', 'hello'), ('y', 1)] | ||
Brian Granger
|
r2498 | """ | ||
depth = kw.get('depth',0) | ||||
luz.paz
|
r24494 | |||
Brian Granger
|
r2498 | callerNS = sys._getframe(depth+1).f_locals | ||
return dict((k,callerNS[k]) for k in names) | ||||
M Bussonnier
|
r28899 | def extract_vars_above(*names: list[str]): | ||
Brian Granger
|
r2498 | """Extract a set of variables by name from another frame. | ||
Similar to extractVars(), but with a specified depth of 1, so that names | ||||
luz.paz
|
r24494 | are extracted exactly from above the caller. | ||
Brian Granger
|
r2498 | |||
This is simply a convenience function so that the very common case (for us) | ||||
of skipping exactly 1 frame doesn't have to construct a special dict for | ||||
keyword passing.""" | ||||
callerNS = sys._getframe(2).f_locals | ||||
return dict((k,callerNS[k]) for k in names) | ||||
M Bussonnier
|
r28899 | def debugx(expr: str, pre_msg: str = ""): | ||
Brian Granger
|
r2498 | """Print the value of an expression from the caller's frame. | ||
Takes an expression, evaluates it in the caller's frame and prints both | ||||
the given expression and the resulting value (as well as a debug mark | ||||
indicating the name of the calling function. The input must be of a form | ||||
suitable for eval(). | ||||
An optional message can be passed, which will be prepended to the printed | ||||
expr->value pair.""" | ||||
cf = sys._getframe(1) | ||||
Thomas Kluyver
|
r13348 | print('[DBG:%s] %s%s -> %r' % (cf.f_code.co_name,pre_msg,expr, | ||
eval(expr,cf.f_globals,cf.f_locals))) | ||||
Brian Granger
|
r2498 | |||
# deactivate it by uncommenting the following line, which makes it a no-op | ||||
#def debugx(expr,pre_msg=''): pass | ||||
M Bussonnier
|
r28899 | |||
def extract_module_locals(depth: int = 0) -> tuple[Any, Any]: | ||||
Thomas Ballinger
|
r21856 | """Returns (module, locals) of the function `depth` frames away from the caller""" | ||
Scott Tsai
|
r6225 | f = sys._getframe(depth + 1) | ||
global_ns = f.f_globals | ||||
Scott Tsai
|
r6224 | module = sys.modules[global_ns['__name__']] | ||
Scott Tsai
|
r6225 | return (module, f.f_locals) | ||