| from cbh.api.availability.schemas import UpdateAvailabilityRequest |
| from cbh.api.availability.models import AvailabilityModel |
| from cbh.api.common.db_requests import get_obj_by_id |
| from cbh.core.config import settings |
|
|
|
|
| async def update_availability_obj( |
| availability: UpdateAvailabilityRequest, coach_id: str |
| ) -> AvailabilityModel: |
| """ |
| Update availability object. |
| """ |
| availability_obj = await get_obj_by_id( |
| AvailabilityModel, None, additional_filter={"coach.id": coach_id} |
| ) |
| availability_obj.weeklySchedule = availability.weeklySchedule |
| await settings.DB_CLIENT.availabilities.update_one( |
| {"coach.id": coach_id}, |
| {"$set": availability_obj.to_mongo()}, |
| ) |
| return availability_obj |
|
|
