summaryrefslogtreecommitdiffstats
path: root/openbb_platform/core/openbb_core/provider/standard_models/analyst_estimates.py
diff options
context:
space:
mode:
Diffstat (limited to 'openbb_platform/core/openbb_core/provider/standard_models/analyst_estimates.py')
-rw-r--r--openbb_platform/core/openbb_core/provider/standard_models/analyst_estimates.py16
1 files changed, 11 insertions, 5 deletions
diff --git a/openbb_platform/core/openbb_core/provider/standard_models/analyst_estimates.py b/openbb_platform/core/openbb_core/provider/standard_models/analyst_estimates.py
index 7ffe2462b0a..4a4e76c796d 100644
--- a/openbb_platform/core/openbb_core/provider/standard_models/analyst_estimates.py
+++ b/openbb_platform/core/openbb_core/provider/standard_models/analyst_estimates.py
@@ -1,7 +1,7 @@
"""Analyst Estimates Standard Model."""
from datetime import date as dateType
-from typing import List, Literal, Set, Union
+from typing import List, Literal, Optional, Set, Union
from pydantic import Field, field_validator
@@ -24,10 +24,16 @@ class AnalystEstimatesQueryParams(QueryParams):
@field_validator("symbol", mode="before", check_fields=False)
@classmethod
- def upper_symbol(cls, v: str) -> str:
- """Convert symbol to uppercase."""
+ def to_upper(cls, v: str) -> str:
+ """Convert field to uppercase."""
return v.upper()
+ @field_validator("period", mode="before", check_fields=False)
+ @classmethod
+ def to_lower(cls, v: Optional[str]) -> Optional[str]:
+ """Convert field to lowercase."""
+ return v.lower() if v else v
+
class AnalystEstimatesData(Data):
"""Analyst Estimates data."""
@@ -71,8 +77,8 @@ class AnalystEstimatesData(Data):
@field_validator("symbol", mode="before", check_fields=False)
@classmethod
- def upper_symbol(cls, v: Union[str, List[str], Set[str]]):
- """Convert symbol to uppercase."""
+ def to_upper(cls, v: Union[str, List[str], Set[str]]):
+ """Convert field to uppercase."""
if isinstance(v, str):
return v.upper()
return ",".join([symbol.upper() for symbol in list(v)]) if v else None