nado-protocol 0.2.7__py3-none-any.whl → 0.2.8__py3-none-any.whl

nado_protocol/engine_client/query.py

@@ -1,3 +1,4 @@
+import json
 from typing import Optional
 import requests
 
@@ -182,7 +183,10 @@ class EngineQueryClient:
         )
 
     def get_subaccount_info(
-        self, subaccount: str, txs: Optional[list[QuerySubaccountInfoTx]] = None
+        self,
+        subaccount: str,
+        txs: Optional[list[QuerySubaccountInfoTx]] = None,
+        pre_state: Optional[bool] = None,
     ) -> SubaccountInfoData:
         """
         Query the engine for the state of a subaccount, including balances.
@@ -193,11 +197,26 @@ class EngineQueryClient:
             txs (list[QuerySubaccountInfoTx], optional): You can optionally provide a list of txs, to get an estimated view
             of what the subaccount state would look like if the transactions were applied.
 
+            pre_state (bool, optional): When True and txs are provided, returns the subaccount state before the
+            transactions were applied in the pre_state field. Defaults to False.
+
         Returns:
             SubaccountInfoData: Information about the specified subaccount.
         """
+        txns_str = None
+        if txs is not None:
+            txns_str = json.dumps([tx.dict() for tx in txs])
+
+        pre_state_str = None
+        if pre_state is not None:
+            pre_state_str = str(pre_state).lower()
+
         return ensure_data_type(
-            self.query(QuerySubaccountInfoParams(subaccount=subaccount, txs=txs)).data,
+            self.query(
+                QuerySubaccountInfoParams(
+                    subaccount=subaccount, txns=txns_str, pre_state=pre_state_str
+                )
+            ).data,
             SubaccountInfoData,
         )
 

nado_protocol/engine_client/types/__init__.py

