From d50138a9ff0a7ef7c9942b149913f68f9e405320 Mon Sep 17 00:00:00 2001
From: Backend IM Bot <bot@backend.im>
Date: Tue, 25 Mar 2025 13:44:16 -0500
Subject: [PATCH] Add Drug schema

---
 schemas/drug.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 schemas/drug.py

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