59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
import py
|
|
import py.__.code.safe_repr
|
|
|
|
class Frame(object):
|
|
"""Wrapper around a Python frame holding f_locals and f_globals
|
|
in which expressions can be evaluated."""
|
|
|
|
def __init__(self, frame):
|
|
self.code = py.code.Code(frame.f_code)
|
|
self.lineno = frame.f_lineno - 1
|
|
self.f_globals = frame.f_globals
|
|
self.f_locals = frame.f_locals
|
|
self.raw = frame
|
|
|
|
def statement(self):
|
|
return self.code.fullsource.getstatement(self.lineno)
|
|
statement = property(statement, None, None,
|
|
"statement this frame is at")
|
|
|
|
def eval(self, code, **vars):
|
|
""" evaluate 'code' in the frame
|
|
|
|
'vars' are optional additional local variables
|
|
|
|
returns the result of the evaluation
|
|
"""
|
|
f_locals = self.f_locals.copy()
|
|
f_locals.update(vars)
|
|
return eval(code, self.f_globals, f_locals)
|
|
|
|
def exec_(self, code, **vars):
|
|
""" exec 'code' in the frame
|
|
|
|
'vars' are optiona; additional local variables
|
|
"""
|
|
f_locals = self.f_locals.copy()
|
|
f_locals.update(vars)
|
|
exec code in self.f_globals, f_locals
|
|
|
|
def repr(self, object):
|
|
""" return a 'safe' (non-recursive, one-line) string repr for 'object'
|
|
"""
|
|
return py.__.code.safe_repr._repr(object)
|
|
|
|
def is_true(self, object):
|
|
return object
|
|
|
|
def getargs(self):
|
|
""" return a list of tuples (name, value) for all arguments
|
|
"""
|
|
retval = []
|
|
for arg in self.code.getargs():
|
|
try:
|
|
retval.append((arg, self.f_locals[arg]))
|
|
except KeyError:
|
|
pass # this can occur when using Psyco
|
|
return retval
|
|
|