104 lines
3.4 KiB
Python
104 lines
3.4 KiB
Python
from typing import List
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.database import get_db
|
|
from app.core.deps import get_current_user, require_role
|
|
from app.models.user import User
|
|
from app.schemas.batch import BatchCreate, BatchUpdate, BatchResponse, BatchDispense, BatchAddStock
|
|
from app.services.batch import BatchService
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/medicine/{medicine_id}", response_model=List[BatchResponse])
|
|
async def list_batches_by_medicine(
|
|
medicine_id: int,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(get_current_user)
|
|
):
|
|
service = BatchService(db)
|
|
batches = await service.get_batches_by_medicine(medicine_id)
|
|
return [BatchResponse.model_validate(b) for b in batches]
|
|
|
|
|
|
@router.get("/{batch_id}", response_model=BatchResponse)
|
|
async def get_batch(
|
|
batch_id: int,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(get_current_user)
|
|
):
|
|
service = BatchService(db)
|
|
batch = await service.get_batch(batch_id)
|
|
if not batch:
|
|
raise HTTPException(status_code=404, detail="批次不存在")
|
|
return BatchResponse.model_validate(batch)
|
|
|
|
|
|
@router.post("/medicine/{medicine_id}", response_model=BatchResponse)
|
|
async def create_batch(
|
|
medicine_id: int,
|
|
data: BatchCreate,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(require_role(["admin", "user"]))
|
|
):
|
|
service = BatchService(db)
|
|
batch = await service.create_batch(medicine_id, data)
|
|
return BatchResponse.model_validate(batch)
|
|
|
|
|
|
@router.put("/{batch_id}", response_model=BatchResponse)
|
|
async def update_batch(
|
|
batch_id: int,
|
|
data: BatchUpdate,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(require_role(["admin", "user"]))
|
|
):
|
|
service = BatchService(db)
|
|
batch = await service.update_batch(batch_id, data)
|
|
if not batch:
|
|
raise HTTPException(status_code=404, detail="批次不存在")
|
|
return BatchResponse.model_validate(batch)
|
|
|
|
|
|
@router.delete("/{batch_id}")
|
|
async def delete_batch(
|
|
batch_id: int,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(require_role(["admin"]))
|
|
):
|
|
service = BatchService(db)
|
|
success = await service.delete_batch(batch_id)
|
|
if not success:
|
|
raise HTTPException(status_code=404, detail="批次不存在")
|
|
return {"message": "删除成功"}
|
|
|
|
|
|
@router.post("/{batch_id}/dispense", response_model=BatchResponse)
|
|
async def dispense_batch(
|
|
batch_id: int,
|
|
data: BatchDispense,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(require_role(["admin", "user"]))
|
|
):
|
|
service = BatchService(db)
|
|
try:
|
|
batch = await service.dispense(batch_id, data.quantity, current_user.id)
|
|
return BatchResponse.model_validate(batch)
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
|
|
|
|
@router.post("/{batch_id}/add-stock", response_model=BatchResponse)
|
|
async def add_stock_batch(
|
|
batch_id: int,
|
|
data: BatchAddStock,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(require_role(["admin", "user"]))
|
|
):
|
|
service = BatchService(db)
|
|
try:
|
|
batch = await service.add_stock(batch_id, data.quantity, current_user.id)
|
|
return BatchResponse.model_validate(batch)
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=400, detail=str(e)) |