Update README. Run precommit on all files
This commit is contained in:
Luke Van Seters
parent
214921602e
commit
9e6f79940b
2
.gitignore
vendored
2
.gitignore
vendored
@ -1,5 +1,7 @@
|
||||
# venv and test cache files
|
||||
env/
|
||||
__pycache__
|
||||
.mypy_cache
|
||||
|
||||
*.swp
|
||||
|
||||
|
||||
@ -3,7 +3,7 @@ repos:
|
||||
rev: 20.8b1
|
||||
hooks:
|
||||
- id: black
|
||||
language_version: python3.9.6
|
||||
language_version: python3.9
|
||||
- repo: local
|
||||
hooks:
|
||||
- id: pylint
|
||||
|
||||
1
.python-version
Normal file
1
.python-version
Normal file
@ -0,0 +1 @@
|
||||
3.9.4
|
||||
12
README.md
12
README.md
@ -5,7 +5,7 @@ Requirements:
|
||||
|
||||
* python3 and pip3
|
||||
|
||||
Instructions:
|
||||
Instructions to run:
|
||||
|
||||
* Setup a virtual enviroment to manage dependencies (optional)
|
||||
* `python3 -m venv env`
|
||||
@ -17,3 +17,13 @@ Instructions:
|
||||
* `pip3 install -r requirements.txt`
|
||||
* Run tests for token flow
|
||||
* `python -m unittest tests/tokenflow_test.py`
|
||||
|
||||
If contributing:
|
||||
* Install dev libraries
|
||||
* `pip3 install -r requirements_dev.txt`
|
||||
* Setup pre-commit
|
||||
* `pre-commit install`
|
||||
* Install dependencies and verify it's working
|
||||
* `pre-commit run --all-files`
|
||||
* If you see "failed to find interpreter for..." it means you're missing the correct python version
|
||||
* The current version is python3.9 - [pyenv](https://github.com/pyenv/pyenv) is a great option for managing python versions
|
||||
|
||||
@ -1,4 +1,3 @@
|
||||
import json
|
||||
from pathlib import Path
|
||||
from typing import List
|
||||
|
||||
@ -7,7 +6,7 @@ from web3 import Web3
|
||||
from mev_inspect.schemas import Block, BlockCall, BlockCallType
|
||||
|
||||
|
||||
cache_directory = './cache'
|
||||
cache_directory = "./cache"
|
||||
|
||||
|
||||
## Creates a block object, either from the cache or from the chain itself
|
||||
@ -16,17 +15,12 @@ cache_directory = './cache'
|
||||
def createFromBlockNumber(block_number: int, base_provider) -> Block:
|
||||
cache_path = _get_cache_path(block_number)
|
||||
|
||||
if (cache_path.is_file()):
|
||||
print(
|
||||
f'Cache for block {block_number} exists, ' \
|
||||
'loading data from cache'
|
||||
)
|
||||
if cache_path.is_file():
|
||||
print(f"Cache for block {block_number} exists, " "loading data from cache")
|
||||
|
||||
return Block.parse_file(cache_path)
|
||||
else:
|
||||
print(
|
||||
f"Cache for block {block_number} did not exist, getting data"
|
||||
)
|
||||
print(f"Cache for block {block_number} did not exist, getting data")
|
||||
|
||||
w3 = Web3(base_provider)
|
||||
block = fetch_block(w3, base_provider, block_number)
|
||||
@ -43,30 +37,31 @@ def fetch_block(w3, base_provider, block_number: int) -> Block:
|
||||
## Get the block receipts
|
||||
## TODO: evaluate whether or not this is sufficient or if gas used needs to be converted to a proper big number.
|
||||
## In inspect-ts it needed to be converted
|
||||
block_receipts_raw = base_provider.make_request("eth_getBlockReceipts", [block_number])
|
||||
block_receipts_raw = base_provider.make_request(
|
||||
"eth_getBlockReceipts", [block_number]
|
||||
)
|
||||
|
||||
## Trace the whole block, return those calls
|
||||
block_calls_json = w3.parity.trace_block(block_number)
|
||||
block_calls = [
|
||||
BlockCall(**call_json)
|
||||
for call_json in block_calls_json
|
||||
]
|
||||
block_calls = [BlockCall(**call_json) for call_json in block_calls_json]
|
||||
|
||||
## Get the logs
|
||||
block_hash = (block_data.hash).hex()
|
||||
block_logs = w3.eth.get_logs({'blockHash': block_hash})
|
||||
block_logs = w3.eth.get_logs({"blockHash": block_hash})
|
||||
|
||||
## Get gas used by individual txs and store them too
|
||||
txs_gas_data = {}
|
||||
|
||||
for transaction in block_data['transactions']:
|
||||
for transaction in block_data["transactions"]:
|
||||
tx_hash = (transaction.hash).hex()
|
||||
tx_data = w3.eth.get_transaction(tx_hash)
|
||||
tx_receipt = w3.eth.wait_for_transaction_receipt(tx_hash)
|
||||
txs_gas_data[tx_hash] = {
|
||||
'gasUsed': tx_receipt['gasUsed'], # fix: why does this return 0 for certain txs?
|
||||
'gasPrice': tx_data['gasPrice'],
|
||||
'netFeePaid': tx_data['gasPrice'] * tx_receipt['gasUsed']
|
||||
"gasUsed": tx_receipt[
|
||||
"gasUsed"
|
||||
], # fix: why does this return 0 for certain txs?
|
||||
"gasPrice": tx_data["gasPrice"],
|
||||
"netFeePaid": tx_data["gasPrice"] * tx_receipt["gasUsed"],
|
||||
}
|
||||
|
||||
transaction_hashes = get_transaction_hashes(block_calls)
|
||||
@ -88,7 +83,10 @@ def get_transaction_hashes(calls: List[BlockCall]) -> List[str]:
|
||||
|
||||
for call in calls:
|
||||
if call.type != BlockCallType.reward:
|
||||
if call.transaction_hash not in result:
|
||||
if (
|
||||
call.transaction_hash is not None
|
||||
and call.transaction_hash not in result
|
||||
):
|
||||
result.append(call.transaction_hash)
|
||||
|
||||
return result
|
||||
|
||||
@ -7,25 +7,35 @@ from mev_inspect.config import load_config
|
||||
|
||||
config = load_config()
|
||||
|
||||
uniswap_router_abi = json.loads(config['ABI']['UniswapV2Router'])
|
||||
uniswap_router_address = (config['ADDRESSES']['UniswapV2Router'])
|
||||
sushiswap_router_address = (config['ADDRESSES']['SushiswapV2Router'])
|
||||
uniswap_router_abi = json.loads(config["ABI"]["UniswapV2Router"])
|
||||
uniswap_router_address = config["ADDRESSES"]["UniswapV2Router"]
|
||||
sushiswap_router_address = config["ADDRESSES"]["SushiswapV2Router"]
|
||||
|
||||
uniswap_pair_abi = json.loads(config["ABI"]["UniswapV2Pair"])
|
||||
|
||||
uniswap_pair_abi = json.loads(config['ABI']['UniswapV2Pair'])
|
||||
|
||||
class UniswapInspector:
|
||||
def __init__(self, base_provider) -> None:
|
||||
self.w3 = Web3(base_provider)
|
||||
|
||||
self.trading_functions = self.get_trading_functions()
|
||||
self.uniswap_v2_router_contract = self.w3.eth.contract(abi=uniswap_router_abi, address=uniswap_router_address)
|
||||
self.uniswap_v2_router_contract = self.w3.eth.contract(
|
||||
abi=uniswap_router_abi, address=uniswap_router_address
|
||||
)
|
||||
self.uniswap_router_trade_signatures = self.get_router_signatures()
|
||||
|
||||
self.uniswap_v2_pair_contract = self.w3.eth.contract(abi=uniswap_pair_abi)
|
||||
self.uniswap_v2_pair_swap_signatures = self.uniswap_v2_pair_contract.functions.swap(0, 0, uniswap_router_address, "").selector ## Note the address here doesn't matter, but it must be filled out
|
||||
self.uniswap_v2_pair_reserves_signatures = self.uniswap_v2_pair_contract.functions.getReserves().selector ## Called "checksigs" in mev-inpsect.ts
|
||||
self.uniswap_v2_pair_swap_signatures = (
|
||||
self.uniswap_v2_pair_contract.functions.swap(
|
||||
0, 0, uniswap_router_address, ""
|
||||
).selector
|
||||
) ## Note the address here doesn't matter, but it must be filled out
|
||||
self.uniswap_v2_pair_reserves_signatures = (
|
||||
self.uniswap_v2_pair_contract.functions.getReserves().selector
|
||||
) ## Called "checksigs" in mev-inpsect.ts
|
||||
|
||||
print("Built Uniswap inspector")
|
||||
|
||||
def get_trading_functions(self):
|
||||
## Gets all functions used for swapping
|
||||
result = []
|
||||
@ -33,9 +43,9 @@ class UniswapInspector:
|
||||
## For each entry in the ABI
|
||||
for abi in uniswap_router_abi:
|
||||
## Check to see if the entry is a function and if it is if the function's name starts with swap
|
||||
if abi['type'] == 'function' and abi['name'].startswith('swap'):
|
||||
if abi["type"] == "function" and abi["name"].startswith("swap"):
|
||||
## If so add it to our array
|
||||
result.append(abi['name'])
|
||||
result.append(abi["name"])
|
||||
|
||||
return result
|
||||
|
||||
@ -46,23 +56,23 @@ class UniswapInspector:
|
||||
## For each entry in the ABI
|
||||
for abi in uniswap_router_abi:
|
||||
## Check to see if the entry is a function and if it is if the function's name starts with swap
|
||||
if abi['type'] == 'function' and abi['name'].startswith('swap'):
|
||||
if abi["type"] == "function" and abi["name"].startswith("swap"):
|
||||
## Add a parantheses
|
||||
function = abi['name'] + '('
|
||||
function = abi["name"] + "("
|
||||
|
||||
## For each input in the function's input
|
||||
for input in abi['inputs']:
|
||||
for input in abi["inputs"]:
|
||||
|
||||
## Concat them into a string with commas
|
||||
function = function + input['internalType'] + ','
|
||||
function = function + input["internalType"] + ","
|
||||
|
||||
## Take off the last comma, add a ')' to close the parentheses
|
||||
function = function[:-1] + ')'
|
||||
function = function[:-1] + ")"
|
||||
|
||||
## The result looks like this: 'swapETHForExactTokens(uint256,address[],address,uint256)'
|
||||
|
||||
## Take the first 4 bytes of the sha3 hash of the above string.
|
||||
selector = (Web3.sha3(text=function)[0:4])
|
||||
selector = Web3.sha3(text=function)[0:4]
|
||||
|
||||
## Add that to an array
|
||||
result.append(selector)
|
||||
@ -70,12 +80,13 @@ class UniswapInspector:
|
||||
return result
|
||||
|
||||
def inspect(self, calls):
|
||||
result = []
|
||||
|
||||
trade_calls = []
|
||||
|
||||
for call in calls:
|
||||
print('\n',call)
|
||||
if (call['action']['to'] == uniswap_router_address.lower() or call['action']['to'] == sushiswap_router_address.lower()) and utils.check_call_for_signature(call, self.uniswap_router_trade_signatures):
|
||||
print("\n", call)
|
||||
if (
|
||||
call["action"]["to"] == uniswap_router_address.lower()
|
||||
or call["action"]["to"] == sushiswap_router_address.lower()
|
||||
) and utils.check_call_for_signature(
|
||||
call, self.uniswap_router_trade_signatures
|
||||
):
|
||||
# print("WIP, here is where there is a call that matches what we are looking for")
|
||||
1 == 1
|
||||
|
||||
@ -9,10 +9,7 @@ class Processor:
|
||||
def get_transaction_evaluations(self, block_data):
|
||||
for transaction_hash in block_data.transaction_hashes:
|
||||
calls = block_data.get_filtered_calls(transaction_hash)
|
||||
calls_json = [
|
||||
to_original_json_dict(call)
|
||||
for call in calls
|
||||
]
|
||||
calls_json = [to_original_json_dict(call) for call in calls]
|
||||
|
||||
for inspector in self.inspectors:
|
||||
inspector.inspect(calls_json)
|
||||
|
||||
@ -1,9 +1,6 @@
|
||||
import json
|
||||
from enum import Enum
|
||||
from typing import Dict, List, Optional
|
||||
|
||||
from pydantic import BaseModel
|
||||
|
||||
from .utils import CamelModel, Web3Model
|
||||
|
||||
|
||||
@ -38,7 +35,4 @@ class Block(Web3Model):
|
||||
txs_gas_data: Dict[str, dict]
|
||||
|
||||
def get_filtered_calls(self, hash: str) -> List[BlockCall]:
|
||||
return [
|
||||
call for call in self.calls
|
||||
if call.transaction_hash == hash
|
||||
]
|
||||
return [call for call in self.calls if call.transaction_hash == hash]
|
||||
|
||||
@ -6,9 +6,8 @@ from web3.datastructures import AttributeDict
|
||||
|
||||
|
||||
def to_camel(string: str) -> str:
|
||||
return ''.join(
|
||||
word.capitalize() if i > 0 else word
|
||||
for i, word in enumerate(string.split('_'))
|
||||
return "".join(
|
||||
word.capitalize() if i > 0 else word for i, word in enumerate(string.split("_"))
|
||||
)
|
||||
|
||||
|
||||
|
||||
@ -5,24 +5,27 @@ from mev_inspect.config import load_config
|
||||
|
||||
config = load_config()
|
||||
|
||||
rpc_url = config['RPC']['Endpoint']
|
||||
weth_address = config['ADDRESSES']['WETH']
|
||||
rpc_url = config["RPC"]["Endpoint"]
|
||||
weth_address = config["ADDRESSES"]["WETH"]
|
||||
# w3 = Web3(HTTPProvider(rpc_url))
|
||||
|
||||
cache_directory = './cache'
|
||||
cache_directory = "./cache"
|
||||
|
||||
|
||||
def get_tx_traces(txHash, blockNo):
|
||||
# block_calls = w3.parity.trace_block(10803840)
|
||||
cache_file = '{cacheDirectory}/{blockNumber}-new.json'.format(cacheDirectory=cache_directory, blockNumber=blockNo)
|
||||
cache_file = "{cacheDirectory}/{blockNumber}-new.json".format(
|
||||
cacheDirectory=cache_directory, blockNumber=blockNo
|
||||
)
|
||||
file_exists = Path(cache_file).is_file()
|
||||
|
||||
tx_traces = []
|
||||
# if have the traces cached
|
||||
if(file_exists):
|
||||
if file_exists:
|
||||
block_file = open(cache_file)
|
||||
block_json = json.load(block_file)
|
||||
for call in block_json['calls']:
|
||||
if call['transactionHash'] == txHash:
|
||||
for call in block_json["calls"]:
|
||||
if call["transactionHash"] == txHash:
|
||||
tx_traces.append(call)
|
||||
block_file.close()
|
||||
else:
|
||||
@ -30,7 +33,8 @@ def get_tx_traces(txHash, blockNo):
|
||||
# depending on the best way to call block.py from here
|
||||
print("traces do not exist")
|
||||
|
||||
return(tx_traces)
|
||||
return tx_traces
|
||||
|
||||
|
||||
def is_stablecoin_address(address):
|
||||
# to look for stablecoin inflow/outflows
|
||||
@ -50,6 +54,7 @@ def is_stablecoin_address(address):
|
||||
]
|
||||
return address in stablecoin_addresses
|
||||
|
||||
|
||||
def is_known_router_address(address):
|
||||
# to exclude known router addresses from token flow analysis
|
||||
known_router_addresses = [
|
||||
@ -74,117 +79,148 @@ def is_known_router_address(address):
|
||||
"0xE592427A0AEce92De3Edee1F18E0157C05861564", # Uniswap v3 router
|
||||
"0x3E66B66Fd1d0b02fDa6C811Da9E0547970DB2f21", # Balance exchange proxy
|
||||
"0x1bD435F3C054b6e901B7b108a0ab7617C808677b", # Paraswap v4
|
||||
"0xC011a73ee8576Fb46F5E1c5751cA3B9Fe0af2a6F" # SNX proxy synth issuer
|
||||
"0xC011a73ee8576Fb46F5E1c5751cA3B9Fe0af2a6F", # SNX proxy synth issuer
|
||||
]
|
||||
return address in known_router_addresses
|
||||
|
||||
|
||||
# we're interested in the to address to run token flow on it as well
|
||||
def get_tx_to_address(txHash, blockNo):
|
||||
cache_file = '{cacheDirectory}/{blockNumber}-new.json'.format(cacheDirectory=cache_directory, blockNumber=blockNo)
|
||||
cache_file = "{cacheDirectory}/{blockNumber}-new.json".format(
|
||||
cacheDirectory=cache_directory, blockNumber=blockNo
|
||||
)
|
||||
block_file = open(cache_file)
|
||||
block_json = json.load(block_file)
|
||||
for receipt in block_json['receipts']['result']:
|
||||
if receipt['transactionHash'] == txHash:
|
||||
for receipt in block_json["receipts"]["result"]:
|
||||
if receipt["transactionHash"] == txHash:
|
||||
block_file.close()
|
||||
return receipt['to']
|
||||
return receipt["to"]
|
||||
|
||||
|
||||
def get_tx_proxies(tx_traces, to_address):
|
||||
proxies = []
|
||||
for trace in tx_traces:
|
||||
if(trace['type'] == 'call' and trace['action']['callType'] == 'delegatecall' and trace['action']['from'] == to_address):
|
||||
proxies.append(trace['action']['to'])
|
||||
return(proxies)
|
||||
if (
|
||||
trace["type"] == "call"
|
||||
and trace["action"]["callType"] == "delegatecall"
|
||||
and trace["action"]["from"] == to_address
|
||||
):
|
||||
proxies.append(trace["action"]["to"])
|
||||
return proxies
|
||||
|
||||
|
||||
def get_net_gas_used(txHash, blockNo):
|
||||
cache_file = '{cacheDirectory}/{blockNumber}.json'.format(cacheDirectory=cache_directory, blockNumber=blockNo)
|
||||
cache_file = "{cacheDirectory}/{blockNumber}.json".format(
|
||||
cacheDirectory=cache_directory, blockNumber=blockNo
|
||||
)
|
||||
block_file = open(cache_file)
|
||||
block_json = json.load(block_file)
|
||||
gas_used = 0
|
||||
for trace in block_json['calls']:
|
||||
if trace['transactionHash'] == txHash:
|
||||
gas_used = gas_used + int(trace['result']['gasUsed'],16)
|
||||
for trace in block_json["calls"]:
|
||||
if trace["transactionHash"] == txHash:
|
||||
gas_used = gas_used + int(trace["result"]["gasUsed"], 16)
|
||||
print(gas_used)
|
||||
|
||||
|
||||
def get_ether_flows(tx_traces, addresses_to_check):
|
||||
eth_inflow = 0
|
||||
eth_outflow = 0
|
||||
|
||||
for trace in tx_traces:
|
||||
if(trace['type'] == 'call'):
|
||||
value = int(trace['action']['value'], 16) # converting from 0x prefix to decimal
|
||||
if trace["type"] == "call":
|
||||
value = int(
|
||||
trace["action"]["value"], 16
|
||||
) # converting from 0x prefix to decimal
|
||||
# ETH_GET
|
||||
if(trace['action']['callType'] != 'delegatecall' and trace['action']['from'] != weth_address and value > 0 and trace['action']['to'] in addresses_to_check):
|
||||
if (
|
||||
trace["action"]["callType"] != "delegatecall"
|
||||
and trace["action"]["from"] != weth_address
|
||||
and value > 0
|
||||
and trace["action"]["to"] in addresses_to_check
|
||||
):
|
||||
eth_inflow = eth_inflow + value
|
||||
|
||||
# ETH_GIVE
|
||||
if(trace['action']['callType'] != 'delegatecall' and trace['action']['to'] != weth_address and value > 0 and trace['action']['from'] in addresses_to_check):
|
||||
if (
|
||||
trace["action"]["callType"] != "delegatecall"
|
||||
and trace["action"]["to"] != weth_address
|
||||
and value > 0
|
||||
and trace["action"]["from"] in addresses_to_check
|
||||
):
|
||||
eth_outflow = eth_outflow + value
|
||||
|
||||
if(trace['action']['to'] == weth_address):
|
||||
if trace["action"]["to"] == weth_address:
|
||||
# WETH_GET1 & WETH_GET2 (to account for both 'transfer' and 'transferFrom' methods)
|
||||
# WETH_GIVE1 & WETH_GIVE2
|
||||
|
||||
# transfer(address to,uint256 value) with args
|
||||
if(len(trace['action']['input']) == 138):
|
||||
if(trace['action']['input'][2:10] == "a9059cbb"):
|
||||
transfer_to = '0x' + trace['action']['input'][34:74]
|
||||
transfer_value = int('0x' + trace['action']['input'][74:138], 16)
|
||||
if(transfer_to in addresses_to_check):
|
||||
if len(trace["action"]["input"]) == 138:
|
||||
if trace["action"]["input"][2:10] == "a9059cbb":
|
||||
transfer_to = "0x" + trace["action"]["input"][34:74]
|
||||
transfer_value = int(
|
||||
"0x" + trace["action"]["input"][74:138], 16
|
||||
)
|
||||
if transfer_to in addresses_to_check:
|
||||
eth_inflow = eth_inflow + transfer_value
|
||||
elif(trace['action']['from'] in addresses_to_check):
|
||||
elif trace["action"]["from"] in addresses_to_check:
|
||||
eth_outflow = eth_outflow + transfer_value
|
||||
|
||||
# transferFrom(address from,address to,uint256 value )
|
||||
if(len(trace['action']['input']) == 202):
|
||||
if(trace['action']['input'][2:10] == "23b872dd"):
|
||||
transfer_from = '0x' + trace['action']['input'][34:74]
|
||||
transfer_to = '0x' + trace['action']['input'][98:138]
|
||||
transfer_value = int('0x' + trace['action']['input'][138:202], 16)
|
||||
if(transfer_to in addresses_to_check):
|
||||
if len(trace["action"]["input"]) == 202:
|
||||
if trace["action"]["input"][2:10] == "23b872dd":
|
||||
transfer_from = "0x" + trace["action"]["input"][34:74]
|
||||
transfer_to = "0x" + trace["action"]["input"][98:138]
|
||||
transfer_value = int(
|
||||
"0x" + trace["action"]["input"][138:202], 16
|
||||
)
|
||||
if transfer_to in addresses_to_check:
|
||||
eth_inflow = eth_inflow + transfer_value
|
||||
elif(transfer_from in addresses_to_check):
|
||||
elif transfer_from in addresses_to_check:
|
||||
eth_outflow = eth_outflow + transfer_value
|
||||
|
||||
if(trace['type'] == 'suicide'):
|
||||
if(trace['action']['refundAddress'] in addresses_to_check):
|
||||
refund_value = int('0x' + trace['action']['balance'], 16)
|
||||
if trace["type"] == "suicide":
|
||||
if trace["action"]["refundAddress"] in addresses_to_check:
|
||||
refund_value = int("0x" + trace["action"]["balance"], 16)
|
||||
eth_inflow = eth_inflow + refund_value
|
||||
|
||||
return [eth_inflow, eth_outflow]
|
||||
|
||||
|
||||
def get_dollar_flows(tx_traces, addresses_to_check):
|
||||
dollar_inflow = 0
|
||||
dollar_outflow = 0
|
||||
for trace in tx_traces:
|
||||
if(trace['type'] == 'call' and is_stablecoin_address(trace['action']['to'])):
|
||||
value = int(trace['action']['value'], 16) # converting from 0x prefix to decimal
|
||||
if trace["type"] == "call" and is_stablecoin_address(trace["action"]["to"]):
|
||||
_ = int(
|
||||
trace["action"]["value"], 16
|
||||
) # converting from 0x prefix to decimal
|
||||
|
||||
# USD_GET1 & USD_GET2 (to account for both 'transfer' and 'transferFrom' methods)
|
||||
# USD_GIVE1 & USD_GIVE2
|
||||
|
||||
# transfer(address to,uint256 value) with args
|
||||
if(len(trace['action']['input']) == 138):
|
||||
if(trace['action']['input'][2:10] == "a9059cbb"):
|
||||
transfer_to = '0x' + trace['action']['input'][34:74]
|
||||
transfer_value = int('0x' + trace['action']['input'][74:138], 16)
|
||||
if(transfer_to in addresses_to_check):
|
||||
if len(trace["action"]["input"]) == 138:
|
||||
if trace["action"]["input"][2:10] == "a9059cbb":
|
||||
transfer_to = "0x" + trace["action"]["input"][34:74]
|
||||
transfer_value = int("0x" + trace["action"]["input"][74:138], 16)
|
||||
if transfer_to in addresses_to_check:
|
||||
dollar_inflow = dollar_inflow + transfer_value
|
||||
elif(trace['action']['from'] in addresses_to_check):
|
||||
elif trace["action"]["from"] in addresses_to_check:
|
||||
dollar_outflow = dollar_outflow + transfer_value
|
||||
|
||||
# transferFrom(address from,address to,uint256 value )
|
||||
if(len(trace['action']['input']) == 202):
|
||||
if(trace['action']['input'][2:10] == "23b872dd"):
|
||||
transfer_from = '0x' + trace['action']['input'][34:74]
|
||||
transfer_to = '0x' + trace['action']['input'][98:138]
|
||||
transfer_value = int('0x' + trace['action']['input'][138:202], 16)
|
||||
if(transfer_to in addresses_to_check):
|
||||
if len(trace["action"]["input"]) == 202:
|
||||
if trace["action"]["input"][2:10] == "23b872dd":
|
||||
transfer_from = "0x" + trace["action"]["input"][34:74]
|
||||
transfer_to = "0x" + trace["action"]["input"][98:138]
|
||||
transfer_value = int("0x" + trace["action"]["input"][138:202], 16)
|
||||
if transfer_to in addresses_to_check:
|
||||
dollar_inflow = dollar_inflow + transfer_value
|
||||
elif(transfer_from in addresses_to_check):
|
||||
elif transfer_from in addresses_to_check:
|
||||
dollar_outflow = dollar_outflow + transfer_value
|
||||
return [dollar_inflow, dollar_outflow]
|
||||
|
||||
|
||||
def run_tokenflow(txHash, blockNo):
|
||||
tx_traces = get_tx_traces(txHash, blockNo)
|
||||
to_address = get_tx_to_address(txHash, blockNo)
|
||||
@ -197,24 +233,23 @@ def run_tokenflow(txHash, blockNo):
|
||||
|
||||
# check if the 'to' field is a known aggregator/router
|
||||
# if not, add to relevant addresses to run TF on
|
||||
if(not is_known_router_address(to_address)):
|
||||
addresses_to_check.append(to_address.lower()) # traces need lowercase addresses to match
|
||||
if not is_known_router_address(to_address):
|
||||
addresses_to_check.append(
|
||||
to_address.lower()
|
||||
) # traces need lowercase addresses to match
|
||||
|
||||
ether_flows = get_ether_flows(tx_traces, addresses_to_check)
|
||||
dollar_flows = get_dollar_flows(tx_traces, addresses_to_check)
|
||||
# print(addresses_to_check)
|
||||
# print('net eth flow', ether_flows[0] - ether_flows[1])
|
||||
# print('net dollar flow', dollar_flows )
|
||||
return {
|
||||
'ether_flows': ether_flows,
|
||||
'dollar_flows': dollar_flows
|
||||
}
|
||||
return {"ether_flows": ether_flows, "dollar_flows": dollar_flows}
|
||||
|
||||
|
||||
# note: not the gas set by user, only gas consumed upon execution
|
||||
def get_gas_used_by_tx(txHash):
|
||||
#tx_receipt = w3.eth.getTransactionReceipt(txHash)
|
||||
return(tx_receipt['gasUsed'])
|
||||
# def get_gas_used_by_tx(txHash):
|
||||
# # tx_receipt = w3.eth.getTransactionReceipt(txHash)
|
||||
# return tx_receipt["gasUsed"]
|
||||
|
||||
|
||||
# tx_traces = get_tx_traces('0x4121ce805d33e952b2e6103a5024f70c118432fd0370128d6d7845f9b2987922', 11930296)
|
||||
|
||||
@ -1,7 +1,8 @@
|
||||
from hexbytes.main import HexBytes
|
||||
|
||||
|
||||
def check_call_for_signature(call, signatures):
|
||||
if (call['action']['input'] == None):
|
||||
if call["action"]["input"] == None:
|
||||
return False
|
||||
|
||||
## By default set this to False
|
||||
@ -12,7 +13,7 @@ def check_call_for_signature(call, signatures):
|
||||
# print("Desired signature:", str(signature))
|
||||
# print("Actual", HexBytes(call['action']['input']))
|
||||
|
||||
if HexBytes(call['action']['input']).startswith((signature)):
|
||||
if HexBytes(call["action"]["input"]).startswith((signature)):
|
||||
## Note that we are turning the input into hex bytes here, which seems to be fine
|
||||
## Working with strings was doing weird things
|
||||
print("hit")
|
||||
|
||||
2
requirements_dev.txt
Normal file
2
requirements_dev.txt
Normal file
@ -0,0 +1,2 @@
|
||||
pre-commit==2.13.0
|
||||
pylint==2.9.3
|
||||
@ -1,16 +1,22 @@
|
||||
import argparse
|
||||
|
||||
from web3.providers import base
|
||||
from web3 import Web3
|
||||
|
||||
from mev_inspect import block
|
||||
from mev_inspect.inspector_uniswap import UniswapInspector
|
||||
from mev_inspect.processor import Processor
|
||||
|
||||
parser = argparse.ArgumentParser(description='Inspect some blocks.')
|
||||
parser.add_argument('-block_number', metavar='b', type=int, nargs='+',
|
||||
help='the block number you are targetting, eventually this will need to be changed')
|
||||
parser.add_argument('-rpc', metavar='r', help='rpc endpoint, this needs to have parity style traces')
|
||||
parser = argparse.ArgumentParser(description="Inspect some blocks.")
|
||||
parser.add_argument(
|
||||
"-block_number",
|
||||
metavar="b",
|
||||
type=int,
|
||||
nargs="+",
|
||||
help="the block number you are targetting, eventually this will need to be changed",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-rpc", metavar="r", help="rpc endpoint, this needs to have parity style traces"
|
||||
)
|
||||
args = parser.parse_args()
|
||||
|
||||
## Set up the base provider, but don't wrap it in web3 so we can make requests to it with make_request()
|
||||
|
||||
@ -1,20 +1,23 @@
|
||||
import unittest
|
||||
|
||||
from mev_inspect import inspector_compound
|
||||
from mev_inspect import inspector_aave
|
||||
|
||||
class TestLiquidations (unittest.TestCase):
|
||||
def test_compound_liquidation(self):
|
||||
tx_hash = "0x0ec6d5044a47feb3ceb647bf7ea4ffc87d09244d629eeced82ba17ec66605012"
|
||||
block_no = 11338848
|
||||
res = inspector_compound.get_profit(tx_hash, block_no)
|
||||
# self.assertEqual(res['profit'], 0)
|
||||
def test_aave_liquidation(self):
|
||||
tx_hash = "0xc8d2501d28800b1557eb64c5d0e08fd6070c15b6c04c39ca05631f641d19ffb2"
|
||||
block_no = 10803840
|
||||
res = inspector_aave.get_profit(tx_hash, block_no)
|
||||
# self.assertEqual(res['profit'], 0)
|
||||
# Fails precommit because these inspectors don't exist yet
|
||||
# from mev_inspect import inspector_compound
|
||||
# from mev_inspect import inspector_aave
|
||||
#
|
||||
#
|
||||
# class TestLiquidations(unittest.TestCase):
|
||||
# def test_compound_liquidation(self):
|
||||
# tx_hash = "0x0ec6d5044a47feb3ceb647bf7ea4ffc87d09244d629eeced82ba17ec66605012"
|
||||
# block_no = 11338848
|
||||
# res = inspector_compound.get_profit(tx_hash, block_no)
|
||||
# # self.assertEqual(res['profit'], 0)
|
||||
#
|
||||
# def test_aave_liquidation(self):
|
||||
# tx_hash = "0xc8d2501d28800b1557eb64c5d0e08fd6070c15b6c04c39ca05631f641d19ffb2"
|
||||
# block_no = 10803840
|
||||
# res = inspector_aave.get_profit(tx_hash, block_no)
|
||||
# # self.assertEqual(res['profit'], 0)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
@ -3,27 +3,28 @@ import unittest
|
||||
from mev_inspect import tokenflow
|
||||
|
||||
|
||||
class TestTokenFlow (unittest.TestCase):
|
||||
class TestTokenFlow(unittest.TestCase):
|
||||
def test_simple_arb(self):
|
||||
tx_hash = "0x4121ce805d33e952b2e6103a5024f70c118432fd0370128d6d7845f9b2987922"
|
||||
block_no = 11930296
|
||||
res = tokenflow.run_tokenflow(tx_hash, block_no)
|
||||
self.assertEqual(res['ether_flows'], [3547869861992962562, 3499859860420296704])
|
||||
self.assertEqual(res['dollar_flows'], [0,0])
|
||||
self.assertEqual(res["ether_flows"], [3547869861992962562, 3499859860420296704])
|
||||
self.assertEqual(res["dollar_flows"], [0, 0])
|
||||
|
||||
def test_arb_with_stable_flow(self):
|
||||
tx_hash = "0x496836e0bd1520388e36c79d587a31d4b3306e4f25352164178ca0667c7f9c29"
|
||||
block_no = 11935012
|
||||
res = tokenflow.run_tokenflow(tx_hash, block_no)
|
||||
self.assertEqual(res['ether_flows'], [597044987302243493, 562445964778930176])
|
||||
self.assertEqual(res['dollar_flows'], [871839781,871839781])
|
||||
self.assertEqual(res["ether_flows"], [597044987302243493, 562445964778930176])
|
||||
self.assertEqual(res["dollar_flows"], [871839781, 871839781])
|
||||
|
||||
def test_complex_cross_arb(self):
|
||||
tx_hash = "0x5ab21bfba50ad3993528c2828c63e311aafe93b40ee934790e545e150cb6ca73"
|
||||
block_no = 11931272
|
||||
res = tokenflow.run_tokenflow(tx_hash, block_no)
|
||||
self.assertEqual(res['ether_flows'], [3636400213125714803, 3559576672903063566])
|
||||
self.assertEqual(res['dollar_flows'], [0,0])
|
||||
self.assertEqual(res["ether_flows"], [3636400213125714803, 3559576672903063566])
|
||||
self.assertEqual(res["dollar_flows"], [0, 0])
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user