moonstream/crawlers/mooncrawl/mooncrawl/data.py

from dataclasses import dataclass
from datetime import datetime
from enum import Enum
from typing import List, Any, Dict

from pydantic import BaseModel, Field


class AvailableBlockchainType(Enum):
    ETHEREUM = "ethereum"
    POLYGON = "polygon"


class StatsUpdateRequest(BaseModel):
    dashboard_id: str
    timescales: List[str]
    token: str


@dataclass
class DateRange:
    start_time: datetime
    end_time: datetime
    include_start: bool
    include_end: bool


class PingResponse(BaseModel):
    """
    Schema for ping response
    """

    status: str


class VersionResponse(BaseModel):
    """
    Schema for responses on /version endpoint
    """

    version: str


class NowResponse(BaseModel):
    """
    Schema for responses on /now endpoint
    """

    epoch_time: float


class QueryDataUpdate(BaseModel):

    file_type: str
    query: str
    params: Dict[str, Any] = Field(default_factory=dict)