145 lines
5.3 KiB
Python
145 lines
5.3 KiB
Python
import logging
|
|
from typing import List
|
|
|
|
import strawberry
|
|
from sqlalchemy import and_, select
|
|
from sqlalchemy.exc import SQLAlchemyError
|
|
from sqlalchemy.orm import aliased
|
|
from strawberry.schema.config import StrawberryConfig
|
|
from strawberry_sqlalchemy_mapper import StrawberrySQLAlchemyMapper
|
|
|
|
from orm.author import Author
|
|
from orm.notification import Notification as NotificationMessage
|
|
from orm.notification import NotificationSeen
|
|
from services.auth import LoginRequiredMiddleware
|
|
from services.db import local_session
|
|
|
|
strawberry_sqlalchemy_mapper = StrawberrySQLAlchemyMapper()
|
|
|
|
# Инициализация логгера
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@strawberry_sqlalchemy_mapper.type(NotificationMessage)
|
|
class Notification:
|
|
id: int
|
|
action: str # create update delete join follow etc.
|
|
entity: str # REACTION SHOUT
|
|
created_at: int
|
|
payload: str # JSON data
|
|
seen: List[int]
|
|
|
|
|
|
@strawberry.type
|
|
class NotificationSeenResult:
|
|
error: str | None
|
|
|
|
|
|
@strawberry.type
|
|
class NotificationsResult:
|
|
notifications: List[Notification]
|
|
unread: int
|
|
total: int
|
|
|
|
|
|
def get_notifications(author_id: int, session, after: int, limit: int = 9999, offset: int = 0) -> List[Notification]:
|
|
NotificationSeenAlias = aliased(NotificationSeen)
|
|
query = (
|
|
select(NotificationMessage, NotificationSeenAlias.viewer.label("seen"))
|
|
.outerjoin(
|
|
NotificationSeen,
|
|
and_(NotificationSeen.viewer == author_id, NotificationSeen.notification == NotificationMessage.id),
|
|
)
|
|
.filter(NotificationMessage.created_at > after)
|
|
.group_by(NotificationSeen.notification)
|
|
)
|
|
if limit:
|
|
query = query.limit(limit)
|
|
if offset:
|
|
query = query.offset(offset)
|
|
|
|
notifications = []
|
|
for n, seen in session.execute(query):
|
|
ntf = Notification(
|
|
id=n.id,
|
|
payload=n.payload,
|
|
entity=n.entity,
|
|
action=n.action,
|
|
created_at=n.created_at,
|
|
seen=seen,
|
|
)
|
|
if ntf:
|
|
notifications.append(ntf)
|
|
return notifications
|
|
|
|
|
|
@strawberry.type
|
|
class Query:
|
|
@strawberry.field
|
|
async def load_notifications(self, info, limit: int = 50, offset: int = 0) -> NotificationsResult:
|
|
author_id = info.context.get("author_id")
|
|
with local_session() as session:
|
|
try:
|
|
if author_id:
|
|
author = session.query(Author).filter(Author.id == author_id).first()
|
|
after = author.last_seen
|
|
notifications = get_notifications(author_id, session, after, limit, offset)
|
|
if notifications and len(notifications) > 0:
|
|
nr = NotificationsResult(
|
|
notifications=notifications,
|
|
unread=sum(1 for n in notifications if author_id in n.seen),
|
|
total=session.query(NotificationMessage).count(),
|
|
)
|
|
return nr
|
|
except Exception as ex:
|
|
import traceback
|
|
|
|
traceback.print_exc()
|
|
logger.error(f"[load_notifications] Ошибка при выполнении запроса к базе данных: {ex}")
|
|
return NotificationsResult(notifications=[], total=0, unread=0)
|
|
|
|
|
|
@strawberry.type
|
|
class Mutation:
|
|
@strawberry.mutation
|
|
async def mark_notification_as_read(self, info, notification_id: int) -> NotificationSeenResult:
|
|
author_id = info.context.get("author_id")
|
|
if author_id:
|
|
with local_session() as session:
|
|
try:
|
|
ns = NotificationSeen(notification=notification_id, viewer=author_id)
|
|
session.add(ns)
|
|
session.commit()
|
|
except SQLAlchemyError as e:
|
|
session.rollback()
|
|
logger.error(
|
|
f"[mark_notification_as_read] Ошибка при обновлении статуса прочтения уведомления: {str(e)}"
|
|
)
|
|
return NotificationSeenResult(error="cant mark as read")
|
|
return NotificationSeenResult(error=None)
|
|
|
|
@strawberry.mutation
|
|
async def mark_all_notifications_as_read(self, info) -> NotificationSeenResult:
|
|
author_id = info.context.get("author_id")
|
|
if author_id:
|
|
with local_session() as session:
|
|
try:
|
|
nslist = get_notifications(author_id, session)
|
|
for n in nslist:
|
|
if author_id not in n.seen:
|
|
ns = NotificationSeen(viewer=author_id, notification=n.id)
|
|
session.add(ns)
|
|
session.commit()
|
|
except SQLAlchemyError as e:
|
|
session.rollback()
|
|
logger.error(
|
|
f"[mark_all_notifications_as_read] Ошибка обновления статуса прочтения всех уведомлений: {e}"
|
|
)
|
|
return NotificationSeenResult(error="cant mark as read")
|
|
return NotificationSeenResult(error=None)
|
|
|
|
|
|
schema = strawberry.Schema(
|
|
query=Query, mutation=Mutation, config=StrawberryConfig(auto_camel_case=False), extensions=[LoginRequiredMiddleware]
|
|
)
|