from pydantic import BaseModel, Field
from typing import Optional

class CountryBase(BaseModel):
    name: str = Field(..., min_length=1, description="Country name")
    code: str = Field(..., min_length=2, max_length=3, description="Country code")
    flag: Optional[str] = Field(None, description="Country flag URL")
    capital: Optional[str] = Field(None, description="Capital city")
    region: Optional[str] = Field(None, description="Geographic region")
    subregion: Optional[str] = Field(None, description="Geographic subregion")
    population: Optional[str] = Field(None, description="Population count")
    languages: Optional[str] = Field(None, description="Official languages")
    currencies: Optional[str] = Field(None, description="Official currencies")

class CountryCreate(CountryBase):
    class Config:
        schema_extra = {
            "example": {
                "name": "United States",
                "code": "USA",
                "flag": "https://example.com/us-flag.png",
                "capital": "Washington, D.C.",
                "region": "Americas",
                "subregion": "North America",
                "population": "331,002,651",
                "languages": "English",
                "currencies": "USD"
            }
        }

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

    class Config:
        orm_mode = True
        schema_extra = {
            "example": {
                "id": 1,
                "name": "United States",
                "code": "USA",
                "flag": "https://example.com/us-flag.png",
                "capital": "Washington, D.C.",
                "region": "Americas",
                "subregion": "North America",
                "population": "331,002,651",
                "languages": "English",
                "currencies": "USD"
            }
        }