from pydantic import BaseModel, Field
from typing import Optional

class CountryBase(BaseModel):
    name: str = Field(..., index=True, min_length=1, description="Country name")
    iso2: str = Field(..., min_length=2, max_length=2, description="ISO 3166-1 alpha-2 code")
    iso3: str = Field(..., min_length=3, max_length=3, description="ISO 3166-1 alpha-3 code") 
    capital: str = Field(..., min_length=1, description="Capital city")
    region: str = Field(..., min_length=1, description="Geographic region")
    subregion: str = Field(..., min_length=1, description="Geographic subregion")
    population: str = Field(..., min_length=1, description="Population count")
    latitude: str = Field(..., min_length=1, description="Latitude coordinates")
    longitude: str = Field(..., min_length=1, description="Longitude coordinates")
    flag: str = Field(..., min_length=1, description="Flag image URL")

class CountryCreate(CountryBase):
    class Config:
        schema_extra = {
            "example": {
                "name": "United States",
                "iso2": "US",
                "iso3": "USA",
                "capital": "Washington, D.C.",
                "region": "Americas",
                "subregion": "North America", 
                "population": "331002651",
                "latitude": "37.09024",
                "longitude": "-95.712891",
                "flag": "https://example.com/us-flag.png"
            }
        }

class Country(CountryBase):
    id: int = Field(..., description="Unique identifier")

    class Config:
        orm_mode = True