3

如何在我的代码中嵌入 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)

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

a = 2
b = a
ipshell()

在调用者范围内启动一个 IPython shell,允许我检查 和 的ab

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

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

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

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

我怎样才能做到这一点?

4

1 回答 1

6

您可以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在检索新 IPython shell 的命名空间时要求ipshell上一级(默认为1)。
  • currentframe().f_back()检索前一帧,以便您可以打印ipsh()调用位置的行号和文件。
于 2013-03-27T23:03:28.927 回答