initial-draft

resolvers/notifications.py

@@ -0,0 +1,84 @@
+from sqlalchemy import and_, desc, select, update
+
+from services.auth import login_required
+from services.db import local_session
+from services.schema import mutation, query
+from orm.notification import Notification
+
+
+@query.field("loadNotifications")
+@login_required
+async def load_notifications(_, info, params=None):
+    if params is None:
+        params = {}
+
+    user_id = info.context["user_id"]
+
+    limit = params.get("limit", 50)
+    offset = params.get("offset", 0)
+    q = (
+        select(Notification)
+        .order_by(desc(Notification.created_at))
+        .limit(limit)
+        .offset(offset)
+    )
+
+    notifications = []
+    with local_session() as session:
+        total_count = session.query(Notification).where(Notification.author == author_id).count()
+
+        total_unread_count = (
+            session.query(Notification)
+            .where(and_(Notification.author == author_id, Notification.seen == False))  # noqa: E712
+            .count()
+        )
+
+        for [notification] in session.execute(q):
+            notification.type = notification.type.name
+            notifications.append(notification)
+
+    return {
+        "notifications": notifications,
+        "totalCount": total_count,
+        "totalUnreadCount": total_unread_count,
+    }
+
+
+@mutation.field("markNotificationAsRead")
+@login_required
+async def mark_notification_as_read(_, info, notification_id: int):
+    author_id = info.context["author_id"]
+
+    with local_session() as session:
+        notification = (
+            session.query(Notification)
+            .where(and_(Notification.id == notification_id, Notification.user == user_id))
+            .one()
+        )
+        notification.seen = True
+        session.commit()
+
+    return {}
+
+
+@mutation.field("markAllNotificationsAsRead")
+@login_required
+async def mark_all_notifications_as_read(_, info):
+    auth: AuthCredentials = info.context["request"].auth
+    user_id = auth.user_id
+
+    statement = (
+        update(Notification)
+        .where(and_(Notification.user == user_id, Notification.seen == False))  # noqa: E712
+        .values(seen=True)
+    )
+
+    with local_session() as session:
+        try:
+            session.execute(statement)
+            session.commit()
+        except Exception as e:
+            session.rollback()
+            print(f"[mark_all_notifications_as_read] error: {str(e)}")
+
+    return {}