Break into two functions

mev_inspect/sandwiches.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Optional
 
 from mev_inspect.schemas.sandwiches import Sandwich
 from mev_inspect.schemas.swaps import Swap
@@ -14,10 +14,22 @@ def get_sandwiches(swaps: List[Swap]) -> List[Sandwich]:
 
     sandwiches: List[Sandwich] = []
 
-    for index, front_swap in enumerate(ordered_swaps):
-        sandwicher_address = front_swap.from_address
-        rest_swaps = ordered_swaps[index:]
+    for index, swap in enumerate(ordered_swaps):
+        rest_swaps = ordered_swaps[index + 1 :]
+        sandwich = _get_sandwich_starting_with_swap(swap, rest_swaps)
 
+        if sandwich is not None:
+            sandwiches.append(sandwich)
+
+    return sandwiches
+
+
+def _get_sandwich_starting_with_swap(
+    front_swap: Swap,
+    rest_swaps: List[Swap],
+) -> Optional[Sandwich]:
+
+    sandwicher_address = front_swap.from_address
     sandwiched_swaps = []
 
     for other_swap in rest_swaps:
@@ -36,16 +48,12 @@ def get_sandwiches(swaps: List[Swap]) -> List[Sandwich]:
                 and other_swap.token_in_address == front_swap.token_out_address
                 and other_swap.from_address == sandwicher_address
             ):
-                    sandwiches.append(
-                        Sandwich(
+                return Sandwich(
                     block_number=front_swap.block_number,
                     sandwicher_address=sandwicher_address,
                     frontrun_transaction_hash=front_swap.transaction_hash,
                     backrun_transaction_hash=other_swap.transaction_hash,
                     sandwiched_swaps=sandwiched_swaps,
                 )
-                    )
 
-                    break
-
-    return sandwiches
+    return None