cleanup in stocks.ca.get

This commit is contained in:
Pratyush Shukla
2023-07-15 22:12:12 +05:30
parent 492aae6e37
commit ee95fc4eeb
2 changed files with 3 additions and 10 deletions

View File

@@ -9,9 +9,6 @@ from openbb_provider.helpers import data_transformer
# IMPORT INTERNAL
from openbb_provider.models.stock_peers import StockPeersData, StockPeersQueryParams
# IMPORT THIRD-PARTY
from pydantic import Field
from .helpers import create_url, get_data_many
# FMP SPECIFIC FUNCTIONALITY CURRENTLY
@@ -32,9 +29,6 @@ class FMPStockPeersQueryParams(StockPeersQueryParams):
class FMPStockPeersData(StockPeersData):
"""FMP Stock Peers data."""
symbol: str = Field(min_length=1)
peers_list: List[str] = Field(alias="peersList")
class FMPStockPeersFetcher(
Fetcher[
@@ -55,7 +49,7 @@ class FMPStockPeersFetcher(
query: FMPStockPeersQueryParams, api_key: str
) -> List[FMPStockPeersData]:
url = create_url(4, "stock_peers", api_key, query)
return get_data_many(url, FMPStockPeersData, "peersList")
return get_data_many(url, FMPStockPeersData)
@staticmethod
def transform_data(data: List[FMPStockPeersData]) -> List[StockPeersData]:

View File

@@ -1,7 +1,6 @@
"""Stock Peers data model."""
from typing import List
from typing import List, Optional
from openbb_provider.abstract.data import Data, QueryParams
from openbb_provider.models.base import BaseSymbol
@@ -29,4 +28,4 @@ class StockPeersData(Data):
"""
symbol: str
peers_list: List[str]
peers_list: Optional[List[str]]