feat: let display hook handle clear_output (#1135)

ipython · Jul 20, 2023 · 58e9d15 · 58e9d15
1 parent 1abb019
commit 58e9d15
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/ipykernel/zmqshell.py b/ipykernel/zmqshell.py
@@ -144,11 +144,17 @@ def clear_output(self, wait=False):
         """
         content = dict(wait=wait)
         self._flush_streams()
+        msg = self.session.msg("clear_output", json_clean(content), parent=self.parent_header)
+
+        # see publish() for details on how this works
+        for hook in self._hooks:
+            msg = hook(msg)
+            if msg is None:
+                return
+
         self.session.send(
             self.pub_socket,
-            "clear_output",
-            content,
-            parent=self.parent_header,
+            msg,
             ident=self.topic,
         )