使用代码自定义 Jupyter Notebook 单元行为

Question

我正在 Jupyter Notebook 中编写和制作代码插图。我的用例是从某些代码单元中获取最终代码并将其放入 HTML 文档中。我找到了一个非常好的使用pygments包的管道，它为我突出显示代码并将其放入适当的 HTML 中。

from pygments import highlight

from pygments.lexers import PythonLexer

from pygments.formatters import HtmlFormatter

def PyHighlight(code):

    return highlight(code, PythonLexer(), HtmlFormatter())

PyHighlight("print('Hello world!')")

输出：

'<div class="highlight"><pre><span></span><span class="nb">print</span><span class="p">(</span><span class="s1">&#39;Hello world!&#39;</span><span class="p">)</span>\n</pre></div>\n'

但是将每个代码单元格转换成字符串然后传递给函数PyHighlight并最终得到 HTML 对我来说非常繁琐。

有没有办法可以将每个单元格的内容作为字符串获取？更好的是，我可以PyHighlight在每个单元格内容作为参数的单元格之后触发运行，PyHightlight这样我就可以复制粘贴突出显示的代码 HTML 吗？

Answer 1

我从关于Custom Magic 的IPython 文档中获得了一些灵感在单元格中定义并注册以下魔法。from IPython.core.magic import  (Magics, cell_magic)@magics_classclass MyMagics(Magics):    @cell_magic    def pygmented(self, line, cell):        print(PyHighlight(cell))get_ipython().register_magics(MyMagics)之后，只需添加%%pygmented到每个单元格的顶部，运行单元格后，单元格的内容也会在所有突出显示后打印出来（如问题中所问）。