880992362b
* feat: save profit to db Signed-off-by: Arthurim <arthurbdauphine@gmail.com> * fix: update total profit table and related crud Signed-off-by: Luca Georges Francois <luca@quartz.technology> Signed-off-by: Arthurim <arthurbdauphine@gmail.com> Signed-off-by: Luca Georges Francois <luca@quartz.technology> Co-authored-by: Luca Georges Francois <luca@quartz.technology>
104 lines
3.7 KiB
Python
104 lines
3.7 KiB
Python
import logging
|
|
from typing import Any, Dict, List
|
|
|
|
from sqlalchemy import orm
|
|
from web3 import Web3
|
|
|
|
from mev_inspect.arbitrages import get_arbitrages
|
|
from mev_inspect.block import get_classified_traces_from_events
|
|
from mev_inspect.crud.total_profits import write_total_profits_for_blocks
|
|
from mev_inspect.schemas.total_profits import TotalProfits
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
TRAILING_ZEROS = "000000000000000000000000"
|
|
|
|
|
|
async def inspect_block(
|
|
inspect_db_session: orm.Session,
|
|
w3: Web3,
|
|
block_number: int,
|
|
):
|
|
await inspect_many_blocks(
|
|
inspect_db_session,
|
|
w3,
|
|
block_number,
|
|
block_number + 1,
|
|
)
|
|
|
|
|
|
async def inspect_many_blocks(
|
|
inspect_db_session: orm.Session,
|
|
w3: Web3,
|
|
after_block_number: int,
|
|
before_block_number: int,
|
|
):
|
|
count = 0
|
|
arbitrages_payload = []
|
|
liquidations_payload = []
|
|
|
|
profits: List[TotalProfits] = []
|
|
async for swaps, liquidations in get_classified_traces_from_events(
|
|
w3, after_block_number, before_block_number
|
|
):
|
|
arbitrages = get_arbitrages(swaps)
|
|
|
|
if len(arbitrages) > 0:
|
|
for arb in arbitrages:
|
|
arb_payload: Dict[str, Any] = dict()
|
|
arb_payload["block_number"] = arb.block_number
|
|
arb_payload["transaction"] = arb.transaction_hash
|
|
arb_payload["account"] = arb.account_address
|
|
arb_payload["profit_amt"] = arb.profit_amount
|
|
arb_payload["token"] = arb.profit_token_address
|
|
arbitrages_payload.append(arb_payload)
|
|
count += 1
|
|
profits.append(
|
|
TotalProfits(
|
|
block_number=arb.block_number,
|
|
transaction_hash=arb.transaction_hash,
|
|
token_debt="",
|
|
amount_debt=0,
|
|
token_received=str(arb.profit_token_address).replace(
|
|
TRAILING_ZEROS, ""
|
|
),
|
|
amount_received=arb.profit_amount,
|
|
)
|
|
)
|
|
|
|
if len(liquidations) > 0:
|
|
for liq in liquidations:
|
|
liq_payload: Dict[str, Any] = dict()
|
|
liq_payload["block_number"] = liq.block_number
|
|
liq_payload["transaction"] = liq.transaction_hash
|
|
liq_payload["liquidator"] = liq.liquidator_user
|
|
liq_payload["purchase_addr"] = liq.debt_token_address
|
|
liq_payload["receive_addr"] = liq.received_token_address
|
|
liq_payload["purchase_amount"] = liq.debt_purchase_amount
|
|
liq_payload["receive_amount"] = liq.received_amount
|
|
liquidations_payload.append(liq_payload)
|
|
count += 1
|
|
profits.append(
|
|
TotalProfits(
|
|
block_number=liq.block_number,
|
|
transaction_hash=liq.transaction_hash,
|
|
token_debt=str(liq.debt_token_address).replace(
|
|
TRAILING_ZEROS, ""
|
|
),
|
|
amount_debt=liq.debt_purchase_amount,
|
|
token_received=str(liq.received_amount).replace(
|
|
TRAILING_ZEROS, ""
|
|
),
|
|
amount_received=liq.received_amount,
|
|
)
|
|
)
|
|
|
|
if count > 0:
|
|
print("writing profits of {0} mev transactions".format(count))
|
|
write_total_profits_for_blocks(
|
|
inspect_db_session=inspect_db_session, total_profits_for_blocks=profits
|
|
)
|
|
arbitrages_payload = []
|
|
liquidations_payload = []
|
|
count = 0
|