cleanup code

Author: jack60612

tools/generate_chain.py

@@ -151,6 +151,10 @@ async def main(length: int, fill_rate: int, profile: bool, block_refs: bool, out
                 prev_tx_hash: Optional[bytes32] = None
                 for bl in blocks:
                     if bl.is_transaction_block():
+                        reward_coins = bl.get_included_reward_coins()
+                        for coin in reward_coins:
+                            if coin.puzzle_hash in {farmer_puzzlehash, pool_puzzlehash}:
+                                unspent_coins.append(coin)
                         # Update coin store and mempool with new block
                         br = BenchBlockRecord(
                             header_hash=bl.header_hash,
@@ -159,16 +163,13 @@ async def main(length: int, fill_rate: int, profile: bool, block_refs: bool, out
                             prev_transaction_block_height=prev_tx_height,
                             prev_transaction_block_hash=prev_tx_hash,
                         )
-                        await coin_store.new_block(bl.height, block_timestamp, bl.get_included_reward_coins(), [], [])
+                        await coin_store.new_block(bl.height, block_timestamp, reward_coins, [], [])
                         await mempool_manager.new_peak(br, [])
                         prev_tx_height = bl.height
                         prev_tx_hash = bl.header_hash
                         prev_tx_block = bl  # last transaction block
 
                     block_timestamp = uint64(block_timestamp + int(time_per_block))
-                    for coin in bl.get_included_reward_coins():
-                        if coin.puzzle_hash in {farmer_puzzlehash, pool_puzzlehash}:
-                            unspent_coins.append(coin)
                     db.execute(
                         "INSERT INTO full_blocks VALUES(?, ?, ?, ?, ?)",
                         (