2121
2222import json
2323import re
24- from datetime import datetime
24+ from datetime import datetime , timezone
2525from enum import Enum
2626from typing import Any , Callable , Dict , Optional , Tuple , cast
2727from urllib .parse import urlparse
3333)
3434from packages .valory .protocols .http .message import HttpMessage
3535from packages .valory .protocols .ipfs import IpfsMessage
36+ from packages .valory .skills .abstract_round_abci .base import RoundSequence
3637from packages .valory .skills .abstract_round_abci .handlers import (
3738 ABCIRoundHandler as BaseABCIRoundHandler ,
3839)
7071TendermintHandler = BaseTendermintHandler
7172
7273
74+ FSM_REPR_MAX_DEPTH = 25
75+
76+
7377class IpfsHandler (AbstractResponseHandler ):
7478 """IPFS message handler."""
7579
@@ -104,6 +108,7 @@ def handle(self, message: IpfsMessage) -> None:
104108OK_CODE = 200
105109NOT_FOUND_CODE = 404
106110BAD_REQUEST_CODE = 400
111+ TOO_EARLY_CODE = 425
107112AVERAGE_PERIOD_SECONDS = 10
108113
109114
@@ -156,12 +161,15 @@ def setup(self) -> None:
156161
157162 self .rounds_info = load_rounds_info_with_transitions ()
158163
164+ @property
165+ def round_sequence (self ) -> RoundSequence :
166+ """Return the round sequence."""
167+ return self .context .state .round_sequence
168+
159169 @property
160170 def synchronized_data (self ) -> SynchronizedData :
161171 """Return the synchronized data."""
162- return SynchronizedData (
163- db = self .context .state .round_sequence .latest_synchronized_data .db
164- )
172+ return SynchronizedData (db = self .round_sequence .latest_synchronized_data .db )
165173
166174 def _get_handler (self , url : str , method : str ) -> Tuple [Optional [Callable ], Dict ]:
167175 """Check if an url is meant to be handled in this handler
@@ -176,7 +184,7 @@ def _get_handler(self, url: str, method: str) -> Tuple[Optional[Callable], Dict]
176184 # Check base url
177185 if not re .match (self .handler_url_regex , url ):
178186 self .context .logger .info (
179- f"The url { url } DynamicNFT HttpHandler's pattern"
187+ f"The url { url }
180188 )
181189 return None , {}
182190
@@ -193,7 +201,7 @@ def _get_handler(self, url: str, method: str) -> Tuple[Optional[Callable], Dict]
193201
194202 # No route found
195203 self .context .logger .info (
196- f"The message [{ method } { url } DynamicNFT HttpHandler but did not match any valid pattern"
204+ f"The message [{ method } { url }
197205 )
198206 return self ._handle_bad_request , {}
199207
@@ -265,6 +273,36 @@ def _handle_bad_request(
265273 self .context .logger .info ("Responding with: {}" .format (http_response ))
266274 self .context .outbox .put_message (message = http_response )
267275
276+ def _has_transitioned (self ) -> bool :
277+ """Check if the agent has transitioned."""
278+ try :
279+ return bool (self .round_sequence .last_round_transition_height )
280+ except ValueError :
281+ return False
282+
283+ def _handle_too_early (
284+ self , http_msg : HttpMessage , http_dialogue : HttpDialogue
285+ ) -> None :
286+ """
287+ Handle a request when the FSM's loop has not started yet.
288+
289+ :param http_msg: the http message
290+ :param http_dialogue: the http dialogue
291+ """
292+ http_response = http_dialogue .reply (
293+ performative = HttpMessage .Performative .RESPONSE ,
294+ target_message = http_msg ,
295+ version = http_msg .version ,
296+ status_code = TOO_EARLY_CODE ,
297+ status_text = "The state machine has not started yet! Please try again later..." ,
298+ headers = http_msg .headers ,
299+ body = b"" ,
300+ )
301+
302+ # Send response
303+ self .context .logger .info ("Responding with: {}" .format (http_response ))
304+ self .context .outbox .put_message (message = http_response )
305+
268306 def _handle_get_health (
269307 self , http_msg : HttpMessage , http_dialogue : HttpDialogue
270308 ) -> None :
@@ -274,40 +312,41 @@ def _handle_get_health(
274312 :param http_msg: the http message
275313 :param http_dialogue: the http dialogue
276314 """
277- seconds_since_last_transition = None
278- is_tm_unhealthy = None
279- is_transitioning_fast = None
280- current_round = None
281- rounds = None
315+ if not self ._has_transitioned ():
316+ self ._handle_too_early (http_msg , http_dialogue )
317+ return
318+
282319 has_required_funds = self ._check_required_funds ()
283320 is_receiving_mech_responses = self ._check_is_receiving_mech_responses ()
284321 is_staking_kpi_met = self .synchronized_data .is_staking_kpi_met
285322 staking_status = self .synchronized_data .service_staking_state .name .lower ()
286323
287- round_sequence = cast (SharedState , self .context .state ).round_sequence
324+ round_sequence = self .round_sequence
325+ is_tm_unhealthy = round_sequence .block_stall_deadline_expired
288326
289- if round_sequence ._last_round_transition_timestamp :
290- is_tm_unhealthy = cast (
291- SharedState , self .context .state
292- ).round_sequence .block_stall_deadline_expired
293-
294- current_time = datetime .now ().timestamp ()
295- seconds_since_last_transition = current_time - datetime .timestamp (
296- round_sequence ._last_round_transition_timestamp
297- )
327+ current_time = datetime .now ().timestamp ()
328+ seconds_since_last_transition = current_time - datetime .timestamp (
329+ round_sequence .last_round_transition_timestamp
330+ )
298331
299- is_transitioning_fast = (
300- not is_tm_unhealthy
301- and seconds_since_last_transition
302- < 2 * self .context .params .reset_pause_duration
303- )
332+ abci_app = self .round_sequence .abci_app
333+ previous_rounds = abci_app ._previous_rounds
334+ previous_round_cls = type (previous_rounds [- 1 ])
335+ previous_round_events = abci_app .transition_function .get (
336+ previous_round_cls , {}
337+ ).keys ()
338+ previous_round_timeouts = {
339+ abci_app .event_to_timeout .get (event , - 1 ) for event in previous_round_events
340+ }
341+ last_round_timeout = max (previous_round_timeouts )
342+ is_transitioning_fast = (
343+ not is_tm_unhealthy
344+ and seconds_since_last_transition < 2 * last_round_timeout
345+ )
304346
305- if round_sequence ._abci_app :
306- current_round = round_sequence ._abci_app .current_round .round_id
307- rounds = [
308- r .round_id for r in round_sequence ._abci_app ._previous_rounds [- 25 :]
309- ]
310- rounds .append (current_round )
347+ rounds = [r .round_id for r in previous_rounds [- FSM_REPR_MAX_DEPTH :]] + [
348+ round_sequence .current_round_id
349+ ]
311350
312351 data = {
313352 "seconds_since_last_transition" : seconds_since_last_transition ,
@@ -348,7 +387,7 @@ def _send_ok_response(
348387 def _send_not_found_response (
349388 self , http_msg : HttpMessage , http_dialogue : HttpDialogue
350389 ) -> None :
351- """Send an not found response"""
390+ """Send a not found response"""
352391 http_response = http_dialogue .reply (
353392 performative = HttpMessage .Performative .RESPONSE ,
354393 target_message = http_msg ,
@@ -375,6 +414,6 @@ def _check_is_receiving_mech_responses(self) -> bool:
375414 # (an on chain transaction)
376415 return (
377416 self .synchronized_data .decision_receive_timestamp
378- < int (datetime .utcnow ( ).timestamp ())
417+ < int (datetime .now ( timezone . utc ).timestamp ())
379418 - self .context .params .expected_mech_response_time
380419 )
0 commit comments