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

Make Classifier a union

Browse Source
This commit is contained in:
Luke Van Seters 2021-10-06 15:14:24 -04:00
parent d57a2d021d
commit 86ee26dd1a
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mev_inspect/schemas/classifiers.py
View File

@ -1,4 +1,4 @@
from typing import Dict, List, Optional from typing import Dict, List, Optional, Union
from typing_extensions import Literal from typing_extensions import Literal
from pydantic import BaseModel from pydantic import BaseModel
@ -6,22 +6,21 @@ from pydantic import BaseModel
from .classified_traces import Classification, Protocol from .classified_traces import Classification, Protocol
class Classifier(BaseModel): class TransferClassifier(BaseModel):
classification: Classification
class TransferClassifier(Classifier):
classification: Literal[Classification.transfer] = Classification.transfer classification: Literal[Classification.transfer] = Classification.transfer
class SwapClassifier(Classifier): class SwapClassifier(BaseModel):
classification: Literal[Classification.swap] = Classification.swap classification: Literal[Classification.swap] = Classification.swap
class LiquidationClassifier(Classifier): class LiquidationClassifier(BaseModel):
classification: Literal[Classification.liquidate] = Classification.liquidate classification: Literal[Classification.liquidate] = Classification.liquidate
Classifier = Union[TransferClassifier, SwapClassifier, LiquidationClassifier]
class ClassifierSpec(BaseModel): class ClassifierSpec(BaseModel):
abi_name: str abi_name: str
protocol: Optional[Protocol] = None protocol: Optional[Protocol] = None