Add BannerNotification mutations

api/graphql/banner_notification/filtersets.py

@@ -7,8 +7,8 @@
 
 class BannerNotificationOrderingFilter(CustomOrderingFilter):
     @staticmethod
-    def order_by_type(qs: BannerNotificationQuerySet, desc: bool) -> BannerNotificationQuerySet:
-        return qs.order_by_type(desc)
+    def order_by_level(qs: BannerNotificationQuerySet, desc: bool) -> BannerNotificationQuerySet:
+        return qs.order_by_level(desc)
 
     @staticmethod
     def order_by_target(qs: BannerNotificationQuerySet, desc: bool) -> BannerNotificationQuerySet:
@@ -29,7 +29,7 @@ class BannerNotificationFilterSet(django_filters.FilterSet):
             ("active_until", "ends"),
         ),
         custom_fields=(
-            "type",
+            "level",
             "state",
             "target",
         ),

api/graphql/banner_notification/mutations.py

@@ -0,0 +1,34 @@
+from api.graphql.base_mutations import CreateAuthMutation, DeleteAuthMutation, UpdateAuthMutation
+from common.models import BannerNotification
+
+from .permissions import BannerNotificationPermission
+from .serializers import BannerNotificationSerializer
+
+__all__ = [
+    "BannerNotificationCreateMutation",
+    "BannerNotificationUpdateMutation",
+    "BannerNotificationDeleteMutation",
+]
+
+
+class BannerNotificationCreateMutation(CreateAuthMutation):
+    permission_classes = (BannerNotificationPermission,)
+
+    class Meta:
+        serializer_class = BannerNotificationSerializer
+
+
+class BannerNotificationUpdateMutation(UpdateAuthMutation):
+    permission_classes = (BannerNotificationPermission,)
+
+    class Meta:
+        serializer_class = BannerNotificationSerializer
+        lookup_field = "pk"
+
+
+class BannerNotificationDeleteMutation(DeleteAuthMutation):
+    permission_classes = (BannerNotificationPermission,)
+
+    class Meta:
+        model = BannerNotification
+        lookup_field = "pk"

api/graphql/banner_notification/permissions.py

@@ -1,8 +1,16 @@
+from typing import Any
+
 from graphene_permissions.permissions import BasePermission
-from graphql import GraphQLResolveInfo
+
+from common.typing import GQLInfo
+from permissions.helpers import can_manage_banner_notifications
 
 
 class BannerNotificationPermission(BasePermission):
     @classmethod
-    def has_permission(cls, info: GraphQLResolveInfo) -> bool:
+    def has_permission(cls, info: GQLInfo) -> bool:
         return True
+
+    @classmethod
+    def has_mutation_permission(cls, root: Any, info: GQLInfo, input: dict) -> bool:
+        return can_manage_banner_notifications(info.context.user)

api/graphql/banner_notification/serializers.py

@@ -0,0 +1,57 @@
+from collections import defaultdict
+from datetime import datetime
+from types import NoneType
+from typing import Any, Optional
+
+from rest_framework import serializers
+
+from common.models import BannerNotification
+from common.serializers import TranslatedModelSerializer
+
+__all__ = [
+    "BannerNotificationSerializer",
+]
+
+
+class BannerNotificationSerializer(TranslatedModelSerializer):
+    class Meta:
+        model = BannerNotification
+        fields = [
+            "pk",
+            "name",
+            "message",
+            "draft",
+            "level",
+            "target",
+            "active_from",
+            "active_until",
+        ]
+
+    def validate(self, attrs: dict[str, Any]) -> dict[str, Any]:
+        errors: dict[str, list[str]] = defaultdict(list)
+
+        active_from: Optional[datetime] = self.get_or_default("active_from", attrs)
+        active_until: Optional[datetime] = self.get_or_default("active_until", attrs)
+        draft: bool = self.get_or_default("draft", attrs)
+        message: str = self.get_or_default("message", attrs)
+
+        if not isinstance(active_from, type(active_until)) or not isinstance(active_from, (datetime, NoneType)):
+            msg = "Both 'active_from' and 'active_until' must be either set or null."
+            errors["active_until"].append(msg)
+            errors["active_from"].append(msg)
+
+        if active_from and active_until and active_from >= active_until:
+            errors["active_from"].append("'active_from' must be before 'active_until'.")
+
+        if not draft:
+            if not active_from:
+                errors["active_from"].append("Non-draft notifications must set 'active_from'")
+            if not active_until:
+                errors["active_until"].append("Non-draft notifications must set 'active_until'")
+            if not message:
+                errors["message"].append("Non-draft notifications must have a message.")
+
+        if errors:
+            raise serializers.ValidationError(errors)
+
+        return attrs

api/graphql/banner_notification/types.py

@@ -26,7 +26,7 @@ class Meta:
             "message_en",
             "message_sv",
             "draft",
-            "type",
+            "level",
             "target",
             "active_from",
             "active_until",