=move punk classifiers out of classifer.py

This commit is contained in:
Robert Miller 2021-11-25 16:48:48 -05:00
parent 976ac9ea77
commit bb3ace07a1
2 changed files with 15 additions and 29 deletions

View File

@ -1,13 +1,23 @@
from mev_inspect.schemas.traces import (
Protocol,
)
from mev_inspect.schemas.traces import Protocol, Classification
from mev_inspect.schemas.classifiers import (
ClassifierSpec,
PunkBidClassifier,
PunkBidAcceptanceClassifier,
Classifier,
)
class PunkBidAcceptanceClassifier(Classifier):
@staticmethod
def get_classification() -> Classification:
return Classification.punk_accept_bid
class PunkBidClassifier(Classifier):
@staticmethod
def get_classification() -> Classification:
return Classification.punk_bid
CRYPTO_PUNKS_SPEC = ClassifierSpec(
abi_name="cryptopunks",
protocol=Protocol.cryptopunks,

View File

@ -5,8 +5,6 @@ from pydantic import BaseModel
from .traces import Classification, DecodedCallTrace, Protocol
from .transfers import Transfer
from .punk_bid import PunkBid
from .punk_accept_bid import PunkBidAcceptance
class Classifier(ABC):
@ -16,28 +14,6 @@ class Classifier(ABC):
raise NotImplementedError()
class PunkBidClassifier(Classifier):
@staticmethod
def get_classification() -> Classification:
return Classification.punk_bid
@staticmethod
@abstractmethod
def get_bid(trace: DecodedCallTrace) -> PunkBid:
raise NotImplementedError()
class PunkBidAcceptanceClassifier(Classifier):
@staticmethod
def get_classification() -> Classification:
return Classification.punk_accept_bid
@staticmethod
@abstractmethod
def get_accept_bid(trace: DecodedCallTrace) -> PunkBidAcceptance:
raise NotImplementedError()
class TransferClassifier(Classifier):
@staticmethod
def get_classification() -> Classification: