add profit_token_address to sandwiches

This commit is contained in:
Caleb 2022-01-17 17:15:10 -08:00
parent 189b7d1220
commit 5467459004
5 changed files with 7 additions and 2 deletions

View File

@ -18,10 +18,11 @@ depends_on = None
def upgrade(): def upgrade():
op.add_column( op.add_column(
"sandwiches", "sandwiches", sa.Column("profit_token_address", sa.String(256), nullable=True)
sa.Column("profit_amount", sa.Numeric, nullable=True),
) )
op.add_column("sandwiches", sa.Column("profit_amount", sa.Numeric, nullable=True))
def downgrade(): def downgrade():
op.drop_column("sandwiches", "profit_token_address")
op.drop_column("sandwiches", "profit_amount") op.drop_column("sandwiches", "profit_amount")

View File

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

View File

@ -14,4 +14,5 @@ class SandwichModel(Base):
frontrun_swap_trace_address = Column(ARRAY(Integer), nullable=False) frontrun_swap_trace_address = Column(ARRAY(Integer), nullable=False)
backrun_swap_transaction_hash = Column(String(256), nullable=False) backrun_swap_transaction_hash = Column(String(256), nullable=False)
backrun_swap_trace_address = Column(ARRAY(Integer), 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) profit_amount = Column(Numeric, nullable=False)

View File

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

View File

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