from typing import Any, Optional

from pydantic import BaseModel


class AbacusStateBaseModel(BaseModel):
    """Common ABACUS state fields."""

    abacus_state_id: int
    action_name: str
    action_status: str
    message: Optional[str] = None


class AbacusStateGetResponseItemModel(AbacusStateBaseModel):
    """Abacus state get response item model."""

    pass


class AbacusStatePutResponseModel(AbacusStateBaseModel):
    """Abacus state PUT response model."""

    parent_table_id: int
    parent_table_name: str