2

我不知道如何使用inspect / inspect_shell检查当前正在执行的函数

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

给定这个例子九.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)]
4

2 回答 2

3

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

在“九.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')

您可以使用 simple exec（如果它在 'number_nine' 中，它将替换 'number_nine' globals），或者exec in，并将其放在临时模块中：

# 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!

于 2012-01-04T20:25:39.720 回答
1

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

所以真正的建议是，使 x 全局，然后你应该能够改变它并且你的 number_nine() 函数将返回。

希望有帮助！

于 2012-02-03T07:28:25.930 回答