indices snapshot addition

Author: antdjohns

polygon/rest/models/snapshot.py

@@ -15,7 +15,6 @@ class MinuteSnapshot:
     close: Optional[float] = None
     volume: Optional[float] = None
     vwap: Optional[float] = None
-    value: Optional[float] = None
     otc: Optional[bool] = None
 
     @staticmethod
@@ -28,7 +27,6 @@ def from_dict(d):
             d.get("c", None),
             d.get("v", None),
             d.get("vw", None),
-            d.get("val", None),
             d.get("otc", None),
         )
 
@@ -53,13 +51,34 @@ def from_dict(d):
 
 
 @modelclass
-class TickerSnapshot:
-    "Contains the most up-to-date market data for all traded ticker symbols."
+class IndicesSnapshot:
     value: Optional[float] = None
     name: Optional[str] = None
     type: Optional[str] = None
+    ticket: Optional[str] = None
     market_status: Optional[str] = None
     session: Optional[DayOptionContractSnapshot] = None
+    error: Optional[str] = None
+    message: Optional[str] = None
+
+    @staticmethod
+    def from_dict(d):
+        return IndicesSnapshot(
+            value=d.get("value", None),
+            name=d.get("name", None),
+            type=d.get("type", None),
+            market_status=d.get("market_status", None),
+            session=None
+            if "session" not in d
+            else DayOptionContractSnapshot.from_dict(d["session"]),
+            error=d.get("error", None),
+            message=d.get("message", None),
+        )
+
+
+@modelclass
+class TickerSnapshot:
+    "Contains the most up-to-date market data for all traded ticker symbols."
     day: Optional[Agg] = None
     last_quote: Optional[LastQuote] = None
     last_trade: Optional[LastTrade] = None
@@ -73,13 +92,6 @@ class TickerSnapshot:
     @staticmethod
     def from_dict(d):
         return TickerSnapshot(
-            value=d.get("value", None),
-            name=d.get("name", None),
-            type=d.get("type", None),
-            market_status=d.get("market_status", None),
-            session=None
-            if "session" not in d
-            else DayOptionContractSnapshot.from_dict(d["session"]),
             day=None if "day" not in d else Agg.from_dict(d["day"]),
             last_quote=None
             if "lastQuote" not in d

polygon/rest/snapshot.py

@@ -6,6 +6,7 @@
     OptionContractSnapshot,
     SnapshotMarketType,
     SnapshotTickerFullBook,
+    IndicesSnapshot,
 )
 from urllib3 import HTTPResponse
 
@@ -184,3 +185,22 @@ def get_snapshot_crypto_book(
             raw=raw,
             options=options,
         )
+
+    def get_snapshot_indices(
+        self,
+        ticker: Optional[str] = None,
+        ticker_any_of: Optional[Union[str, List[str]]] = None,
+        params: Optional[Dict[str, Any]] = None,
+        raw: bool = False,
+        options: Optional[RequestOptionBuilder] = None,
+    ) -> Union[List[IndicesSnapshot], HTTPResponse]:
+        url = f"/v3/snapshot/indices"
+
+        return self._get(
+            path=url,
+            params=self._get_params(self.get_snapshot_indices, locals()),
+            deserializer=IndicesSnapshot.from_dict,
+            raw=raw,
+            result_key="results",
+            options=options,
+        )