olas-operate-middleware 0.10.10__tar.gz → 0.10.11__tar.gz

{olas_operate_middleware-0.10.10 → olas_operate_middleware-0.10.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: olas-operate-middleware
-Version: 0.10.10
+Version: 0.10.11
 Summary: 
 Author: David Vilela
 Author-email: dvilelaf@gmail.com

{olas_operate_middleware-0.10.10 → olas_operate_middleware-0.10.11}/operate/services/manage.py

@@ -1895,21 +1895,23 @@ class ServiceManager:
         service = self.load(service_config_id=service_config_id)
         chain_config = service.chain_configs[chain]
         ledger_config = chain_config.ledger_config
-        staking_program_id = chain_config.chain_data.user_params.staking_program_id
         wallet = self.wallet_manager.load(ledger_config.chain.ledger_type)
         ledger_api = wallet.ledger_api(chain=ledger_config.chain, rpc=ledger_config.rpc)
         self.logger.info(
             f"OLAS Balance on service Safe {chain_config.chain_data.multisig}: "
             f"{get_asset_balance(ledger_api, OLAS[Chain(chain)], chain_config.chain_data.multisig)}"
         )
+        current_staking_program = self._get_current_staking_program(
+            service=service, chain=chain
+        )
         staking_contract = get_staking_contract(
             chain=ledger_config.chain,
-            staking_program_id=staking_program_id,
+            staking_program_id=current_staking_program,
         )
         if staking_contract is None:
             raise RuntimeError(
-                "No staking contract found for the current staking_program_id: "
-                f"{staking_program_id}. Not claiming the rewards."
+                "No staking contract found for the "
+                f"{current_staking_program=}. Not claiming the rewards."
             )
 
         sftxb = self.get_eth_safe_tx_builder(ledger_config=ledger_config)
@@ -1925,15 +1927,18 @@ class ServiceManager:
             .add(
                 sftxb.get_claiming_data(
                     service_id=chain_config.chain_data.token,
-                    staking_contract=get_staking_contract(
-                        chain=ledger_config.chain,
-                        staking_program_id=staking_program_id,
-                    ),
+                    staking_contract=staking_contract,
                 )
             )
             .settle()
         )
 
+        if receipt.status != 1:
+            self.logger.error(
+                f"Failed to claim staking rewards. Tx hash: {receipt.tx_hash}"
+            )
+            return 0
+
         # transfer claimed amount from agents safe to master safe
         # TODO: remove after staking contract directly starts sending the rewards to master safe
         amount_claimed = int(receipt["logs"][0]["data"].hex(), 16)
@@ -2323,8 +2328,8 @@ class ServiceManager:
                         await loop.run_in_executor(
                             executor,
                             self.claim_on_chain_from_safe,
-                            service_config_id=service_config_id,
-                            chain=service.home_chain,
+                            service_config_id,
+                            service.home_chain,
                         )
                     except Exception:  # pylint: disable=broad-except
                         logging.info(

{olas_operate_middleware-0.10.10 → olas_operate_middleware-0.10.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "olas-operate-middleware"
-version = "0.10.10"
+version = "0.10.11"
 description = ""
 authors = ["David Vilela <dvilelaf@gmail.com>", "Viraj Patel <vptl185@gmail.com>"]
 readme = "README.md"