31 lines
885 B
Python
31 lines
885 B
Python
from pydantic import BaseModel, Field
|
|
|
|
# Base schema
|
|
class PenBase(BaseModel):
|
|
name: str = Field(..., description="Name of the pen")
|
|
color: str = Field(..., description="Color of the pen")
|
|
brand: str = Field(..., description="Brand of the pen")
|
|
price: int = Field(..., gt=0, description="Price of the pen")
|
|
stock: int = Field(..., ge=0, description="Stock quantity of the pen")
|
|
|
|
class Config:
|
|
schema_extra = {
|
|
"example": {
|
|
"name": "Classic Pen",
|
|
"color": "Blue",
|
|
"brand": "PenBrand",
|
|
"price": 5,
|
|
"stock": 100
|
|
}
|
|
}
|
|
|
|
# Schema for creating a new Pen
|
|
class PenCreate(PenBase):
|
|
pass
|
|
|
|
# Schema for Pen responses
|
|
class Pen(PenBase):
|
|
id: int = Field(..., description="Unique identifier for the pen")
|
|
|
|
class Config:
|
|
orm_mode = True |