Update api_server.py

api_server.py

@@ -124,4 +124,127 @@ def predict_completion(request: PredictionRequest):
 if __name__ == "__main__":
     import uvicorn
     # host 0.0.0.0 允許外部訪問，port 8000 與前端設定一致
-    uvicorn.run("api_server:app", host="0.0.0.0", port=8000, reload=True)
+    uvicorn.run("api_server:app", host="0.0.0.0", port=8000, reload=True)
+
+# =================================================================
+# 4. 資料庫設定（SQLite + SQLAlchemy）
+# =================================================================
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker, Session
+from models import Base, Patient, Nurse, Record
+
+DATABASE_URL = "sqlite:///./nursing.db"
+
+engine = create_engine(
+    DATABASE_URL, connect_args={"check_same_thread": False}
+)
+
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+# 建立資料表
+Base.metadata.create_all(bind=engine)
+
+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+# =================================================================
+# 5. Schemas
+# =================================================================
+from pydantic import BaseModel
+
+# 病患
+class PatientCreate(BaseModel):
+    name: str
+    birth: str | None = None
+    gender: str | None = None
+    phone: str | None = None
+    email: str | None = None
+
+    emg_name: str | None = None
+    emg_phone: str | None = None
+    emg_relation: str | None = None
+
+    room: str
+    department: str
+    doctor: str | None = None
+    diagnosis: str
+    risk: str
+    admit_date: str
+
+
+# 護士
+class NurseCreate(BaseModel):
+    name: str
+    staff_id: str
+    department: str | None = None
+    position: str | None = None
+    phone: str | None = None
+    email: str | None = None
+
+
+# 護理紀錄
+class RecordCreate(BaseModel):
+    patient_id: int
+    nurse_id: int
+    content: str
+    created_at: str
+
+# =================================================================
+# 6. 病患 API
+# =================================================================
+@app.post("/patients")
+def create_patient(data: PatientCreate, db: Session = next(get_db())):
+    patient = Patient(**data.dict())
+    db.add(patient)
+    db.commit()
+    db.refresh(patient)
+    return patient
+
+
+@app.get("/patients")
+def list_patients(db: Session = next(get_db())):
+    return db.query(Patient).all()
+
+
+@app.get("/patients/{patient_id}")
+def get_patient(patient_id: int, db: Session = next(get_db())):
+    patient = db.query(Patient).filter(Patient.id == patient_id).first()
+    if not patient:
+        raise HTTPException(status_code=404, detail="找不到病患")
+    return patient
+
+# =================================================================
+# 7. 護士 API
+# =================================================================
+@app.post("/nurses")
+def create_nurse(data: NurseCreate, db: Session = next(get_db())):
+    nurse = Nurse(**data.dict())
+    db.add(nurse)
+    db.commit()
+    db.refresh(nurse)
+    return nurse
+
+
+@app.get("/nurses")
+def list_nurses(db: Session = next(get_db())):
+    return db.query(Nurse).all()
+
+# =================================================================
+# 8. 護理紀錄 API
+# =================================================================
+@app.post("/records")
+def create_record(data: RecordCreate, db: Session = next(get_db())):
+    record = Record(**data.dict())
+    db.add(record)
+    db.commit()
+    db.refresh(record)
+    return record
+
+
+@app.get("/records/{patient_id}")
+def list_records(patient_id: int, db: Session = next(get_db())):
+    return db.query(Record).filter(Record.patient_id == patient_id).all()