config.py

from enum import Enum
from typing import Dict, List, Optional

from pydantic import BaseModel, Field


class Status(str, Enum):
    ACTIVE = "ACTIVE"
    DISABLED = "DISABLED"
    DELETED = "DELETED"


class Contract(BaseModel):
    id: int
    created_at: int = Field(alias="createdAt")
    updated_at: int = Field(alias="updatedAt")
    project_id: int = Field(alias="projectId")
    tenant_id: int = Field(alias="tenantId")
    chain_uid: Optional[str] = Field(alias="chainUid", default=None)
    implementation: Optional[str] = None 
    address: Optional[str] = None
    name: str


class Schema(BaseModel):
    id: int
    created_at: int = Field(alias="createdAt")
    updated_at: int = Field(alias="updatedAt")
    status: Status
    name: str
    version: str
    jsonschema: Dict = Field(alias="schema", default_factory=dict)


class Configuration(BaseModel):
    id: int
    created_at: int = Field(alias="createdAt")
    updated_at: int = Field(alias="updatedAt")
    status: Status
    name: str
    source: str
    contract: Contract
    config_schema: Schema = Field(alias="schema")
    tags: List[str] = Field(default_factory=list)

    # Configuration based on published scheme for detector
    # TODO need to make data validation before use in a detector
    config: Dict = Field(default_factory=Dict)