Add authentification
This commit is contained in:
@@ -23,25 +23,41 @@ def get_shipments(
|
||||
)
|
||||
|
||||
@router.get('/{id}', response_model=models.ShipmentPublic)
|
||||
def get_shipment(id: int, session: Session = Depends(get_session)):
|
||||
def get_shipment(
|
||||
id: int,
|
||||
user: models.User = Depends(get_current_user),
|
||||
session: Session = Depends(get_session)
|
||||
):
|
||||
result = service.get_one(session, id)
|
||||
if result is None:
|
||||
raise HTTPException(status_code=404, detail=messages.notfound)
|
||||
return result
|
||||
|
||||
@router.post('/', response_model=models.ShipmentPublic)
|
||||
def create_shipment(shipment: models.ShipmentCreate, session: Session = Depends(get_session)):
|
||||
def create_shipment(
|
||||
shipment: models.ShipmentCreate,
|
||||
user: models.User = Depends(get_current_user),
|
||||
session: Session = Depends(get_session)
|
||||
):
|
||||
return service.create_one(session, shipment)
|
||||
|
||||
@router.put('/{id}', response_model=models.ShipmentPublic)
|
||||
def update_shipment(id: int, shipment: models.ShipmentUpdate, session: Session = Depends(get_session)):
|
||||
def update_shipment(
|
||||
id: int, shipment: models.ShipmentUpdate,
|
||||
user: models.User = Depends(get_current_user),
|
||||
session: Session = Depends(get_session)
|
||||
):
|
||||
result = service.update_one(session, id, shipment)
|
||||
if result is None:
|
||||
raise HTTPException(status_code=404, detail=messages.notfound)
|
||||
return result
|
||||
|
||||
@router.delete('/{id}', response_model=models.ShipmentPublic)
|
||||
def delete_shipment(id: int, session: Session = Depends(get_session)):
|
||||
def delete_shipment(
|
||||
id: int,
|
||||
user: models.User = Depends(get_current_user),
|
||||
session: Session = Depends(get_session)
|
||||
):
|
||||
result = service.delete_one(session, id)
|
||||
if result is None:
|
||||
raise HTTPException(status_code=404, detail=messages.notfound)
|
||||
|
||||
Reference in New Issue
Block a user