stash

core/notifications/api_v2.py

@@ -5,7 +5,7 @@ from rest_framework.decorators import api_view, permission_classes
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.response import Response
 
-from notifications.models import MessageTemplate
+from notifications.models import MessageTemplate, UserNotificationChannel
 from rest_framework import serializers
 
 from notifications.templates import TEMPLATE_VARS
@@ -17,11 +17,22 @@ class MessageTemplateSerializer(serializers.ModelSerializer):
         fields = '__all__'
 
 
+class UserNotificationChannelSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = UserNotificationChannel
+        fields = '__all__'
+
+
 class MessageTemplateViewSet(viewsets.ModelViewSet):
     serializer_class = MessageTemplateSerializer
     queryset = MessageTemplate.objects.all()
 
 
+class UserNotificationChannelViewSet(viewsets.ModelViewSet):
+    serializer_class = UserNotificationChannelSerializer
+    queryset = UserNotificationChannel.objects.all()
+
+
 @api_view(['GET'])
 @permission_classes([IsAuthenticated])
 @permission_required('tickets.add_issuethread_manual', raise_exception=True)  # TDOO: change this permission
@@ -31,7 +42,7 @@ def get_template_vars(self):
 
 router = routers.SimpleRouter()
 router.register(r'message_templates', MessageTemplateViewSet)
-
+router.register(r'user_notification_channels', UserNotificationChannelViewSet)
 urlpatterns = ([
                    re_path('message_template_variables', get_template_vars),
                ] + router.urls)