import base64
from typing import Any

from pydantic import BaseModel, Json, field_validator


class KafkaRecord(BaseModel):
    value: Json[dict[str, Any]]

    @field_validator("value", mode="before")
    def _parse_event(cls, value: Any) -> Any:
        if isinstance(value, str | bytes):
            return base64.b64decode(value)
        return value


class KafkaEvent(BaseModel):
    records: dict[str, list[KafkaRecord]]