Function new

marimo/_cli/cli.py:732–830

(
    port: int | None,
    host: str,
    proxy: str | None,
    headless: bool,
    token: bool,
    token_password: str | None,
    token_password_file: str | None,
    base_url: str,
    sandbox: bool | None,
    skew_protection: bool,
    timeout: float | None,
    prompt: str | None,
)

)
@click.argument("prompt", required=False)
def new(
    port: int | None,
    host: str,
    proxy: str | None,
    headless: bool,
    token: bool,
    token_password: str | None,
    token_password_file: str | None,
    base_url: str,
    sandbox: bool | None,
    skew_protection: bool,
    timeout: float | None,
    prompt: str | None,
) -> None:
    if sandbox:
        from marimo._cli.sandbox import run_in_sandbox

        # TODO: consider adding recommended as well
        run_in_sandbox(sys.argv[1:], name=None, additional_features=["lsp"])
        return

    workspace: NotebookWorkspace | None = None

    if prompt is None:
        # We support unix-style prompting, cat prompt.txt | marimo new
        prompt = _get_stdin_contents()

    if prompt is not None:
        import tempfile

        from marimo._ai.text_to_notebook import text_to_notebook

        try:
            _maybe_path = Path(prompt)
            if _maybe_path.is_file():
                prompt = _maybe_path.read_text(encoding="utf-8")
        except OSError:
            # is_file() fails when, for example, the "filename" (prompt) is too long
            pass

        temp_file = None
        try:
            notebook_content = text_to_notebook(prompt)
            # On Windows, NamedTemporaryFile cannot be reopened unless
            # delete=False.
            with tempfile.NamedTemporaryFile(
                suffix=".py", mode="w", encoding="utf-8", delete=False
            ) as temp_file:
                temp_file.write(notebook_content)
            workspace = infer_workspace(temp_file.name)

            def _cleanup() -> None:
                try:
                    os.unlink(temp_file.name)
                except Exception:
                    pass

            atexit.register(_cleanup)