Spaces:
Sleeping
Sleeping
File size: 2,567 Bytes
c073c9d 9797175 c073c9d 3afdd84 d7cc4ba 9797175 d7cc4ba 3afdd84 9797175 c073c9d 9797175 c073c9d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 |
from src.profile.models import UserDevices
from src.profile.models import Leave
from sqlmodel import select
from requests.api import delete
from src.notifications.schemas import RegisterDeviceRequest
from src.notifications.service import register_device
from fastapi import APIRouter, Depends
from src.auth.utils import get_current_user
from src.core.database import get_async_session
from sqlalchemy.ext.asyncio import AsyncSession
from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy import text
import uuid
router = APIRouter(prefix="/notifications", tags=["Notifications"])
@router.post("/register-device")
async def register_device_route(
body: RegisterDeviceRequest,
session: AsyncSession = Depends(get_async_session),
user=Depends(get_current_user),
):
device = await register_device(user, body, session)
return {"message": "Device registered", "device": str(device.id)}
@router.post("/{notif_id}/mark-read")
async def mark_notification_read(
notif_id: str,
session: AsyncSession = Depends(get_async_session),
user_id: str = Depends(get_current_user),
):
notif = await session.get(Leave, uuid.UUID(notif_id))
if not notif:
raise HTTPException(404, "Notification not found")
# Only owner or mentor should mark
if str(notif.user_id) != user_id and str(notif.mentor_id) != user_id:
raise HTTPException(403, "Unauthorized")
notif.is_read = True
await session.commit()
return {"message": "Marked as read"}
@router.post("/logout")
async def logout(
body: RegisterDeviceRequest,
session: AsyncSession = Depends(get_async_session),
user_id: str = Depends(get_current_user),
):
stmt = select(UserDevices).where(
UserDevices.user_id == user_id, UserDevices.device_token == body.device_token
)
result = await session.execute(stmt)
device = result.scalar_one_or_none()
if not device:
return {"message": "Device already removed"}
await session.delete(device)
await session.commit()
return {"message": "Logged out — device token removed"}
@router.post("/mark-all-read")
async def mark_all_read(
session: AsyncSession = Depends(get_async_session),
user_id: str = Depends(get_current_user),
):
await session.execute(
text(
"""
UPDATE leave
SET is_read = TRUE
WHERE user_id = :uid OR mentor_id = :uid
"""
),
{"uid": user_id},
)
await session.commit()
return {"message": "All notifications cleared"}
|