python - 如何使调用者的命名空间可用于导入函数中的 IPython 魔法？

Question

我正在尝试调用在 'timex.py' 中定义的一个小辅助函数，它是从 'caller.py' 调用的，而后者又应该通过“%run caller.py”从(PyCharm 的)IPython 控制台调用。

这当前导致“NameError:名称'pickle'未定义”。

# timex.py

from IPython import get_ipython

def time_expression(exp: str):    
    print(f"time expression: {exp}", flush=True)
    return get_ipython().magic(f"timeit {exp}")

# caller.py

import pickle
from timex import time_expression

packet = [*range(10**2)]
time_expression("pickle.dumps(packet, protocol=-1)")

# Then run from IPython-console:

>>>%run caller.py
time expression: pickle.dumps(packet, protocol=-1)
NameError: name 'pickle' is not defined

显然 ipython timeit-magic 的命名空间不包含
caller.py-globals。将 globals() 作为 time_expression 的参数传递并从 time_expression 中更新 globals 没有帮助，并且将 get_ipython() 作为参数传递也不起作用。

有没有办法让它工作？

如果我在 caller.py 中定义 time_expression ，它会起作用并使用“ipython caller.py”从终端调用，但我更愿意将它放在一个库中，然后将其导入到需要的地方。

(Python 3.6.3，IPython 6.2.1)

Answer 1

好吧，看来我很幸运。
我在这里找到了一个提示 link ，shell 有一个 'user_ns' 参数。
然后用调用者的本地人更新这个 user_ns-dictionary 是一个解决方案，通过从 IPython 控制台运行 %run caller.py 来使其从 caller.py 导入。

这有点老套，我现在觉得很脏;)
也许有更优雅的方式？

import sys
from IPython import get_ipython

def time_expression(exp: str):
    """"""
    ipy = get_ipython()
    ipy.__dict__['user_ns'].update(sys._getframe(1).f_locals)
    print(f"time expression: {exp}", flush=True)
    return ipy.magic(f"timeit {exp}")