未知（继上次中断）

backend/app/schemas/ae.py

@@ -1,8 +1,17 @@
 import uuid
 from datetime import date, datetime
 from typing import Optional
+from enum import Enum
 
-from pydantic import BaseModel, ConfigDict
+from pydantic import BaseModel, ConfigDict, field_validator
+
+
+class AESeriousness(str, Enum):
+    I = "I"
+    II = "II"
+    III = "III"
+    IV = "IV"
+    V = "V"
 
 
 class AECreate(BaseModel):
@@ -11,8 +20,7 @@ class AECreate(BaseModel):
     visit_id: Optional[uuid.UUID] = None
     term: str
     onset_date: Optional[date] = None
-    seriousness: str
-    severity: str
+    seriousness: AESeriousness
     causality: Optional[str] = None
     description: Optional[str] = None
     outcome: Optional[str] = None
@@ -23,8 +31,7 @@ class AEUpdate(BaseModel):
     term: Optional[str] = None
     onset_date: Optional[date] = None
     resolution_date: Optional[date] = None
-    seriousness: Optional[str] = None
-    severity: Optional[str] = None
+    seriousness: Optional[AESeriousness] = None
     causality: Optional[str] = None
     action_taken: Optional[str] = None
     outcome: Optional[str] = None
@@ -42,8 +49,7 @@ class AERead(BaseModel):
     term: str
     onset_date: Optional[date]
     resolution_date: Optional[date]
-    seriousness: str
-    severity: str
+    seriousness: AESeriousness
     causality: Optional[str]
     action_taken: Optional[str]
     outcome: Optional[str]
@@ -57,3 +63,26 @@ class AERead(BaseModel):
     is_overdue: bool | None = None
 
     model_config = ConfigDict(from_attributes=True)
+
+    @field_validator("seriousness", mode="before")
+    @classmethod
+    def normalize_seriousness(cls, value):
+        if value is None:
+            return AESeriousness.I
+        normalized = str(value).strip().upper()
+        digit_map = {
+            "1": "I",
+            "2": "II",
+            "3": "III",
+            "4": "IV",
+            "5": "V",
+        }
+        legacy_map = {
+            "SERIOUS": "IV",
+            "NON_SERIOUS": "II",
+        }
+        if normalized in digit_map:
+            return digit_map[normalized]
+        if normalized in legacy_map:
+            return legacy_map[normalized]
+        return normalized