python - IPython 运行魔术 : How create an alias for "run -i"?

Question

我正在编写一个 python 用户界面来控制实验室中的各种仪器。如果脚本不是交互式运行，与仪器的连接会在脚本结束时丢失，这可能非常糟糕。我想帮助用户“记住”以交互方式运行脚本。

我正在考虑两种可能的方法。首先，如标题中所述，我可以为 run -i 创建一个别名:

%alias_magic lab_run run -i

但这会返回一个错误:

UsageError: unrecognized arguments: -i 

有办法解决这个问题吗？

或者，我可以检测脚本内部是否传递了 -i 标志，如果没有传递则引发错误。但是，它没有出现在 sys.argv 列表中:

In [1]: import sys
In [2]: run -i test.py random args
['test.py', 'random', 'args']

我不能使用 ipy 文件，因为我需要读取 %run 标志，正如我之前的问题中所解释的: How to add a custom flag to IPython's magic commands? (.ipy files)

有人看到这个问题的解决方案吗？

Answer 1

你可以定义你自己的魔法函数并在其中使用%run -i:

from IPython.core.magic import register_line_magic

@register_line_magic
def r(line):
    get_ipython().magic('run -i ' + line)

del r

编辑

正如 hpaulj 指出的那样，magic 已被弃用。这里有一个新的版本 run_line_magic:

from IPython.core.magic import register_line_magic

@register_line_magic
def r(line):
    get_ipython().run_line_magic('run', ' -i ' + line)

del r
Now:

%r

做同样的事:

%run -i