mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
2.4 KiB
73 lines
2.4 KiB
4 years ago
|
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)
|