File size: 742 Bytes
a7ed1bd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | 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
|