@@ -60,6 +60,7 @@ __all__ = [
     "ContractsData",
     "NoncesData",
     "OrderData",
+    "PreState",
     "SubaccountInfoData",
     "SubaccountOpenOrdersData",
     "MarketLiquidityData",

nado_protocol/engine_client/types/query.py

@@ -101,7 +101,8 @@ class QuerySubaccountInfoParams(NadoBaseModel):
 
     type = EngineQueryType.SUBACCOUNT_INFO.value
     subaccount: str
-    txs: Optional[list[QuerySubaccountInfoTx]]
+    txns: Optional[str]
+    pre_state: Optional[str]
 
 
 class QuerySubaccountOpenOrdersParams(NadoBaseModel):
@@ -299,6 +300,17 @@ class OrderData(NadoBaseModel):
     placed_at: str
 
 
+class PreState(NadoBaseModel):
+    """
+    Model for subaccount state before simulated transactions were applied.
+    """
+
+    healths: list[SubaccountHealth]
+    health_contributions: list[list[str]]
+    spot_balances: list[SpotProductBalance]
+    perp_balances: list[PerpProductBalance]
+
+
 class SubaccountInfoData(NadoBaseModel):
     """
     Model for detailed info about a subaccount, including balances.
@@ -314,6 +326,7 @@ class SubaccountInfoData(NadoBaseModel):
     perp_balances: list[PerpProductBalance]
     spot_products: list[SpotProduct]
     perp_products: list[PerpProduct]
+    pre_state: Optional[PreState]
 
     def parse_subaccount_balance(
         self, product_id: int

{nado_protocol-0.2.7.dist-info → nado_protocol-0.2.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nado-protocol
-Version: 0.2.7
+Version: 0.2.8
 Summary: Nado Protocol SDK
 Keywords: nado protocol,nado sdk,nado protocol api
 Author: Jeury Mejia

{nado_protocol-0.2.7.dist-info → nado_protocol-0.2.8.dist-info}/RECORD

@@ -41,11 +41,11 @@ nado_protocol/contracts/loader.py,sha256=_0W3ImcARs1bTu1lLMpHMrRNuEVYjrRWctvnEbK
 nado_protocol/contracts/types.py,sha256=km29Md8hCE8yivGdgbgMuQoO7GyE0yOVt9BaJatvjFY,4280
 nado_protocol/engine_client/__init__.py,sha256=RE_MxQaQVIp0K5EDhR1_uTjLJgEJBx2fLnhepc3cuAg,1137
 nado_protocol/engine_client/execute.py,sha256=Tmd1yvLTlvFFaSYqwOWTkzSxdE6QllGB8LHgsOim3Tc,15314
-nado_protocol/engine_client/query.py,sha256=zCBLeBOY1wkRM42EMsluBRtVw2TuaqJHp4UuIdHWh4c,16265
-nado_protocol/engine_client/types/__init__.py,sha256=pPjKLDJHclcbqROhU8B8OoxUjZkJI9a2qq5fUCVK7j4,2721
+nado_protocol/engine_client/query.py,sha256=v7PqRsnX5jZskNnBpv38HB7BQ7mi8RBhUgczf5msEIc,16858
+nado_protocol/engine_client/types/__init__.py,sha256=aCw_aNxCRy5PhrYpvUHuB7gQPcxHld9iMV5eTFWdEfw,2737
 nado_protocol/engine_client/types/execute.py,sha256=ANF_QwtJAqb9zvga1fjOIiSDt_wjUo4sbf8rVkc-sQ0,20709
 nado_protocol/engine_client/types/models.py,sha256=JIDyOLUBprGcMKHxW2M53_nCc7hoerglnFvfn7wgdXo,3862
-nado_protocol/engine_client/types/query.py,sha256=N6kPuWSUowMeZnUKpxF9NnMKZBKKe-0Q_B9ccFZkjQA,12185
+nado_protocol/engine_client/types/query.py,sha256=5j4KcYUwGAoJFF0m2bmrqYXbU4iJ5IyzUTiEIzn6rh0,12516
 nado_protocol/engine_client/types/stream.py,sha256=F_AKqU4pClzUcM6D9Dc79f6RmraahJzj2-hQSXtQ0vQ,159
 nado_protocol/indexer_client/__init__.py,sha256=ea2exePtguCxJsBlisOQPtnFk8hRmFugeQAuSAaPhoc,922
 nado_protocol/indexer_client/query.py,sha256=aMJoJfKXd9eE4j5me5FPOtsrHY8b5nXCpCV_NCZzh2E,13710
@@ -76,7 +76,7 @@ nado_protocol/utils/order.py,sha256=Q9TlcotvnB395dPhaKpn0EeN1WNTkpYBTUovlirr1_Y,
 nado_protocol/utils/subaccount.py,sha256=WJ7lgU2RekuzJAZH-hhCTbIBlRsl2oHozBm7OEMRV74,495
 nado_protocol/utils/time.py,sha256=tEwmrkc5VdzKLlgkJIAq2ce-nhrduJZNtVPydrrnTHs,360
 nado_protocol/utils/twap.py,sha256=hfBVK0CBa8m4uBArxTnNRoJr3o1rJucyopR_8_9gkOo,6197
-nado_protocol-0.2.7.dist-info/METADATA,sha256=Bo4AhqpGlOdtJAw4-ehu_9CrTvv2pRguokrTUsQ6qhQ,9558
-nado_protocol-0.2.7.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-nado_protocol-0.2.7.dist-info/entry_points.txt,sha256=Df0O9lFc-m0SyOh6_d9FHeG1OT-esxGm-p_z7rTT9h0,340
-nado_protocol-0.2.7.dist-info/RECORD,,
+nado_protocol-0.2.8.dist-info/METADATA,sha256=W26ydlZNiR7NihwnL_gzayKfPKqSWetgcxusYz3ioic,9558
+nado_protocol-0.2.8.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+nado_protocol-0.2.8.dist-info/entry_points.txt,sha256=Df0O9lFc-m0SyOh6_d9FHeG1OT-esxGm-p_z7rTT9h0,340
+nado_protocol-0.2.8.dist-info/RECORD,,