立项配置页初步优化

backend/app/crud/ae.py

@@ -64,12 +64,14 @@ async def create_ae(
         term=ae_in.term,
         onset_date=ae_in.onset_date,
         resolution_date=None,
-        seriousness=str(ae_in.seriousness),
-        severity=str(ae_in.seriousness),
+        seriousness=ae_in.seriousness.value,
+        severity=ae_in.seriousness.value,
         causality=ae_in.causality,
         action_taken=None,
         outcome=ae_in.outcome,
         reported_to_sponsor=False,
+        is_sae=bool(ae_in.is_sae or ae_in.is_susar),
+        is_susar=bool(ae_in.is_susar),
         report_due_date=due_date,
         status="NEW",
         description=ae_in.description,
@@ -125,8 +127,14 @@ async def list_ae(
 
 async def update_ae(db: AsyncSession, study_id: uuid.UUID, ae: AdverseEvent, ae_in: AEUpdate) -> AdverseEvent:
     update_data = ae_in.model_dump(exclude_unset=True)
+    if "is_susar" in update_data and update_data["is_susar"]:
+        update_data["is_sae"] = True
+    if "is_sae" in update_data and update_data["is_sae"] is False:
+        update_data["is_susar"] = False
     if "seriousness" in update_data:
-        update_data["seriousness"] = str(update_data["seriousness"])
+        severity_value = update_data["seriousness"].value if hasattr(update_data["seriousness"], "value") else str(update_data["seriousness"])
+        update_data["seriousness"] = severity_value
+        update_data["severity"] = severity_value
     if "subject_id" in update_data:
         await _validate_site_subject(db, study_id, None, update_data["subject_id"])
     if "onset_date" in update_data or "seriousness" in update_data: