diff --git a/schemas/drug.py b/schemas/drug.py new file mode 100644 index 0000000..4342632 --- /dev/null +++ b/schemas/drug.py @@ -0,0 +1,29 @@ +from pydantic import BaseModel, Field +from typing import List + +class DrugBase(BaseModel): + name: str = Field(..., description="Name of the drug") + description: str | None = Field(None, description="Description of the drug") + drug_types: List[str] | None = Field(None, description="Types of the drug") + +class DrugCreate(DrugBase): + name: str = Field(..., description="Name of the drug") + description: str | None = Field(None, description="Description of the drug") + drug_types: List[str] | None = Field(None, description="Types of the drug") + + class Config: + schema_extra = { + "example": { + "name": "Paracetamol", + "description": "Paracetamol is a commonly used over-the-counter medication to treat pain and fever.", + "drug_types": ["Analgesic", "Antipyretic"] + } + } + +class DrugResponse(DrugBase): + name: str + description: str | None + drug_types: List[str] | None + + class Config: + orm_mode = True \ No newline at end of file