Skip to content

Commit 216d327

Browse files
committed
fix
1 parent 60e311d commit 216d327

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

src/routers/requests.py

+12-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
from fastapi import APIRouter, BackgroundTasks, Depends, HTTPException
22
from sqlmodel import Session, select
33
from typing import List
4-
from src.models import MedicineRequest, User
4+
from src.models import Medicine, MedicineRequest, User
55
from src.schemas import (
66
RequestCreate,
77
RequestRead,
@@ -21,6 +21,17 @@ def create_request(
2121
session: Session = Depends(get_session),
2222
current_user: User = Depends(get_current_user),
2323
):
24+
# Query the medicine using SQLModel's select
25+
statement = select(Medicine).where(Medicine.name == request.name)
26+
result = session.exec(statement)
27+
medicine = result.first()
28+
29+
if not medicine:
30+
raise HTTPException(status_code=404, detail="Medicine not found.")
31+
32+
if request.quantity > medicine.quantity:
33+
raise HTTPException(status_code=400, detail="Not enough quantity available.")
34+
2435
db_request = MedicineRequest(**request.dict())
2536
session.add(db_request)
2637
session.commit()

0 commit comments

Comments
 (0)