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

Add field for transfer classifiers

Browse Source
This commit is contained in:
Luke Van Seters 2021-09-28 12:55:03 -04:00
parent 7645726708
commit f87016a561
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mev_inspect/schemas/classifier_specs.py
View File

@ -1,8 +1,15 @@
from typing import Dict, List, Optional from typing import Callable, Dict, List, Optional
from pydantic import BaseModel from pydantic import BaseModel
from .classified_traces import Classification, Protocol from .classified_traces import Classification, DecodedCallTrace, Protocol
from .transfers import Transfer
TransferClassifier = Callable[
[DecodedCallTrace, List[DecodedCallTrace], List[DecodedCallTrace]],
Transfer,
]
class ClassifierSpec(BaseModel): class ClassifierSpec(BaseModel):
@ -10,3 +17,4 @@ class ClassifierSpec(BaseModel):
protocol: Optional[Protocol] = None protocol: Optional[Protocol] = None
valid_contract_addresses: Optional[List[str]] = None valid_contract_addresses: Optional[List[str]] = None
classifications: Dict[str, Classification] = {} classifications: Dict[str, Classification] = {}
transfer_classifiers: Dict[str, TransferClassifier] = {}