diff options
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.py | 16 |
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 |