crowetic/mev-inspect-py
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

style: improve schema naming bcuz imagine complained

Browse Source
This commit is contained in:
Robert Miller 2021-11-16 19:57:58 -05:00
parent 0d6215f82e
commit 96d2171daa
4 changed files with 16 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
mev_inspect/punks.py
View File

@ -5,12 +5,12 @@ from mev_inspect.schemas.traces import (
Classification, Classification,
DecodedCallTrace, DecodedCallTrace,
) )
from mev_inspect.schemas.punk_bid import Punk_Bid from mev_inspect.schemas.punk_bid import PunkBid
from mev_inspect.schemas.punk_accept_bid import Punk_Accept_Bid from mev_inspect.schemas.punk_accept_bid import PunkAcceptBid
from mev_inspect.traces import get_traces_by_transaction_hash from mev_inspect.traces import get_traces_by_transaction_hash
def get_punk_accept_bids(traces: List[ClassifiedTrace]) -> List[Punk_Accept_Bid]: def get_punk_accept_bids(traces: List[ClassifiedTrace]) -> List[PunkAcceptBid]:
punk_accept_bids = [] punk_accept_bids = []
for _, transaction_traces in get_traces_by_transaction_hash(traces).items(): for _, transaction_traces in get_traces_by_transaction_hash(traces).items():
@ -23,7 +23,7 @@ def get_punk_accept_bids(traces: List[ClassifiedTrace]) -> List[Punk_Accept_Bid]
def _get_punk_accept_bids_for_transaction( def _get_punk_accept_bids_for_transaction(
traces: List[ClassifiedTrace], traces: List[ClassifiedTrace],
) -> List[Punk_Accept_Bid]: ) -> List[PunkAcceptBid]:
ordered_traces = list(sorted(traces, key=lambda t: t.trace_address)) ordered_traces = list(sorted(traces, key=lambda t: t.trace_address))
punk_accept_bids = [] punk_accept_bids = []
@ -33,7 +33,7 @@ def _get_punk_accept_bids_for_transaction(
continue continue
elif trace.classification == Classification.punk_accept_bid: elif trace.classification == Classification.punk_accept_bid:
punk_accept_bid = Punk_Accept_Bid( punk_accept_bid = PunkAcceptBid(
block_number=trace.block_number, block_number=trace.block_number,
transaction_hash=trace.transaction_hash, transaction_hash=trace.transaction_hash,
trace_address=trace.trace_address, trace_address=trace.trace_address,
@ -47,7 +47,7 @@ def _get_punk_accept_bids_for_transaction(
return punk_accept_bids return punk_accept_bids
def get_punk_bids(traces: List[ClassifiedTrace]) -> List[Punk_Bid]: def get_punk_bids(traces: List[ClassifiedTrace]) -> List[PunkBid]:
punk_bids = [] punk_bids = []
for _, transaction_traces in get_traces_by_transaction_hash(traces).items(): for _, transaction_traces in get_traces_by_transaction_hash(traces).items():
@ -56,7 +56,7 @@ def get_punk_bids(traces: List[ClassifiedTrace]) -> List[Punk_Bid]:
return punk_bids return punk_bids
def _get_punk_bids_for_transaction(traces: List[ClassifiedTrace]) -> List[Punk_Bid]: def _get_punk_bids_for_transaction(traces: List[ClassifiedTrace]) -> List[PunkBid]:
ordered_traces = list(sorted(traces, key=lambda t: t.trace_address)) ordered_traces = list(sorted(traces, key=lambda t: t.trace_address))
punk_bids = [] punk_bids = []
@ -66,7 +66,7 @@ def _get_punk_bids_for_transaction(traces: List[ClassifiedTrace]) -> List[Punk_B
continue continue
elif trace.classification == Classification.punk_bid: elif trace.classification == Classification.punk_bid:
punk_bid = Punk_Bid( punk_bid = PunkBid(
transaction_hash=trace.transaction_hash, transaction_hash=trace.transaction_hash,
block_number=trace.block_number, block_number=trace.block_number,
trace_address=trace.trace_address, trace_address=trace.trace_address,

8
mev_inspect/schemas/classifiers.py
View File

@ -5,8 +5,8 @@ from pydantic import BaseModel
from .traces import Classification, DecodedCallTrace, Protocol from .traces import Classification, DecodedCallTrace, Protocol
from .transfers import Transfer from .transfers import Transfer
from .punk_bid import Punk_Bid from .punk_bid import PunkBid
from .punk_accept_bid import Punk_Accept_Bid from .punk_accept_bid import PunkAcceptBid
class Classifier(ABC): class Classifier(ABC):
@ -23,7 +23,7 @@ class PunkBidClassifier(Classifier):
@staticmethod @staticmethod
@abstractmethod @abstractmethod
def get_bid(trace: DecodedCallTrace) -> Punk_Bid: def get_bid(trace: DecodedCallTrace) -> PunkBid:
raise NotImplementedError() raise NotImplementedError()
@ -34,7 +34,7 @@ class PunkAcceptBidClassifier(Classifier):
@staticmethod @staticmethod
@abstractmethod @abstractmethod
def get_accept_bid(trace: DecodedCallTrace) -> Punk_Accept_Bid: def get_accept_bid(trace: DecodedCallTrace) -> PunkAcceptBid:
raise NotImplementedError() raise NotImplementedError()

2
mev_inspect/schemas/punk_accept_bid.py
View File

@ -3,7 +3,7 @@ from typing import List
from pydantic import BaseModel from pydantic import BaseModel
class Punk_Accept_Bid(BaseModel): class PunkAcceptBid(BaseModel):
block_number: int block_number: int
transaction_hash: str transaction_hash: str
trace_address: List[int] trace_address: List[int]

3
mev_inspect/schemas/punk_bid.py
View File

@ -2,7 +2,8 @@ from typing import List
from pydantic import BaseModel from pydantic import BaseModel
class Punk_Bid(BaseModel):
class PunkBid(BaseModel):
block_number: int block_number: int
transaction_hash: str transaction_hash: str
trace_address: List[int] trace_address: List[int]