stack/notebooks/_template.py

import marimo

__generated_with = "0.20.4"
app = marimo.App(width="full")


@app.cell(hide_code=True)
def _():
    import marimo as mo

    return (mo,)


@app.cell(hide_code=True)
def _():
    import altair as alt
    import polars as pl

    from conf import connect

    connect.theme()

    con = connect.duckdb()
    _s3 = connect.obstore()  # noqa: F841 — marimo auto-discovers in Files panel

    def q(sql):
        """Run SQL and return a Polars DataFrame."""
        return con.execute(sql).pl()

    return alt, con, pl, q


@app.cell(hide_code=True)
def _(mo):
    mo.md("""
    # Notebook Title

    Description of the analysis.
    """)
    return


@app.cell(hide_code=True)
def _(mo, q):
    schemas = q("""
        SELECT table_schema AS schema, count(*) AS tables
        FROM information_schema.tables
        WHERE table_schema != 'information_schema'
        GROUP BY table_schema
        ORDER BY tables DESC
    """)
    mo.ui.table(schemas, label="Available Schemas")
    return


if __name__ == "__main__":
    app.run()