mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-31 22:41:19 +00:00
4400cd9a8f
* small changes - moving provider inside the core * uptade readme * lint * remove old provider readme * leftovers * refactor website examples * ruff * fix nox * relock * fix fails_to_deliver headers * fix for tests that have renamed endpoints * fix discovery * removed test for renamed endpoint * fix broken syntax - test_search_financial_attributes * fix cash/income/balance * fix derivatives tests * fix etf tests * fix news tests * fix regulators * black * wrong test param * sort equity gainers/losers by percent * platform markdown generation import * Hotfix/fix int tests core merge (#5779) * removing tests on old endpoint * removing visualization on old endpoint * fixing equity tests * fixing examples * fix news tests * fix index test * example * add comments * example --------- Co-authored-by: Diogo Sousa <montezdesousa@gmail.com> --------- Co-authored-by: montezdesousa <79287829+montezdesousa@users.noreply.github.com> Co-authored-by: Diogo Sousa <montezdesousa@gmail.com> Co-authored-by: Danglewood <85772166+deeleeramone@users.noreply.github.com> Co-authored-by: Pratyush Shukla <ps4534@nyu.edu>
88 lines
2.3 KiB
Python
88 lines
2.3 KiB
Python
"""FRED Commercial Paper Model."""
|
|
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
from openbb_core.provider.abstract.fetcher import Fetcher
|
|
from openbb_core.provider.standard_models.cp import (
|
|
CommercialPaperData,
|
|
CommercialPaperParams,
|
|
)
|
|
from openbb_fred.utils.fred_base import Fred
|
|
from openbb_fred.utils.fred_helpers import get_cp_series_id
|
|
from pydantic import field_validator
|
|
|
|
|
|
class FREDCommercialPaperParams(CommercialPaperParams):
|
|
"""FRED Commercial Paper Query."""
|
|
|
|
|
|
class FREDCommercialPaperData(CommercialPaperData):
|
|
"""FRED Commercial Paper Data."""
|
|
|
|
__alias_dict__ = {"rate": "value"}
|
|
|
|
@field_validator("rate", mode="before", check_fields=False)
|
|
@classmethod
|
|
def value_validate(cls, v):
|
|
"""Validate rate."""
|
|
try:
|
|
return float(v)
|
|
except ValueError:
|
|
return None
|
|
|
|
|
|
class FREDCommercialPaperFetcher(
|
|
Fetcher[
|
|
FREDCommercialPaperParams,
|
|
List[FREDCommercialPaperData],
|
|
]
|
|
):
|
|
"""Transform the query, extract and transform the data from the FRED endpoints."""
|
|
|
|
data_type = FREDCommercialPaperData
|
|
|
|
@staticmethod
|
|
def transform_query(params: Dict[str, Any]) -> FREDCommercialPaperParams:
|
|
"""Transform query."""
|
|
return FREDCommercialPaperParams(**params)
|
|
|
|
@staticmethod
|
|
def extract_data(
|
|
query: FREDCommercialPaperParams,
|
|
credentials: Optional[Dict[str, str]],
|
|
**kwargs: Any
|
|
) -> list:
|
|
"""Extract data."""
|
|
key = credentials.get("fred_api_key") if credentials else ""
|
|
fred = Fred(key)
|
|
|
|
series = get_cp_series_id(
|
|
maturity=query.maturity,
|
|
category=query.category,
|
|
grade=query.grade,
|
|
)
|
|
|
|
data = []
|
|
|
|
for s in series:
|
|
id_ = s["FRED Series ID"]
|
|
title = s["Title"]
|
|
d = fred.get_series(
|
|
series_id=id_,
|
|
start_date=query.start_date,
|
|
end_date=query.end_date,
|
|
**kwargs,
|
|
)
|
|
for item in d:
|
|
item["title"] = title
|
|
data.extend(d)
|
|
|
|
return data
|
|
|
|
@staticmethod
|
|
def transform_data(
|
|
query: FREDCommercialPaperParams, data: list, **kwargs: Any
|
|
) -> List[FREDCommercialPaperData]:
|
|
"""Transform data."""
|
|
return [FREDCommercialPaperData.model_validate(d) for d in data]
|