from pydantic import BaseModel, ConfigDict, Field

from location.types import IPvAnyAddress


class HelloResponse(BaseModel):
    status: str = "ok"


class LookupIPResponse(BaseModel):
    ip: str
    resolved: bool = Field(
        description="Indicates that provided IP address is successfully resolved."
    )
    country_short: str
    country_long: str
    region: str
    city: str
    latitude: float
    longitude: float

    model_config = ConfigDict(from_attributes=True)


class LookupIPBatchRequest(BaseModel):
    ips: list[IPvAnyAddress] = Field(max_length=10000, min_length=1)
    only_resolved: bool = Field(False, description="Return only resolved IP addresses")