feat: Add auction data handling to stock historical data client

alpaca/common/rest.py

@@ -425,6 +425,7 @@ def _get_marketdata_entries(response: HTTPResult, no_sub_key: bool) -> RawData:
         "snapshots",
         "trade",
         "trades",
+        "auctions",
     }
     selected_keys = data_keys.intersection(response)
     # Neither of these should ever happen!

alpaca/common/utils.py

@@ -33,7 +33,7 @@ def validate_uuid_id_param(
 
 
 def validate_symbol_or_asset_id(
-    symbol_or_asset_id: Union[UUID, str]
+    symbol_or_asset_id: Union[UUID, str],
 ) -> Union[UUID, str]:
     """
     A helper function to eliminate duplicate checks of symbols or asset ids.
@@ -55,7 +55,7 @@ def validate_symbol_or_asset_id(
 
 
 def validate_symbol_or_contract_id(
-    symbol_or_contract_id: Union[UUID, str]
+    symbol_or_contract_id: Union[UUID, str],
 ) -> Union[UUID, str]:
     """
     A helper function to eliminate duplicate checks of symbols or contract id.

alpaca/data/historical/stock.py

@@ -10,7 +10,7 @@
 from alpaca.data.historical.utils import (
     parse_obj_as_symbol_dict,
 )
-from alpaca.data.models import BarSet, QuoteSet, TradeSet
+from alpaca.data.models import BarSet, QuoteSet, TradeSet, AuctionSet
 from alpaca.data.requests import (
     StockBarsRequest,
     StockLatestBarRequest,
@@ -19,6 +19,7 @@
     StockQuotesRequest,
     StockSnapshotRequest,
     StockTradesRequest,
+    StockAuctionsRequest,
 )
 
 
@@ -72,6 +73,25 @@ def __init__(
             raw_data=raw_data,
         )
 
+    def get_stock_auction(self, request_params: StockAuctionsRequest):
+        """Returns auction data for an equity or list of equities over a given time period.
+
+        Args:
+            request_params (StockAuctionsRequest): The request object for retrieving stock auction data.
+
+        Returns:
+            Union[AuctionSet, RawData]: The auction data either in raw or wrapped form
+        """
+        raw_auctions = self._get_marketdata(
+            path="/stocks/auctions",
+            params=request_params.to_request_fields(),
+        )
+
+        if self._use_raw_data:
+            return raw_auctions
+
+        return AuctionSet(raw_auctions)
+
     def get_stock_bars(
         self, request_params: StockBarsRequest
     ) -> Union[BarSet, RawData]:

alpaca/data/mappings.py

@@ -82,3 +82,11 @@
     "cc": "corrected_conditions",
     "z": "tape",
 }
+
+AUCTION_MAPPING: Dict[str, str] = {
+    "c": "condition",
+    "p": "price",
+    "s": "size",
+    "t": "timestamp",
+    "x": "exchange",
+}

alpaca/data/models/__init__.py

@@ -5,3 +5,4 @@
 from alpaca.data.models.trades import *
 from alpaca.data.models.snapshots import *
 from alpaca.data.models.orderbooks import *
+from alpaca.data.models.auctions import *

alpaca/data/models/auctions.py

@@ -0,0 +1,82 @@
+from datetime import datetime
+from typing import Dict, List, Optional
+
+from alpaca.common.models import ValidateBaseModel as BaseModel
+from alpaca.common.types import RawData
+from alpaca.data.mappings import AUCTION_MAPPING
+from alpaca.data.models.base import BaseDataSet, TimeSeriesMixin
+
+
+class Auction(BaseModel):
+    """Represents one auction of aggregated trade data over a specified interval.
+
+    Attributes:
+        symbol (str): The ticker identifier for the security whose data forms the bar.
+
+    """
+
+    symbol: str
+    timestamp: datetime
+    condition: str
+    price: float
+    size: float
+    exchange: str
+
+    def __init__(self, symbol: str, raw_data: RawData) -> None:
+        """Instantiates an auction
+
+        Args:
+            raw_data (RawData): Raw unparsed auction data from API, contains ohlc and other fields.
+        """
+        mapped_auction = {}
+
+        if raw_data is not None:
+            mapped_auction = {
+                AUCTION_MAPPING[key]: val
+                for key, val in raw_data.items()
+                if key in AUCTION_MAPPING
+            }
+
+        super().__init__(symbol=symbol, **mapped_auction)
+
+
+class AuctionSet(BaseDataSet, TimeSeriesMixin):
+    """A collection of Auctions.
+
+    Attributes:
+        data (Dict[str, List[Auction]]): The collection of Auctions keyed by symbol.
+    """
+
+    data: Dict[str, List[Auction]] = {}
+
+    def __init__(self, raw_data: RawData) -> None:
+        """A collection of Auctions.
+
+        Args:
+            raw_data (RawData): The collection of raw auction data from API keyed by Symbol.
+        """
+
+        parsed_auctions = {}
+
+        raw_auctions = raw_data
+
+        if raw_auctions is not None:
+            for symbol, auctions in raw_auctions.items():
+
+                auction_data = []
+                for auction in auctions:
+                    c = auction.get("c")
+                    o = auction.get("o")
+
+                    if c is not None:
+                        auction_data.extend(c)
+                    if o is not None:
+                        auction_data.extend(o)
+
+                parsed_auctions[symbol] = [
+                    Auction(symbol, auction)
+                    for auction in auction_data
+                    if auction is not None
+                ]
+
+        super().__init__(data=parsed_auctions)

alpaca/data/requests.py

@@ -60,6 +60,14 @@ def __init__(self, **data: Any) -> None:
         super().__init__(**data)
 
 
+# ############################## Auctions ################################# #
+
+
+class StockAuctionsRequest(BaseTimeseriesDataRequest):
+    feed: Optional[DataFeed] = None
+    asof: Optional[str] = None
+
+
 # ############################## Bars ################################# #