diff options
Diffstat (limited to 'openbb_platform/providers/intrinio/openbb_intrinio/models/cash_flow.py')
-rw-r--r-- | openbb_platform/providers/intrinio/openbb_intrinio/models/cash_flow.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/openbb_platform/providers/intrinio/openbb_intrinio/models/cash_flow.py b/openbb_platform/providers/intrinio/openbb_intrinio/models/cash_flow.py index c8fffcc5237..3441983152d 100644 --- a/openbb_platform/providers/intrinio/openbb_intrinio/models/cash_flow.py +++ b/openbb_platform/providers/intrinio/openbb_intrinio/models/cash_flow.py @@ -2,8 +2,8 @@ # pylint: disable=unused-argument -import warnings from typing import Any, Dict, List, Literal, Optional +from warnings import warn from openbb_core.provider.abstract.fetcher import Fetcher from openbb_core.provider.standard_models.cash_flow import ( @@ -14,8 +14,6 @@ from openbb_core.provider.utils.helpers import ClientResponse, amake_requests from openbb_intrinio.utils.helpers import get_data_one from pydantic import Field, field_validator, model_validator -_warn = warnings.warn - class IntrinioCashFlowStatementQueryParams(CashFlowStatementQueryParams): """Intrinio Cash Flow Statement Query. @@ -24,7 +22,10 @@ class IntrinioCashFlowStatementQueryParams(CashFlowStatementQueryParams): Source: https://docs.intrinio.com/documentation/web_api/get_fundamental_standardized_financials_v2 """ - period: Literal["annual", "quarter", "ttm", "ytd"] = Field(default="annual") + period: Literal["annual", "quarter", "ttm", "ytd"] = Field( + default="annual", + json_schema_extra={"choices": ["annual", "quarter", "ttm", "ytd"]}, + ) fiscal_year: Optional[int] = Field( default=None, description="The specific fiscal year. Reports do not go beyond 2008.", @@ -265,7 +266,7 @@ class IntrinioCashFlowStatementFetcher( ) if query.fiscal_year is not None: if query.fiscal_year < 2008: - _warn("Financials data is only available from 2008 and later.") + warn("Financials data is only available from 2008 and later.") query.fiscal_year = 2008 fundamentals_url = fundamentals_url + f"&fiscal_year={query.fiscal_year}" fundamentals_url = fundamentals_url + f"&api_key={api_key}" |