mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-06-06 20:15:17 +02:00
68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
"""Notifications for exceeding the page limit."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from uuid import UUID
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.notifications.persistence import Notification
|
|
from app.notifications.service.base import BaseNotificationHandler
|
|
from app.notifications.service.messages import page_limit as msg
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class PageLimitNotificationHandler(BaseNotificationHandler):
|
|
"""Notifications for exceeding the page limit."""
|
|
|
|
def __init__(self):
|
|
super().__init__("page_limit_exceeded")
|
|
|
|
async def notify_page_limit_exceeded(
|
|
self,
|
|
session: AsyncSession,
|
|
user_id: UUID,
|
|
document_name: str,
|
|
document_type: str,
|
|
search_space_id: int,
|
|
pages_used: int,
|
|
pages_limit: int,
|
|
pages_to_add: int,
|
|
) -> Notification:
|
|
"""Notify that a document was blocked by the page limit."""
|
|
operation_id = msg.operation_id(document_name, search_space_id)
|
|
title, message = msg.summary(
|
|
document_name, pages_used, pages_limit, pages_to_add
|
|
)
|
|
|
|
metadata = {
|
|
"operation_id": operation_id,
|
|
"document_name": document_name,
|
|
"document_type": document_type,
|
|
"pages_used": pages_used,
|
|
"pages_limit": pages_limit,
|
|
"pages_to_add": pages_to_add,
|
|
"status": "failed",
|
|
"error_type": "page_limit_exceeded",
|
|
# Where the inbox item links to.
|
|
"action_url": f"/dashboard/{search_space_id}/more-pages",
|
|
"action_label": "Upgrade Plan",
|
|
}
|
|
|
|
notification = Notification(
|
|
user_id=user_id,
|
|
search_space_id=search_space_id,
|
|
type=self.notification_type,
|
|
title=title,
|
|
message=message,
|
|
notification_metadata=metadata,
|
|
)
|
|
session.add(notification)
|
|
await session.commit()
|
|
await session.refresh(notification)
|
|
logger.info(
|
|
f"Created page_limit_exceeded notification {notification.id} for user {user_id}"
|
|
)
|
|
return notification
|