Update models/entities.py

models/entities.py

@@ -1,31 +1,49 @@
-from pydantic import BaseModel, Field
-from typing import Optional, Dict
-
-class Note(BaseModel):
-    date: str = Field(..., example="2023-01-01T12:00:00Z")
-    type: str = Field(..., example="Progress Note")
-    text: str = Field(..., example="Patient reported improvement in symptoms")
-    author: Optional[str] = Field(None, example="Dr. Smith")
+from pydantic import BaseModel
+from typing import List, Optional
+from datetime import datetime
 
 class Condition(BaseModel):
-    id: str = Field(..., example="cond123")
-    code: str = Field(..., example="Hypertension")
-    status: str = Field(..., example="Active")
-    onset_date: str = Field(..., example="2020-05-15")
-    recorded_date: Optional[str] = Field(None, example="2020-05-16")
-    verification_status: Optional[str] = Field(None, example="Confirmed")
+    id: str
+    code: str
+    status: Optional[str] = ""
+    onset_date: Optional[str] = None
+    recorded_date: Optional[str] = None
+    verification_status: Optional[str] = ""
 
 class Medication(BaseModel):
-    id: str = Field(..., example="med123")
-    name: str = Field(..., example="Lisinopril 10mg")
-    status: str = Field(..., example="Active")
-    prescribed_date: str = Field(..., example="2021-02-10")
-    requester: Optional[str] = Field(None, example="Dr. Smith")
-    dosage: Optional[str] = Field(None, example="10mg daily")
+    id: str
+    name: str
+    status: str
+    prescribed_date: Optional[str] = None
+    requester: Optional[str] = ""
+    dosage: Optional[str] = ""
 
 class Encounter(BaseModel):
-    id: str = Field(..., example="enc123")
-    type: str = Field(..., example="Outpatient Visit")
-    status: str = Field(..., example="Finished")
-    period: Dict = Field(..., example={"start": "2023-01-01T10:00:00Z"})
-    service_provider: Optional[str] = Field(None, example="City Hospital")
+    id: str
+    type: str
+    status: str
+    period: dict
+    service_provider: Optional[str] = ""
+
+class Note(BaseModel):
+    date: str
+    type: str
+    text: str
+    context: Optional[str] = ""
+    author: Optional[str] = "System"
+
+class PatientCreate(BaseModel):
+    full_name: str
+    gender: str
+    date_of_birth: str
+    address: Optional[str] = ""
+    city: Optional[str] = ""
+    state: Optional[str] = ""
+    postal_code: Optional[str] = ""
+    country: Optional[str] = "US"
+    marital_status: Optional[str] = "Never Married"
+    language: Optional[str] = "en"
+    conditions: Optional[List[Condition]] = []
+    medications: Optional[List[Medication]] = []
+    encounters: Optional[List[Encounter]] = []
+    notes: Optional[List[Note]] = []