PythonistaLab / marimo compatibility

[RichardPotthoff]

Hi Ole,

I have been playing with marimo for the last 2 weeks, and I was wondering if the upcoming PythonistaLab app could be made compatible with marimo.

I got some of it working in Pythonista 3 (the parts that don't use "multiprocessing") using this Python script:

# marimo.py (v0)
"""
Custom marimo loader for Pythonista on iOS.

- Adds site-packages to sys.path dynamically.

- Patches sys.std* streams for marimo compatibility.

- Loads marimo explicitly.
"""
modulename = 'marimo'
modulefilepath = "/private/var/mobile/Containers/Data/Application/8C15CF35-580B-488E-9B29-D89BDB144E80/Documents/Documents/site-packages/marimo/__init__.py"
# Add site-packages to path
additional_site_packages="/private/var/mobile/Containers/Data/Application/8C15CF35-580B-488E-9B29-D89BDB144E80/Documents/Documents/site-packages"

import sys
import types
import importlib.util

print(f'Patching module "{modulename}" for iOS.',file=sys.stderr) 

if additional_site_packages and not(additional_site_packages in sys.path):
  sys.path.insert(1,additional_site_packages) #insert after the cwd

# Patch sys.std* streams
for sysstdstream, i in [(sys.stdin, 0), (sys.stdout, 1), (sys.stderr, 2)]:
  if not hasattr(sysstdstream, 'errors'):
    sysstdstream.errors = 'strict'
  if not hasattr(sysstdstream, 'fileno'):
    sysstdstream._fileno = i
    sysstdstream.fileno = types.MethodType(lambda self:self._fileno, sysstdstream)

# Load module explicitly
spec = importlib.util.spec_from_file_location(modulename, modulefilepath)
module = importlib.util.module_from_spec(spec)
sys.modules[modulename] = module
spec.loader.exec_module(module)

if __name__=='__main__':
  from marimo.__main__ import main
  main()
else:
  if ('__main__' in sys.modules) and  not(hasattr(sys.modules['__main__'],'__spec__')):
    sys.modules['__main__'].__spec__==None #necessary for running script in IPython console (for some reason)
  from marimo import *

Here is a Python script that uses marimo to convert an IPython notebook into a marimo notebook, and then exports the marimo notebook into a static wasm html that can be viewed in a browser.

#ipynb_marimo_wasm.py
#This python script uses marimo to first convert an IPython notebook
#inot a marimo notebook, and then convert the marimo notebook into a
#static wasm html that can be viewed interactively in a web browser.
import marimo as mo
from marimo.__main__ import main as marimo_main
import sys,os

if len(sys.argv)>1:
  ipynbPath=sys.argv[1]
else:
  ipynbPath='Counter.ipynb'
  
ipynbName,ipynbExt=os.path.splitext(ipynbPath)

script=[f'marimo -l DEBUG convert {ipynbName+(ipynbExt or ".ipynb")} -o {ipynbName+".py"}',
        f'marimo -l DEBUG export html-wasm {ipynbName+".py"} -o {ipynbName}']
        
for cmdln in script:
  try: 
    sys.argv=cmdln.split()
    print(f"running: {' '.join(sys.argv)}\n ")
    marimo_main()
  except BaseException as e:
    print(f"completed with code: {e}\n")


import webbrowser
from http.server import HTTPServer,SimpleHTTPRequestHandler
from threading import Thread

curdir=os.getcwd()
os.chdir(ipynbName)

Server=HTTPServer(('localhost',8088),SimpleHTTPRequestHandler)
ServerThread=Thread(target=lambda:Server.serve_forever())
ServerThread.daemon=True
ServerThread.start()

webbrowser.open('http://localhost:8088/index.html')

try:
  while True:
    pass
except KeyboardInterrupt:
  Server.shutdown()
  ServerThread.join()
  
os.chdir(curdir)

What does not work is running marimo as server, since that requires multiprocessing, which is not supported on iOS, as far as I know.

I am currently running a marimo server on a RaspberryPi 5 on my local network. That works pretty well, but a native marimo-compatible app for iOS would be even better.