ipython cell magic - kroki

gistfile1.txt

from IPython.core.magic import register_cell_magic
from IPython.display import SVG, display

import requests

diagrams_supported = set(['blockdiag', 'seqdiag', 'mermaid', 'actdiag', 
                          'nwdiag', 'packetdiag', 'rackdiag', 'erd',
                          'nomnoml', 'plantuml', 'umlet', 'wavedrom', 
                          'bpmn', 'bytefield', 'pikchr', 'graphviz',
                          'vega', 'vega-lite', 'ditaa', 'svgbob'])

# see https://kroki.io/examples.html#wbs for working examples

@register_cell_magic
def kroki(line, cell):
    "simple wrapper for kroki.io"
    
    if line not in diagrams_supported:
        raise NotImplemented
        
    url = 'https://kroki.io'
    r = requests.post(f'{url}/{line}/svg', json={'diagram_source': cell})
    if r.status_code == 200:
        return display(SVG(r.text))
    else:
        raise ValueError