mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 20:42:26 +00:00
9a30186174
* stash some changes * add more robust testing * mypy * point PR at V5 * introduce spec file * codespell * test fix * fix workflow environment setup * fix workflow environment setup * fix workflow environment setup * add pyyaml to dependencies * split lint jobs * fix workflow environment setup * fix workflow environment setup * workflow env setup * workflow env setup * clean up code comments * add auth hook entrypoints * codespell * add codegen feature * codespell * move _unpack into dispatchers for consistency with codegen packages * surface nested models in the response * fix missing coverage in CI * socrata updates * test fix * detect plotly output * add --include and --exclude flags from generate-extension command * cap test matrix at python 3.14 * no useless comments * platform controller command description split * merge URL overloads from path params * exclude none and unset from model dump --------- Co-authored-by: deeleeramone <> Co-authored-by: Copilot <copilot@github.com>
110 lines
3.6 KiB
Python
110 lines
3.6 KiB
Python
"""Rich table output adapter."""
|
|
|
|
from typing import Any
|
|
|
|
import pandas as pd
|
|
|
|
from openbb_cli.controllers.utils import print_rich_table
|
|
from openbb_cli.session import Session
|
|
|
|
session = Session()
|
|
|
|
|
|
class RichTableOutput:
|
|
"""Rich table output adapter with truncation for terminal display."""
|
|
|
|
def display( # noqa: PLR0911
|
|
self,
|
|
data: Any,
|
|
title: str = "",
|
|
export: bool = False,
|
|
chart: bool = False,
|
|
) -> None:
|
|
"""Display using rich table with row/column limits."""
|
|
if export:
|
|
return
|
|
|
|
if hasattr(data, "model_dump"):
|
|
if chart:
|
|
try:
|
|
data.show()
|
|
return
|
|
except Exception as e:
|
|
session.console.print(f"[yellow]Chart not available: {e}[/yellow]")
|
|
|
|
if session.settings.USE_INTERACTIVE_DF and session.backend is not None:
|
|
try:
|
|
data.charting.table()
|
|
return
|
|
except Exception as e:
|
|
session.console.print(
|
|
f"[yellow]Interactive table not available: {e}[/yellow]"
|
|
)
|
|
|
|
results = data.model_dump(exclude_unset=True, exclude_none=True).get(
|
|
"results"
|
|
)
|
|
|
|
if results is None:
|
|
session.console.print("[yellow]No results to display[/yellow]")
|
|
return
|
|
|
|
try:
|
|
if isinstance(results, pd.DataFrame):
|
|
df = results
|
|
elif isinstance(results, list):
|
|
if not results:
|
|
session.console.print("[yellow]Empty results[/yellow]")
|
|
return
|
|
df = pd.DataFrame(results)
|
|
elif isinstance(results, dict):
|
|
df = pd.DataFrame([results])
|
|
else:
|
|
session.console.print(results)
|
|
return
|
|
except Exception as e:
|
|
session.console.print(f"[yellow]Cannot display as table: {e}[/yellow]")
|
|
session.console.print(results)
|
|
return
|
|
|
|
elif isinstance(data, pd.DataFrame):
|
|
df = data
|
|
if session.settings.USE_INTERACTIVE_DF and session.backend is not None:
|
|
try:
|
|
session.backend.send_table(
|
|
df_table=df,
|
|
title=title,
|
|
theme=session.user.preferences.table_style,
|
|
)
|
|
return
|
|
except Exception: # noqa: S110
|
|
pass
|
|
if df.empty:
|
|
session.console.print("[yellow]Empty DataFrame[/yellow]")
|
|
return
|
|
elif isinstance(data, pd.Series):
|
|
df = data.to_frame()
|
|
elif isinstance(data, dict):
|
|
try:
|
|
df = pd.DataFrame.from_dict(data, orient="columns")
|
|
except Exception:
|
|
session.console.print(data)
|
|
return
|
|
elif isinstance(data, (list, tuple)):
|
|
if not data:
|
|
session.console.print("[yellow]Empty data[/yellow]")
|
|
return
|
|
try:
|
|
df = pd.DataFrame(data)
|
|
except Exception:
|
|
session.console.print(data)
|
|
return
|
|
else:
|
|
session.console.print(data)
|
|
return
|
|
|
|
if isinstance(df.columns, pd.RangeIndex):
|
|
df.columns = [str(i) for i in df.columns]
|
|
|
|
print_rich_table(df=df, show_index=True, title=title, export=export)
|