以编程方式在 Google Colab 中运行所有单元格命令

Question

我需要以编程方式从 Google Colab 菜单“运行时”运行某些命令“全部运行”。它没有任何明显的“onclick”事件处理程序，我可以从该页面上的 javascript 代码中调用它。 页面上的其他“div”可以从js调用，例如，我可以使用js代码连接到运行时：

document.querySelector('#top-toolbar > colab-connect-button').shadowRoot.querySelector('#connect').click();

运行时菜单是一个下拉菜单，我尝试.click() 其中的每个<div> 项目，但没有效果。

“全部运行”命令也有一个热键Ctrl + F9，但向document 元素分派事件无效。但是我可以使用以下代码将Enter 命令发送到笔记本内的任何输入字段：

document.querySelector('input.raw_input').dispatchEvent(new KeyboardEvent('keydown', {key: 'Enter'}))

使用 Chrome 代码检查器 Ctrl + Shift + I 我查看了“全部运行”命令，它看起来像：

<div command="runall" class="goog-menuitem" role="menuitem" id=":1w" style="user-select: none;"><div class="goog-menuitem-content" style="user-select: none;">Run all<span class="goog-menuitem-accel">Ctrl+F9</span></div></div>

所以我在页面上检查器代码的 Sources 选项卡中进行了搜索，发现 https://colab.research.google.com/v2/external/external_polymer_binary.js 文件中出现了“runall”：

, Eja = X(new W({
        id: "runall",
        description: "Run all cells in notebook",
        shortcut: IG(120)

120 - 顺便说一句，是F9 按钮的键码。我还发现我认为需要的菜单项被调用的确切位置：

        case "runall":
            d.runAll();
            break;

但我几乎不可能理解d. 是什么以及它的参考来源！

我还发现了许多其他有趣且有用的命令，例如this.notebook.getKernel().isRunning() 或c.notebook.getKernel().restart()，但问题始终相同：这些命令的根对象是什么？我尝试了document. 和window.，但结果是“未定义”或“不是函数”。我认为我可以在如下字符串中调用 runall() 命令：

document.**SOMETHING I DONT KNOW**.runAll()

我对前端/js 很不好，而且很难在混淆代码中找到一些东西，但是如果我们在连接到所需菜单项的 javascript 代码中具有 .runAll() 这样的功能，我就可以以编程方式运行它从控制台或 javascript 注入

或者也许可以将键盘事件Ctrl + F9 分派给某个元素以运行此命令，因此问题是分派键盘事件所需的对象是什么

Answer 1

出于类似的原因，我花了一段时间梳理那个 javascript 文件，最后弄清楚了如何使它工作。

这是一个以编程方式运行所有单元格的函数：

function runAll() {
  const F9Event = {key: "F9", code: "F9", metaKey: true, keyCode: 120};
  document.dispatchEvent(new KeyboardEvent("keydown", F9Event));
}

请注意 KeyboardEvent.keyCode is deprecated 支持 KeyboardEvent.code，但您仍需要在此处提供它（截至 21 年 5 月 18 日），因为它是 Colab 用于检查键盘输入的属性。

您还可以互换使用 metaKey: true 和 ctrlKey: true，而不管平台如何，因为 Colab 只会检查是否存在 KeyboardEvent.metaKey 或 KeyboardEvent.ctrlKey 以获取需要它们的快捷方式。

我还发现了许多其他有趣且有用的命令，例如 this.notebook.getKernel().isRunning() 或 c.notebook.getKernel().restart()，但问题始终相同：这些命令的根对象是什么？

有一个全局colab 对象提供对某些（但不是全部）功能的访问。大多数东西都可以通过colab.global 访问，例如要重新启动内核，您可以使用：

colab.global.notebook.kernel.restart()