@@ -197,45 +197,45 @@ def on_init(environment: Environment, **_kwargs):
197197 test_data ["chain_id" ] = {test_data_class_name : chain_id }
198198 if environment .parsed_options :
199199 user_test_data .init_data (environment .parsed_options )
200- test_data [test_data_class_name ] = user_test_data .data .to_json ()
201- send_msg_to_workers (environment .runner , "test_data" , test_data )
202- print ("Fetching blocks..." )
203- if environment .parsed_options .use_latest_blocks :
204- print (f"Using latest { user_test_data .data .size .blocks_len } blocks as test data" )
205- logger .info (f"Using latest { user_test_data .data .size .blocks_len } blocks as test data" )
206- for block_number in range (
207- user_test_data .data .block_range .start , user_test_data .data .block_range .end + 1
208- ):
209- block = None
210- try :
211- block = user_test_data .fetch_block (block_number )
212- except (BlockNotFoundError , InvalidBlockError ):
213- pass
214- while block is None :
200+ test_data [test_data_class_name ] = user_test_data .data .to_json ()
201+ send_msg_to_workers (environment .runner , "test_data" , test_data )
202+ print ("Fetching blocks..." )
203+ if environment .parsed_options .use_latest_blocks :
204+ print (f"Using latest { user_test_data .data .size .blocks_len } blocks as test data" )
205+ logger .info (f"Using latest { user_test_data .data .size .blocks_len } blocks as test data" )
206+ for block_number in range (
207+ user_test_data .data .block_range .start , user_test_data .data .block_range .end + 1
208+ ):
209+ block = None
215210 try :
216- block = user_test_data .fetch_latest_block ( )
211+ block = user_test_data .fetch_block ( block_number )
217212 except (BlockNotFoundError , InvalidBlockError ):
218213 pass
219- user_test_data .data .push_block (block )
220- block_data = {test_data_class_name : block .to_json ()}
221- send_msg_to_workers (environment .runner , "block_data" , block_data )
222- print (user_test_data .data .stats (), end = "\r " )
214+ while block is None :
215+ try :
216+ block = user_test_data .fetch_latest_block ()
217+ except (BlockNotFoundError , InvalidBlockError ):
218+ pass
219+ user_test_data .data .push_block (block )
220+ block_data = {test_data_class_name : block .to_json ()}
221+ send_msg_to_workers (environment .runner , "block_data" , block_data )
222+ print (user_test_data .data .stats (), end = "\r " )
223+ else :
224+ print (user_test_data .data .stats (), end = "\r " )
225+ print ("\n " ) # new line after progress display upon exiting loop
223226 else :
224- print (user_test_data .data .stats (), end = "\r " )
225- print ("\n " ) # new line after progress display upon exiting loop
226- else :
227- while user_test_data .data .size .blocks_len > len (user_test_data .data .blocks ):
228- try :
229- block = user_test_data .fetch_random_block (user_test_data .data .block_numbers )
230- except (BlockNotFoundError , InvalidBlockError ):
231- continue
232- user_test_data .data .push_block (block )
233- block_data = {test_data_class_name : block .to_json ()}
234- send_msg_to_workers (environment .runner , "block_data" , block_data )
235- print (user_test_data .data .stats (), end = "\r " )
236- else :
237- print (user_test_data .data .stats (), end = "\r " )
238- print ("\n " ) # new line after progress display upon exiting loop
227+ while user_test_data .data .size .blocks_len > len (user_test_data .data .blocks ):
228+ try :
229+ block = user_test_data .fetch_random_block (user_test_data .data .block_numbers )
230+ except (BlockNotFoundError , InvalidBlockError ):
231+ continue
232+ user_test_data .data .push_block (block )
233+ block_data = {test_data_class_name : block .to_json ()}
234+ send_msg_to_workers (environment .runner , "block_data" , block_data )
235+ print (user_test_data .data .stats (), end = "\r " )
236+ else :
237+ print (user_test_data .data .stats (), end = "\r " )
238+ print ("\n " ) # new line after progress display upon exiting loop
239239 logger .info ("Test data is ready" )
240240 send_msg_to_workers (environment .runner , "release_lock" , {})
241241 user_test_data .release_lock ()
0 commit comments