nado-protocol 0.2.7__tar.gz → 0.2.8__tar.gz

{nado_protocol-0.2.7 → nado_protocol-0.2.8}/PKG-INFO

@@ -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 → nado_protocol-0.2.8}/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-0.2.7 → nado_protocol-0.2.8}/nado_protocol/engine_client/types/__init__.py

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

{nado_protocol-0.2.7 → nado_protocol-0.2.8}/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 → nado_protocol-0.2.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nado-protocol"
-version = "0.2.7"
+version = "0.2.8"
 description = "Nado Protocol SDK"
 authors = ["Jeury Mejia <jeury@inkfnd.com>"]
 homepage = "https://nado.xyz"