notice but without the t

notice/consumers.py

@@ -0,0 +1,68 @@
+from __future__ import annotations
+import json
+
+from channels.generic.websocket import WebsocketConsumer
+from django.contrib.auth.models import User
+
+from profiles.serializers import ProfileSerializer
+from profiles.models import ProfileModel
+from .models import NoticeModel
+
+
+class NoticeManager:
+    def __init__(self):
+        self._list: list[NoticeConsumer] = []
+
+    def add(self, consumer: NoticeConsumer):
+        self._list.append(consumer)
+
+        unsend_notices = NoticeModel.objects.filter(user=consumer.user)
+        for notice in unsend_notices:
+            self.notify_user(consumer.user, json_data=notice.data)
+            notice.delete()
+
+    def remove(self, consumer: NoticeConsumer):
+        self._list.remove(consumer)
+
+    def get_consumer_by_user(self, user: User):
+        for consumer in self._list:
+            if consumer.user == user:
+                return consumer
+
+    def notify_user(self, user: User, data: dict = None, json_data: str = None):
+        consumer = self.get_consumer_by_user(user)
+        data_str: str = json.dumps(data) if json_data is None else json_data
+        if consumer:
+            consumer.send(data_str)
+        else:
+            NoticeModel(user=user, data=data_str).save()
+
+    def notify_friend_request(self, user: User, friend: ProfileModel):
+        self.notify_user(user, {'type': 'friend_request', 'author': ProfileSerializer(friend).data})
+
+    def notify_friend_request_canceled(self, user: User, friend: ProfileModel):
+        self.notify_user(user, {'type': 'friend_request_canceled', 'author': ProfileSerializer(friend).data})
+
+    def notify_new_friend(self, user: User, friend: ProfileModel):
+        self.notify_user(user, {'type': 'new_friend', 'friend': ProfileSerializer(friend).data})
+
+    def notify_friend_removed(self, user: User, friend: ProfileModel):
+        self.notify_user(user, {'type': 'friend_removed', 'friend': ProfileSerializer(friend).data})
+
+
+notice_manager = NoticeManager()
+
+
+class NoticeConsumer(WebsocketConsumer):
+    def connect(self):
+        self.user: User = self.scope['user']
+        if not self.user.is_authenticated:
+            self.close()
+            return
+
+        self.accept()
+        notice_manager.add(self)
+
+    def disconnect(self, code):
+        notice_manager.remove(self)
+        super().disconnect(code)