add check to files

lidofinance · Apr 25, 2024 · feeb6d9 · feeb6d9
1 parent 46b367c
commit feeb6d9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/bots/pauser.py b/src/bots/pauser.py
@@ -123,6 +123,9 @@ def _send_pause(self, message: PauseMessage):
             (message['signature']['r'], compute_vs(message['signature']['v'], message['signature']['s']))
         )
 
+        if not self.w3.transaction.check(pause_tx):
+            return False
+
         result = self.w3.transaction.send(pause_tx, False, 6)
         logger.info({'msg': f'Transaction send. Result is {result}.', 'value': result})
         return result

diff --git a/src/bots/unvetter.py b/src/bots/unvetter.py
@@ -129,6 +129,9 @@ def _send_unvet_message(self, message: UnvetMessage) -> bool:
             (message['signature']['r'], compute_vs(message['signature']['v'], message['signature']['s']))
         )
 
+        if not self.w3.transaction.check(unvet_tx):
+            return False
+
         result = self.w3.transaction.send(unvet_tx, False, 6)
         logger.info({'msg': f'Transaction send. Result is {result}.', 'value': result})
         return result