jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-28 04:52:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

[BugFix] Fix to_df in OptionsChainsData (#7109)

Browse Source 
* fix optionschains to_df handling

* lint
This commit is contained in:
Danglewood
2025-05-04 22:25:19 -07:00
committed by GitHub
parent 5be41233d4
commit 6a79d12640
1 changed files with 23 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
openbb_platform/core/openbb_core/app/model/obbject.py
View File

@@ -1,5 +1,8 @@
"""The OBBject."""
# pylint: disable=too-many-branches, too-many-locals, too-many-statements
from collections.abc import Hashable
from typing import (
TYPE_CHECKING,
Any,
@@ -7,7 +10,6 @@ from typing import (
ClassVar,
Dict,
Generic,
Hashable,
List,
Literal,
Optional,
@@ -122,7 +124,7 @@ class OBBject(Tagged, Generic[T]):
"""
return self.to_dataframe(index=index, sort_by=sort_by, ascending=ascending)
def to_dataframe(
def to_dataframe( # noqa: PLR0912
self,
index: Optional[Union[str, None]] = "date",
sort_by: Optional[str] = None,
@@ -182,10 +184,25 @@ class OBBject(Tagged, Generic[T]):
# BaseModel
if isinstance(res, BaseModel):
res_dict = res.model_dump(exclude_unset=True, exclude_none=True)
series = Series(res_dict, name=res.__class__.__name__)
df = series.to_frame().reset_index()
sort_columns = False
res_dict = res.model_dump( # pylint: disable=no-member
exclude_unset=True, exclude_none=True
)
# Model is serialized as a dict[str, list] or list[dict]
if (
(
isinstance(res_dict, dict)
and res_dict
and all(isinstance(v, list) for v in res_dict.values())
)
or isinstance(res_dict, list)
and all(isinstance(item, dict) for item in res_dict)
):
df = DataFrame(res_dict)
sort_columns = False
else:
series = Series(res_dict, name=res.__class__.__name__)
df = series.to_frame().reset_index()
sort_columns = False
# Dict[str, Any]
elif isinstance(res, dict):