notice but without the t

notice/admin.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

notice/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class NoticeConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'notice'

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)

notice/models.py

@@ -0,0 +1,7 @@
+from django.db.models import Model, ForeignKey, CharField, CASCADE
+from django.contrib.auth.models import User
+
+
+class NoticeModel(Model):
+    user = ForeignKey(User, on_delete=CASCADE)
+    data = CharField(max_length=100)

notice/routing.py

@@ -0,0 +1,7 @@
+from django.urls import re_path
+
+from .consumers import NoticeConsumer
+
+websocket_urlpatterns = [
+    re_path(r'ws/notice$', NoticeConsumer.as_asgi()),
+]

notice/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

notice/views.py

@@ -0,0 +1,3 @@
+from django.shortcuts import render
+
+# Create your views here.