diff --git a/schemas/state.py b/schemas/state.py new file mode 100644 index 0000000..3f6eb54 --- /dev/null +++ b/schemas/state.py @@ -0,0 +1,28 @@ +from pydantic import BaseModel, Field +from typing import Optional + +class StateBase(BaseModel): + name: str = Field(..., min_length=1, max_length=100) + country_code: str = Field(..., min_length=2, max_length=2) + +class StateCreate(StateBase): + class Config: + schema_extra = { + "example": { + "name": "California", + "country_code": "US" + } + } + +class State(StateBase): + id: int + + class Config: + orm_mode = True + schema_extra = { + "example": { + "id": 1, + "name": "California", + "country_code": "US" + } + } \ No newline at end of file