cbh/api/availability/views.py

from fastapi import Depends
from cbh.api.account.models import AccountModel
from cbh.api.availability.db_requests import update_availability_obj
from cbh.api.availability.schemas import (
    UpdateAvailabilityRequest,
)
from cbh.api.common.db_requests import get_obj_by_id
from cbh.core.security import PermissionDependency
from cbh.api.account.dto import AccountType
from cbh.core.wrappers import CbhResponseWrapper
from cbh.api.availability.models import AvailabilityModel
from cbh.api.availability import availability_router


@availability_router.get("")
async def get_own_availability(
    account: AccountModel = Depends(PermissionDependency([AccountType.COACH])),
) -> CbhResponseWrapper[AvailabilityModel]:

    availability = await get_obj_by_id(
        AvailabilityModel, None, additional_filter={"coach.id": account.id}
    )
    return CbhResponseWrapper(data=availability)


@availability_router.get("/{coachId}")
async def get_availability(
    coachId: str,
    _: AccountModel = Depends(
        PermissionDependency([AccountType.ADMIN, AccountType.USER])
    ),
) -> CbhResponseWrapper[AvailabilityModel]:
    """
    Get availability by coach ID.
    """
    availability = await get_obj_by_id(
        AvailabilityModel, None, additional_filter={"coach.id": coachId}
    )
    return CbhResponseWrapper(data=availability)


@availability_router.put("")
async def update_own_availability(
    availability: UpdateAvailabilityRequest,
    account: AccountModel = Depends(PermissionDependency([AccountType.COACH])),
) -> CbhResponseWrapper[AvailabilityModel]:
    """
    Update own availability.
    """
    availability = await update_availability_obj(availability, account.id)
    return CbhResponseWrapper(data=availability)


@availability_router.put("/{coachId}")
async def update_availability(
    coachId: str,
    availability: UpdateAvailabilityRequest,
    _: AccountModel = Depends(PermissionDependency([AccountType.ADMIN])),
) -> CbhResponseWrapper[AvailabilityModel]:
    """
    Update availability.
    """
    availability = await update_availability_obj(availability, coachId)
    return CbhResponseWrapper(data=availability)