mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-09-07 10:20:57 +00:00
feat: 站内信 (#6183)
* 添加站内信 * s * s * 添加接口 * fix * fix * 重构了一些 * 完成 * 完善 * s * s * s * s * s * s * 测试ok * 替换业务中发送消息的方式 * 修改 * s * 去掉 update 兼容 create * 添加 unread total 接口 * 调整json字段 Co-authored-by: xinwen <coderWen@126.com>
This commit is contained in:
72
apps/notifications/api/notifications.py
Normal file
72
apps/notifications/api/notifications.py
Normal file
@@ -0,0 +1,72 @@
|
||||
from django.http import Http404
|
||||
from rest_framework.mixins import ListModelMixin, UpdateModelMixin
|
||||
from rest_framework.views import APIView
|
||||
from rest_framework.response import Response
|
||||
from rest_framework import status
|
||||
|
||||
from common.drf.api import JmsGenericViewSet
|
||||
from notifications.notifications import system_msgs
|
||||
from notifications.models import SystemMsgSubscription
|
||||
from notifications.backends import BACKEND
|
||||
from notifications.serializers import (
|
||||
SystemMsgSubscriptionSerializer, SystemMsgSubscriptionByCategorySerializer
|
||||
)
|
||||
|
||||
__all__ = ('BackendListView', 'SystemMsgSubscriptionViewSet')
|
||||
|
||||
|
||||
class BackendListView(APIView):
|
||||
def get(self, request):
|
||||
data = [
|
||||
{
|
||||
'name': backend,
|
||||
'name_display': backend.label
|
||||
}
|
||||
for backend in BACKEND
|
||||
if backend.is_enable
|
||||
]
|
||||
return Response(data=data)
|
||||
|
||||
|
||||
class SystemMsgSubscriptionViewSet(ListModelMixin,
|
||||
UpdateModelMixin,
|
||||
JmsGenericViewSet):
|
||||
lookup_field = 'message_type'
|
||||
queryset = SystemMsgSubscription.objects.all()
|
||||
serializer_classes = {
|
||||
'list': SystemMsgSubscriptionByCategorySerializer,
|
||||
'update': SystemMsgSubscriptionSerializer,
|
||||
'partial_update': SystemMsgSubscriptionSerializer
|
||||
}
|
||||
|
||||
def list(self, request, *args, **kwargs):
|
||||
data = []
|
||||
category_children_mapper = {}
|
||||
|
||||
subscriptions = self.get_queryset()
|
||||
msgtype_sub_mapper = {}
|
||||
for sub in subscriptions:
|
||||
msgtype_sub_mapper[sub.message_type] = sub
|
||||
|
||||
for msg in system_msgs:
|
||||
message_type = msg['message_type']
|
||||
message_type_label = msg['message_type_label']
|
||||
category = msg['category']
|
||||
category_label = msg['category_label']
|
||||
|
||||
if category not in category_children_mapper:
|
||||
children = []
|
||||
|
||||
data.append({
|
||||
'category': category,
|
||||
'category_label': category_label,
|
||||
'children': children
|
||||
})
|
||||
category_children_mapper[category] = children
|
||||
|
||||
sub = msgtype_sub_mapper[message_type]
|
||||
sub.message_type_label = message_type_label
|
||||
category_children_mapper[category].append(sub)
|
||||
|
||||
serializer = self.get_serializer(data, many=True)
|
||||
return Response(data=serializer.data)
|
Reference in New Issue
Block a user