2179e7a1c9
# What this PR does Fixes https://github.com/grafana/oncall/issues/2320 ## Checklist - [x] Unit, integration, and e2e (if applicable) tests updated - [x] Documentation added (or `pr:no public docs` PR label added if not required) - [x] `CHANGELOG.md` updated (or `pr:no changelog` PR label added if not required)
64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
from rest_framework import serializers
|
|
|
|
from apps.alerts.models import AlertGroup, ResolutionNote
|
|
from apps.api.serializers.user import FastUserSerializer
|
|
from apps.mobile_app.auth import MobileAppAuthTokenAuthentication
|
|
from common.api_helpers.custom_fields import OrganizationFilteredPrimaryKeyRelatedField
|
|
from common.api_helpers.exceptions import BadRequest
|
|
from common.api_helpers.mixins import EagerLoadingMixin
|
|
|
|
|
|
class ResolutionNoteSerializer(EagerLoadingMixin, serializers.ModelSerializer):
|
|
id = serializers.CharField(read_only=True, source="public_primary_key")
|
|
alert_group = OrganizationFilteredPrimaryKeyRelatedField(
|
|
filter_field="channel__organization",
|
|
queryset=AlertGroup.objects,
|
|
)
|
|
text = serializers.CharField(allow_null=False, source="message_text")
|
|
author = FastUserSerializer(read_only=True)
|
|
|
|
SELECT_RELATED = ["resolution_note_slack_message", "author"]
|
|
|
|
class Meta:
|
|
model = ResolutionNote
|
|
fields = [
|
|
"id",
|
|
"alert_group",
|
|
"source",
|
|
"author",
|
|
"created_at",
|
|
"text",
|
|
]
|
|
read_only_fields = [
|
|
"author",
|
|
"created_at",
|
|
"source",
|
|
]
|
|
|
|
def create(self, validated_data):
|
|
validated_data["author"] = self.context["request"].user
|
|
|
|
if isinstance(self.context["request"].successful_authenticator, MobileAppAuthTokenAuthentication):
|
|
source = ResolutionNote.Source.MOBILE_APP
|
|
else:
|
|
source = ResolutionNote.Source.WEB
|
|
validated_data["source"] = source
|
|
|
|
created_instance = super().create(validated_data)
|
|
return created_instance
|
|
|
|
def to_representation(self, instance):
|
|
result = super().to_representation(instance)
|
|
result["text"] = instance.text
|
|
result["source"] = {"id": instance.source, "display_name": instance.get_source_display()}
|
|
return result
|
|
|
|
|
|
class ResolutionNoteUpdateSerializer(ResolutionNoteSerializer):
|
|
alert_group = OrganizationFilteredPrimaryKeyRelatedField(read_only=True)
|
|
|
|
def update(self, instance, validated_data):
|
|
if instance.source != ResolutionNote.Source.WEB:
|
|
raise BadRequest(detail="Cannot update message with this source type")
|
|
|
|
return super().update(instance, validated_data)
|