diff --git a/schemas/country.py b/schemas/country.py new file mode 100644 index 0000000..9a76d61 --- /dev/null +++ b/schemas/country.py @@ -0,0 +1,28 @@ +from pydantic import BaseModel, Field +from typing import Optional + +class CountryBase(BaseModel): + name: str = Field(..., min_length=1, max_length=100) + code: str = Field(..., min_length=2, max_length=2) + +class CountryCreate(CountryBase): + class Config: + schema_extra = { + "example": { + "name": "United States", + "code": "US" + } + } + +class Country(CountryBase): + id: int + + class Config: + orm_mode = True + schema_extra = { + "example": { + "id": 1, + "name": "United States", + "code": "US" + } + } \ No newline at end of file