33 lines
1.3 KiB
Python
33 lines
1.3 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from fastapi.security import OAuth2PasswordBearer
|
|
from services.shelter_service import ShelterService
|
|
from models.schemas import CreateShelter, UpdateShelter
|
|
from config.database import get_db
|
|
|
|
router = APIRouter()
|
|
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/api/v1/auth/token")
|
|
|
|
@router.post("/", status_code=201)
|
|
async def create_shelter(shelter: CreateShelter, token: str = Depends(oauth2_scheme)):
|
|
async with get_db() as db:
|
|
return await ShelterService.create_shelter(shelter, db, token)
|
|
|
|
@router.get("/{shelter_id}")
|
|
async def get_shelter(shelter_id: int):
|
|
async with get_db() as db:
|
|
return await ShelterService.get_shelter(shelter_id, db)
|
|
|
|
@router.get("/")
|
|
async def get_all_shelters():
|
|
async with get_db() as db:
|
|
return await ShelterService.get_all_shelters(db)
|
|
|
|
@router.put("/{shelter_id}")
|
|
async def update_shelter(shelter_id: int, shelter: UpdateShelter, token: str = Depends(oauth2_scheme)):
|
|
async with get_db() as db:
|
|
return await ShelterService.update_shelter(shelter_id, shelter, db, token)
|
|
|
|
@router.delete("/{shelter_id}")
|
|
async def delete_shelter(shelter_id: int, token: str = Depends(oauth2_scheme)):
|
|
async with get_db() as db:
|
|
return await ShelterService.delete_shelter(shelter_id, db, token) |