Add Price dataclass to types. Add StartSession() to subscription module. Rewrite SubscribeToNode() in node module

Author: freQniK

src/sentinel_sdk/modules/node.py

@@ -14,10 +14,11 @@
 import sentinel_protobuf.sentinel.node.v2.querier_pb2 as sentinel_node_v2_querier_pb2
 import sentinel_protobuf.sentinel.node.v2.querier_pb2_grpc as sentinel_node_v2_querier_pb2_grpc
 import sentinel_protobuf.sentinel.node.v2.msg_pb2 as msg_pb2
+import sentinel_protobuf.sentinel.node.v3.msg_pb2 as msg_pb2_3
 
 from sentinel_sdk.querier.querier import Querier
 from sentinel_sdk.transactor.transactor import Transactor
-from sentinel_sdk.types import PageRequest, TxParams, NodeType
+from sentinel_sdk.types import PageRequest, TxParams, NodeType, Price
 
 from .wireguard import WgKey
 
@@ -129,6 +130,7 @@ def RegisterNode(self, gigabyte_prices: int, hourly_prices: int, remote_url: str
         )
         return self.transaction([msg], tx_params)
 
+    '''
     def SubscribeToNode(self, node_address: str, gigabytes: int = 0, hours: int = 0, denom: str = "udvpn", tx_params: TxParams = TxParams()):
         msg = msg_pb2.MsgSubscribeRequest(
             frm = self._account.address,
@@ -138,7 +140,17 @@ def SubscribeToNode(self, node_address: str, gigabytes: int = 0, hours: int = 0,
             node_address = node_address,
         )
         return self.transaction([msg], tx_params)
-
+    '''
+    
+    def SubscribeToNode(self, node_address: str, gigabytes: int = 0, hours: int = 0, price: Price = Price(), tx_params: TxParams = TxParams()):
+        msg = msg_pb2_3.MsgStartSessionRequest(
+            frm = self._account.address,
+            gigabytes = gigabytes,
+            hours = hours,
+            node_address = node_address,
+            max_price = price,
+        )
+        return self.transaction([msg], tx_params)
     def UpdateNodeDetails(self, gigabyte_prices: int, hourly_prices: int, remote_url: str, tx_params: TxParams = TxParams()):
         msg = msg_pb2.MsgUpdateDetailsRequest(
             frm = self._account.address,

src/sentinel_sdk/modules/subscription.py

@@ -5,6 +5,7 @@
 import sentinel_protobuf.sentinel.subscription.v2.querier_pb2_grpc as sentinel_subscription_v2_querier_pb2_grpc
 import sentinel_protobuf.sentinel.subscription.v2.subscription_pb2 as subscription_pb2
 import sentinel_protobuf.sentinel.subscription.v2.msg_pb2 as msg_pb2
+import sentinel_protobuf.sentinel.subscription.v3.msg_pb2 as msg_pb2_3
 
 from sentinel_sdk.querier.querier import Querier
 from sentinel_sdk.transactor.transactor import Transactor
@@ -168,6 +169,13 @@ def Cancel(self, id: int, tx_params: TxParams = TxParams()):
         )
         return self.transaction([msg], tx_params)
 
+    def StartSession(self, address: str, subscription_id: int, tx_params: TxParams = TxParams()):
+        msg = msg_pb2_3.MsgStartSessionRequest(
+            frm = self._account.address,
+            id = subscription_id,
+            address = address
+        )
+        return self.transaction([msg], tx_params)
     # Node subscriptions are returned by grpc querier in google's 'Any' type and need to be converted into desired protobuf type
     #
     #

src/sentinel_sdk/types.py

@@ -25,6 +25,12 @@ class TxParams:
     fee_amount: int = 31415
     gas: float = 0
     gas_multiplier: float = 1.5
+    
+@dataclass
+class Price:
+    denom: str = "udvpn"
+    base_value: str = "0"
+    quote_value: str  = "0"
 
 
 class PageRequest: