Function _launch_ptpython_shell

tortoise/cli/cli.py:119–148

Launch ptpython shell asynchronously. Args: namespace: The namespace dict to make available in the shell

(namespace: dict[str, Any])

async def _launch_ptpython_shell(namespace: dict[str, Any]) -> None:
    """Launch ptpython shell asynchronously.

    Args:
        namespace: The namespace dict to make available in the shell
    """
    # Apply Windows patch for ptpython signal handler issues
    if platform.system() == "Windows":
        _patch_loop_factory_for_ptpython()

    model_names = [
        k for k in namespace.keys() if k not in ("Tortoise", "tortoise", "connections", "apps")
    ]

    # Print banner before launching ptpython
    models_info = (
        f"Available models: {', '.join(model_names)}" if model_names else "No models loaded"
    )
    print("Tortoise ORM Shell (ptpython)")
    print(models_info)
    print("Use 'await' directly for async operations (e.g., 'await YourModel.all()').\n")

    with contextlib.suppress(EOFError, ValueError):
        await ptpython_embed(
            globals=namespace,
            title="Tortoise Shell",
            vi_mode=True,
            return_asyncio_coroutine=True,
            patch_stdout=True,
        )


@contextlib.asynccontextmanager