Merge pull request #233 from CalebEverett/sandwich_profit

add sandwich profit

alembic/versions/b26ab0051a88_add_profit_amount_column_to_sandwiches_.py

@@ -0,0 +1,27 @@
+"""add profit_amount column to sandwiches table
+
+Revision ID: b26ab0051a88
+Revises: 3c54832385e3
+Create Date: 2022-01-16 13:45:10.190969
+
+"""
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "b26ab0051a88"
+down_revision = "3c54832385e3"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    op.add_column(
+        "sandwiches", sa.Column("profit_token_address", sa.String(256), nullable=True)
+    )
+    op.add_column("sandwiches", sa.Column("profit_amount", sa.Numeric, nullable=True))
+
+
+def downgrade():
+    op.drop_column("sandwiches", "profit_token_address")
+    op.drop_column("sandwiches", "profit_amount")

mev_inspect/crud/sandwiches.py

@@ -39,6 +39,8 @@ def write_sandwiches(
                 frontrun_swap_trace_address=sandwich.frontrun_swap.trace_address,
                 backrun_swap_transaction_hash=sandwich.backrun_swap.transaction_hash,
                 backrun_swap_trace_address=sandwich.backrun_swap.trace_address,
+                profit_token_address=sandwich.profit_token_address,
+                profit_amount=sandwich.profit_amount,
             )
         )
 

mev_inspect/models/sandwiches.py

@@ -14,3 +14,5 @@ class SandwichModel(Base):
     frontrun_swap_trace_address = Column(ARRAY(Integer), nullable=False)
     backrun_swap_transaction_hash = Column(String(256), nullable=False)
     backrun_swap_trace_address = Column(ARRAY(Integer), nullable=False)
+    profit_token_address = Column(String(256), nullable=False)
+    profit_amount = Column(Numeric, nullable=False)

mev_inspect/sandwiches.py

@@ -62,6 +62,9 @@ def _get_sandwich_starting_with_swap(
                         frontrun_swap=front_swap,
                         backrun_swap=other_swap,
                         sandwiched_swaps=sandwiched_swaps,
+                        profit_token_address=front_swap.token_in_address,
+                        profit_amount=other_swap.token_out_amount
+                        - front_swap.token_in_amount,
                     )
 
     return None

mev_inspect/schemas/sandwiches.py

@@ -11,3 +11,5 @@ class Sandwich(BaseModel):
     frontrun_swap: Swap
     backrun_swap: Swap
     sandwiched_swaps: List[Swap]
+    profit_token_address: str
+    profit_amount: int

tests/sandwiches/12775690.json

@@ -1,4 +1,5 @@
-[{
+[
+    {
         "block_number": 12775690,
         "sandwicher_address": "0x000000000027d2efc283613d0c3e24a8b430c4d8",
         "frontrun_swap": {
@@ -112,5 +113,8 @@
                 "protocol": null,
                 "error": null
             }
+        ],
+        "profit_token_address": "0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2",
+        "profit_amount": 49991481915322274
+    }
 ]
-}]