initial magics

Author: OutlyingWest

src/magic_extension/__init__.py

@@ -0,0 +1,8 @@
+"""An example magic"""
+__version__ = '0.0.1'
+
+from .magic import KernelMagics
+
+def load_ipython_extension(ipython):
+    print("magic_extension loaded!")
+    ipython.register_magics(KernelMagics)

src/magic_extension/magic.py

@@ -0,0 +1,65 @@
+import sys
+
+from IPython.core.magic import (Magics, magics_class, line_magic, cell_magic)
+from IPython.display import display, Markdown
+
+from jumper.context import kernel_context
+from jumper.kernel import KernelMode
+
+from jumper.perfdatahandler import PerformanceDataHandler
+
+
+@magics_class
+class KernelMagics(Magics):
+    def __init__(self, shell):
+        super(KernelMagics, self).__init__(shell)
+        self.perfdata_handler = PerformanceDataHandler()
+        self.mode = KernelMode.DEFAULT
+
+    @cell_magic
+    def abra(self, line, cell):
+        print('HELLO!\n', cell)
+
+    @cell_magic
+    def set_perfmonitor(self, line, code):
+        """
+        Read the perfmonitor and try to select it.
+        """
+        if self.mode == KernelMode.DEFAULT:
+            monitor = code.split("\n")[1]
+            if monitor in {"local", "localhost", "LOCAL", "LOCALHOST"}:
+                self.cell_output(
+                    "Selected local monitor. No parallel monitoring."
+                )
+            else:
+                try:
+                    self.perfdata_handler.set_monitor(monitor)
+                    kernel_context.nodelist = self.perfdata_handler.get_nodelist()
+                    if len(kernel_context.nodelist) <= 1:
+                        kernel_context.nodelist = None
+                        self.cell_output(
+                            "Found monitor: "
+                            + str(monitor)
+                            + " but no nodelist, using local setup. "
+                        )
+                    else:
+                        self.cell_output(
+                            "Selected monitor: "
+                            + str(monitor)
+                            + " and got nodes: "
+                            + str(kernel_context.nodelist)
+                        )
+                except Exception as e:
+                    self.cell_output(f"Error setting monitor\n{e}", "stderr")
+        else:
+            self.cell_output(
+                f"KernelWarning: Currently in {self.mode}, command ignored.",
+                "stderr",
+            )
+
+    @staticmethod
+    def cell_output(string, stream="stdout"):
+        if stream == "stderr":
+            print(string, file=sys.stderr)
+        else:
+            print(string)