69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
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)
|