felicity-lims/felicity/apps/notification/repository.py
2024-07-21 09:06:51 +02:00

33 lines
883 B
Python

from domain.notification.ports.repository import (
IActivityFeedRepository,
IActivityStreamRepository,
INotificationRepository,
)
from infrastructure.database.repository.base import BaseRepository
from infrastructure.database.notification.entities import (
ActivityFeed,
ActivityStream,
Notification,
)
class ActivityFeedRepository(BaseRepository[ActivityFeed], IActivityFeedRepository):
def __init__(self) -> None:
self.model = ActivityFeed
super().__init__()
class ActivityStreamRepository(
BaseRepository[ActivityStream], IActivityStreamRepository
):
def __init__(self) -> None:
self.model = ActivityStream
super().__init__()
class NotificationRepository(BaseRepository[Notification], INotificationRepository):
def __init__(self) -> None:
self.model = Notification
super().__init__()