cc2b4eeae2
# What this PR does ## Which issue(s) this PR fixes ## Checklist - [ ] Tests updated - [ ] Documentation added - [ ] `CHANGELOG.md` updated
60 lines
2.4 KiB
Python
60 lines
2.4 KiB
Python
from rest_framework import mixins, status, viewsets
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.response import Response
|
|
|
|
from apps.alerts.models import AlertReceiveChannel
|
|
from apps.api.permissions import RBACPermission
|
|
from apps.api.serializers.alert_receive_channel import AlertReceiveChannelTemplatesSerializer
|
|
from apps.auth_token.auth import PluginAuthentication
|
|
from common.api_helpers.mixins import PublicPrimaryKeyMixin, TeamFilteringMixin
|
|
from common.insight_log import EntityEvent, write_resource_insight_log
|
|
from common.jinja_templater.apply_jinja_template import JinjaTemplateError
|
|
|
|
|
|
class AlertReceiveChannelTemplateView(
|
|
TeamFilteringMixin,
|
|
PublicPrimaryKeyMixin,
|
|
mixins.RetrieveModelMixin,
|
|
mixins.UpdateModelMixin,
|
|
viewsets.GenericViewSet,
|
|
):
|
|
authentication_classes = (PluginAuthentication,)
|
|
permission_classes = (IsAuthenticated, RBACPermission)
|
|
|
|
rbac_permissions = {
|
|
"metadata": [RBACPermission.Permissions.INTEGRATIONS_READ],
|
|
"list": [RBACPermission.Permissions.INTEGRATIONS_READ],
|
|
"retrieve": [RBACPermission.Permissions.INTEGRATIONS_READ],
|
|
"update": [RBACPermission.Permissions.INTEGRATIONS_WRITE],
|
|
"partial_update": [RBACPermission.Permissions.INTEGRATIONS_WRITE],
|
|
}
|
|
|
|
model = AlertReceiveChannel
|
|
serializer_class = AlertReceiveChannelTemplatesSerializer
|
|
|
|
def get_queryset(self, ignore_filtering_by_available_teams=False):
|
|
queryset = AlertReceiveChannel.objects.filter(
|
|
organization=self.request.auth.organization,
|
|
)
|
|
if not ignore_filtering_by_available_teams:
|
|
queryset = queryset.filter(*self.available_teams_lookup_args).distinct()
|
|
|
|
return queryset
|
|
|
|
def update(self, request, *args, **kwargs):
|
|
instance = self.get_object()
|
|
prev_state = instance.insight_logs_serialized
|
|
try:
|
|
result = super().update(request, *args, **kwargs)
|
|
except JinjaTemplateError as e:
|
|
return Response(e.fallback_message, status.HTTP_400_BAD_REQUEST)
|
|
instance = self.get_object()
|
|
new_state = instance.insight_logs_serialized
|
|
write_resource_insight_log(
|
|
instance=instance,
|
|
author=self.request.user,
|
|
event=EntityEvent.UPDATED,
|
|
prev_state=prev_state,
|
|
new_state=new_state,
|
|
)
|
|
return result
|