This commit is contained in:
tehcoderer
2023-08-04 14:59:52 -04:00
parent 4de03eafe5
commit 45840c74fb
3 changed files with 5 additions and 12 deletions

View File

@@ -5,7 +5,6 @@ from datetime import datetime
from typing import Any, Dict, List, Literal, Optional
from openbb_provider.abstract.fetcher import Fetcher
from openbb_provider.descriptions import DATA_DESCRIPTIONS
from openbb_provider.helpers import get_querystring
from openbb_provider.models.major_indices_eod import (
MajorIndicesEODData,
@@ -33,7 +32,6 @@ class FMPMajorIndicesEODQueryParams(MajorIndicesEODQueryParams):
class FMPMajorIndicesEODData(MajorIndicesEODData):
"""FMP Major Indices end of day Data."""
date: datetime = Field(description=DATA_DESCRIPTIONS.get("date", ""))
adjClose: Optional[float] = Field(
description="Adjusted Close Price of the symbol.",
alias="adj_close",
@@ -54,9 +52,6 @@ class FMPMajorIndicesEODData(MajorIndicesEODData):
alias="change_percent",
default=None,
)
vwap: Optional[float] = Field(
description="Volume Weighted Average Price of the symbol.", default=None
)
label: Optional[str] = Field(
description="Human readable format of the date.", default=None
)

View File

@@ -10,7 +10,7 @@ from openbb_provider.models.major_indices_eod import (
MajorIndicesEODData,
MajorIndicesEODQueryParams,
)
from pydantic import Field, PositiveFloat, PositiveInt, validator
from pydantic import Field, PositiveInt, validator
from openbb_polygon.utils.helpers import get_data
@@ -47,15 +47,12 @@ class PolygonMajorIndicesEODData(MajorIndicesEODData):
"low": "l",
"close": "c",
"volume": "v",
"vwap": "vw",
}
t: datetime = Field(description="The timestamp of the data.")
n: PositiveInt = Field(
description="The number of transactions for the symbol in the time period."
)
vw: PositiveFloat = Field(
description="The volume weighted average price of the symbol."
)
@validator("t", pre=True, check_fields=False)
def time_validate(cls, v): # pylint: disable=E0213
@@ -106,5 +103,5 @@ class PolygonMajorIndicesEODFetcher(
@staticmethod
def transform_data(
data: List[PolygonMajorIndicesEODData],
) -> List[PolygonMajorIndicesEODData]:
return data
) -> List[MajorIndicesEODData]:
return [MajorIndicesEODData.parse_obj(d.dict()) for d in data]

View File

@@ -33,3 +33,4 @@ class MajorIndicesEODData(Data):
low: PositiveFloat = Field(description=DATA_DESCRIPTIONS.get("low", ""))
close: PositiveFloat = Field(description=DATA_DESCRIPTIONS.get("close", ""))
volume: float = Field(description=DATA_DESCRIPTIONS.get("volume", ""))
vwap: Optional[float] = Field(description=DATA_DESCRIPTIONS.get("vwap", ""))