python - 使用 IPython 作为有效的调试器

Question

如何在我的代码中嵌入 IPython shell 并让它自动显示行号和调用它的函数？

我目前有以下设置来在我的代码中嵌入 IPython shell:

from IPython.frontend.terminal.embed import InteractiveShellEmbed
from IPython.config.loader import Config

# Configure the prompt so that I know I am in a nested (embedded) shell
cfg = Config()
prompt_config = cfg.PromptManager
prompt_config.in_template = 'N.In <\\#>: '
prompt_config.in2_template = '   .\\D.: '
prompt_config.out_template = 'N.Out<\\#>: '

# Messages displayed when I drop into and exit the shell.
banner_msg = ("\n**Nested Interpreter:\n"
"Hit Ctrl-D to exit interpreter and continue program.\n"
"Note that if you use %kill_embedded, you can fully deactivate\n"
"This embedded instance so it will never turn on again")   
exit_msg = '**Leaving Nested interpreter'

# Put ipshell() anywhere in your code where you want it to open.
ipshell = InteractiveShellEmbed(config=cfg, banner1=banner_msg, exit_msg=exit_msg)

这使我可以通过使用 ipshell() 在我的代码中的任何地方启动一个完整的 IPython shell。例如下面的代码:

a = 2
b = a
ipshell()

在调用者范围内启动一个 IPython shell，允许我检查 a 和 b 的值。

我想做的是在每次调用 ipshell() 时自动运行以下代码:

frameinfo = getframeinfo(currentframe())
print 'Stopped at: ' + frameinfo.filename + ' ' +  str(frameinfo.lineno)

这将始终显示 IPython shell 启动的上下文，以便我知道我正在调试的文件/函数等。

也许我可以用一个装饰器来做到这一点，但到目前为止我所有的尝试都失败了，因为我需要 ipshell() 来运行在原始上下文中(这样我可以从 IPython shell 访问 a 和 b。

我怎样才能做到这一点？

Answer 1

您可以调用ipshell()来自另一个用户定义的函数，例如ipsh()

from inspect import currentframe

def ipsh():
    frame = currentframe().f_back
    msg = 'Stopped at {0.f_code.co_filename} and line {0.f_lineno}'.format(frame)
    ipshell(msg,stack_depth=2) # Go back one level!

然后使用ipsh()每当您想进入 IPython shell 时。

说明:

• stack_depth=2问ipshell在为新的 IPython shell 检索 namespace 时上一级(默认为 1 )。
• currentframe().f_back()检索前一帧，以便您可以打印ipsh()所在位置的行号和文件被称为。