python-3.x - IPython 自定义完成器的扩展信息

Question

我正在使用 set_custom_completer 设置自定义完成器:

import IPython

def foo(self, t):
    return []

IPython.get_ipython().set_custom_completer(foo)

问题出在 foo 的签名上:参数 t 只是一个 string，包含从行首到光标。有没有办法找到整个单元格内容和光标位置？

例如，假设单元格中的状态是:

foo
bar<TAB>baz

然后 t 将是 bar，但我想要像

这样的东西

(`foo\barbaz`, 
1, # line 1
4 # cursor position 4 in the line
)

系统信息为:

The version of the notebook server is 5.0.0b2 and is running on:
Python 3.6.3rc1+ (default, Sep 29 2017, 16:55:05) 
[GCC 5.x 20170328]

Current Kernel Information:
Python 3.6.3rc1+ (default, Sep 29 2017, 16:55:05) 
Type "copyright", "credits" or "license" for more information.

IPython 5.3.0 -- An enhanced Interactive Python.

不幸的是，我无法升级它。

Answer 1

在挖掘源代码和堆栈跟踪之后，我找不到任何明显暴露单元格文本的东西。但是我对 ipython source 没有详细的想法，所以我制定了以下技巧，可以满足您的需求:

import IPython
import inspect

def foo(self, t):
    locals_caller = inspect.currentframe().f_back.f_back.f_back.f_back.f_locals
    code = locals_caller['code']
    cursor_pos = locals_caller['cursor_pos']
    # remove any reference to avoid leakages
    del locals_caller
    return [code]

IPython.get_ipython().set_custom_completer(foo)

我已经对堆栈回溯进行了硬编码，但如果您想要一个跨版本/更新工作的稳定函数，您可能需要围绕它添加一个逻辑。这应该足以让您朝着正确的方向前进。