Export SQL Script

[riziles]

I love Marimo, especially for its ability to combine SQL and Python into a single platform. Recently, though, I used Marimo to write a SQL script that I had to deploy to an environment without Python, so I wrote a script to export JUST the SQL cells. Posting here in case it's useful for anyone else (I don't have the ability to submit a PR right now):

"""Convert Marimo file to SQL script."""

import ast
from pathlib import Path

import cyclopts
from marimo._cli.export.commands import (
    ExportResult,
    MarimoPath,
    export_as_script,
    watch_and_export,
)
from pydantic import FilePath, validate_call


def export_as_sql_script(
    marimo_path: MarimoPath,
) -> ExportResult:
    """Convert Marimo file to SQL script."""
    py_export = export_as_script(marimo_path)
    if py_export.did_error:
        return py_export

    py_parse = ast.parse(py_export.contents)

    sql_string = ""

    for leaf_0 in py_parse.body:
        if (
            isinstance(leaf_0, ast.Assign)
            and isinstance(func_call := leaf_0.value, ast.Call)
            and isinstance(attr1 := func_call.func, ast.Attribute)
            and isinstance(attr_val := attr1.value, ast.Name)
            and attr_val.id == "mo"
            and attr1.attr == "sql"
            and isinstance(jstr := func_call.args[0], ast.JoinedStr)
        ):
            sql_string += "".join(
                [str(i.value) for i in jstr.values if isinstance(i, ast.Constant)],
            )

    return ExportResult(
        contents=sql_string,
        download_filename=py_export.download_filename,
        did_error=False,
    )


@validate_call
def marimo_export_sql_script(
    name: FilePath,
    output: Path | None = None,
    *,
    watch: bool = False,
    sandbox: bool | None = None,
    force: bool = False,
) -> None:
    """Export a marimo notebook as a sql script, in topological order."""
    import sys  # noqa: PLC0415

    # Set default, if not provided
    if sandbox is None:
        from marimo._cli.sandbox import maybe_prompt_run_in_sandbox  # noqa: PLC0415

        sandbox = maybe_prompt_run_in_sandbox(str(name))

    if sandbox:
        from marimo._cli.sandbox import run_in_sandbox  # noqa: PLC0415

        run_in_sandbox(sys.argv[1:], name=str(name))
        return None

    return watch_and_export(
        MarimoPath(name),
        output,
        watch,
        export_as_sql_script,
        force,
    )


if __name__ == "__main__":

    cyclopts.run(marimo_export_sql_script)