使用python inspect来查看函数中的局部变量

Question

我无法弄清楚如何使用inspect / inspect_shell检查当前正在执行的函数

我猜它涉及使用getinnerframe和getouterframe遍历框架层次结构，但我对几个问题感到困惑。

给出这个例子nine.py：

import inspect_shell
import time

def number_nine():
    x = 9
    while x==9:
        time.sleep(1)

number_nine()
print x

我想检查x的值，甚至可能更改它以使函数返回并打印新值。

首先我发布nine.py，然后在一个单独的命令窗口，使用inspect_shell ，我看到getinnerframes不会将当前帧上工作（它需要一个跟踪（也许？））和当前帧没有“痕迹”。 和getouterframes （如果我正在考虑这个向后）似乎只得到与我的函数无关的帧。

>> Inspect Shell v1.0
>> https://github.com/amoffat/Inspect-Shell

localhost:1234> import inspect

localhost:1234> f = inspect.currentframe()

localhost:1234> inspect.getinnerframes(f)
Traceback (most recent call last):
  File "C:\Users\Paul\Desktop\inspect_shell.py", line 143, in run_repl
    try: exec compile(data, "<dummy>", "single") in f_globals, f_globals
  File "<dummy>", line 1, in <module>
  File "C:\Python26\lib\inspect.py", line 942, in getinnerframes
    framelist.append((tb.tb_frame,) + getframeinfo(tb, context))
AttributeError: 'frame' object has no attribute 'tb_frame'


localhost:1234> dir(f)
['__class__', '__delattr__', '__doc__', '__format__', '__getattribute__', '__hash__',
'__init__', '__new__', '__reduce__', '__reduce_ex__', '__repr__',
'__setattr__', '__sizeof__', '__str__', '__subclasshook__', 'f_back', 'f_builtins', 
'f_code', 'f_exc_traceback', 'f_exc_type', 'f_exc_value', 'f_glo
bals', 'f_lasti', 'f_lineno', 'f_locals', 'f_restricted', 'f_trace']

localhost:1234> print f.f_trace
None

localhost:1234> inspect.getouterframes(f)
[(<frame object at 0x0237D470>, '<dummy>', 1, '<module>', None, None), 
(<frame object at 0x02375938>, 'C:\\Users\\Paul\\Desktop\\inspect_shell.py', 14
3, 'run_repl', ['                    try: exec compile(data, "<dummy>", "single") in
f_globals, f_globals\n'], 0), (<frame object at 0x023A2C30>, 'C:\
\Python26\\lib\\threading.py', 484, 'run', 
['                self.__target(*self.__args, **self.__kwargs)\n'], 0), 
(<frame object at 0x02397F28>, 'C:\
\Python26\\lib\\threading.py', 532, '__bootstrap_inner', 
['                self.run()\n'], 0), (<frame object at 0x023A9D68>, 
'C:\\Python26\\lib\\thre
ading.py', 504, '__bootstrap', ['            self.__bootstrap_inner()\n'], 0)]

Answer 1

这有点棘手，但鉴于源代码（或一个好的python反编译器），你可以这样做：

在'nine.py'中：

def number_nine():
    x = 9
    if x == 9:
        print 'x is Nine!'
    else:
        print 'x:', x

你的邪恶代码：

from nine import number_nine

我们需要使用ast，它是抽象语法树 ：

import inspect
import ast

现在我们得到源并将其转换为ast：

# Assuming you have the source, we can generate AST from it
nine_src = inspect.getsource(number_nine)
nine_ast = ast.parse(nine_src)

隔离要更改的特定语句：

# This is the Assign object, which represents the 'x = 9' line
# Try to run it interactivly and see how it looks...
x_assign = nine_ast.body[0].body[0]

# Prints 'x'
print x_assign.targets[0].id
# Prints 9
print x_assign.value.n

并根据您的需要改变它：

# Change the value of x
# Notice, that we change the assignment itself, a.k.a `x = 9` is now `x = "It's a trap!"`
x_assign.value.n = "It's a trap!"

现在剩下要做的就是将修改后的ast对象编译成更有用的东西：

# Compile the new function
new_nine = compile(nine_ast, 'new_nine', 'exec')

您可以使用简单的exec （如果它在globals中将替换'number_nine'）或exec in ，并将其放在temp模块中：

# Now we need to execute our litle new_nine (which is a code object)
# This to create the modified version in 'm'
import types
m = types.ModuleType('m', 'The m module')
exec new_nine in m.__dict__
m.number_nine()

# Or this to create it in the global scope
exec new_nine
number_nine()

砰！ 它打印x: It's a trap!

Answer 2

Inspect Shell的作者:)你可能需要从全局命名空间访问x。 Inspect Shell实际上会将您放入正在运行的脚本的全局命名空间中，因此如果从那里无法访问某些数据，那么实现它将非常棘手。

所以真的建议是，make x global，然后你应该能够改变它，你的number_nine（）函数将返回。

希望有所帮助！