baccenfutter/ohmyapi
1
2
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

🎨 Improve shell experience

Browse source 
- more informative banner
- make project singleton available via identifier `p`
This commit is contained in:
Brian Wiborg 2025-09-27 15:36:07 +02:00
parent 35e6ddfcf5
commit aea68b8128
No known key found for this signature in database
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/ohmyapi/cli.py
View file

@ -9,7 +9,9 @@ from ohmyapi.core import scaffolding, runtime
from pathlib import Path from pathlib import Path
app = typer.Typer(help="OhMyAPI — Django-flavored FastAPI scaffolding with tightly integrated TortoiseORM.") app = typer.Typer(help="OhMyAPI — Django-flavored FastAPI scaffolding with tightly integrated TortoiseORM.")
banner = """OhMyAPI Shell | Project: {project_name}""" banner = """OhMyAPI Shell | Project: {project_name}
Find your loaded project singleton via identifier: `p`
"""
@app.command() @app.command()
@ -46,16 +48,14 @@ def shell(root: str = "."):
try: try:
from IPython import start_ipython from IPython import start_ipython
shell_vars = { shell_vars = {
"settings": project.settings, "p": project,
"project": Path(project_path).resolve(),
} }
from traitlets.config.loader import Config from traitlets.config.loader import Config
c = Config() c = Config()
c.TerminalIPythonApp.display_banner = True c.TerminalIPythonApp.display_banner = True
c.TerminalInteractiveShell.banner1 = banner.format(**{ c.TerminalInteractiveShell.banner2 = banner.format(**{
"project_name": f"{f'{project.settings.PROJECT_NAME} ' if getattr(project.settings, 'PROJECT_NAME', '') else ''}[{Path(project_path).resolve()}]", "project_name": f"{f'{project.settings.PROJECT_NAME} ' if getattr(project.settings, 'PROJECT_NAME', '') else ''}[{Path(project_path).resolve()}]",
}) })
c.TerminalInteractiveShell.banner2 = " "
start_ipython(argv=[], user_ns=shell_vars, config=c) start_ipython(argv=[], user_ns=shell_vars, config=c)
except ImportError: except ImportError:
typer.echo("IPython is not installed. Falling back to built-in Python shell.") typer.echo("IPython is not installed. Falling back to built-in Python shell.")