summaryrefslogtreecommitdiffstats
path: root/openbb_platform/core/openbb_core/provider/standard_models/recent_performance.py
blob: 37314e80c50f7bf3b229c7471ad9703c8d4de903 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
"""Recent Performance Standard Model."""

from typing import Optional

from pydantic import Field, field_validator

from openbb_core.provider.abstract.data import Data
from openbb_core.provider.abstract.query_params import QueryParams
from openbb_core.provider.utils.descriptions import QUERY_DESCRIPTIONS


class RecentPerformanceQueryParams(QueryParams):
    """Recent Performance Query."""

    symbol: str = Field(description=QUERY_DESCRIPTIONS.get("symbol", ""))

    @field_validator("symbol")
    @classmethod
    def to_upper(cls, v: str) -> str:
        """Convert field to uppercase."""
        return v.upper()


class RecentPerformanceData(Data):
    """Recent Performance Data."""

    one_day: Optional[float] = Field(description="One-day return.", default=None)
    wtd: Optional[float] = Field(description="Week to date return.", default=None)
    one_week: Optional[float] = Field(description="One-week return.", default=None)
    mtd: Optional[float] = Field(description="Month to date return.", default=None)
    one_month: Optional[float] = Field(description="One-month return.", default=None)
    qtd: Optional[float] = Field(description="Quarter to date return.", default=None)
    three_month: Optional[float] = Field(
        description="Three-month return.", default=None
    )
    six_month: Optional[float] = Field(description="Six-month return.", default=None)
    ytd: Optional[float] = Field(description="Year to date return.", default=None)
    one_year: Optional[float] = Field(description="One-year return.", default=None)
    three_year: Optional[float] = Field(description="Three-year return.", default=None)
    five_year: Optional[float] = Field(description="Five-year return.", default=None)
    ten_year: Optional[float] = Field(description="Ten-year return.", default=None)
    max: Optional[float] = Field(
        description="Return from the beginning of the time series.", default=None
    )