from datetime import datetime
from typing import List, Optional, Any
from uuid import UUID
from ninja import Schema
from pydantic import field_serializer, model_validator


class MarketOptionSchema(Schema):
    uuid: UUID
    text: str
    position: int

    @field_serializer('uuid')
    def serialize_uuid(self, value: UUID) -> str:
        return str(value)


class MarketListSchema(Schema):
    uuid: UUID
    title: str
    description: str
    type: str
    status: str
    end_date: datetime
    created_at: datetime
    options: List[MarketOptionSchema]
    winning_option: Optional[MarketOptionSchema] = None

    @field_serializer('uuid')
    def serialize_uuid(self, value: UUID) -> str:
        return str(value)


class ResolveMarketSchema(Schema):
    winning_option_uuid: str


class UserBetCreateSchema(Schema):
    option_uuid: str
    amount: int


class UserBetSchema(Schema):
    uuid: UUID
    amount: int
    created_at: datetime
    option: MarketOptionSchema
    market: Optional[MarketListSchema] = None

    @field_serializer('uuid')
    def serialize_uuid(self, value: UUID) -> str:
        return str(value)

    @model_validator(mode='before')
    @classmethod
    def resolve_market_from_option(cls, data: Any) -> Any:
        if hasattr(data, 'option') and hasattr(data.option, 'market'):
            data.market = data.option.market
        return data