# -*- coding: utf-8 -*- # from django.utils.translation import ugettext_lazy as _ from rest_framework import serializers from common.drf.serializers import MethodSerializer from orgs.utils import get_org_by_id from orgs.mixins.serializers import OrgResourceModelSerializerMixin from users.models import User from tickets.models import Ticket from .meta import type_serializer_classes_mapping __all__ = [ 'TicketDisplaySerializer', 'TicketApplySerializer', 'TicketApproveSerializer', ] class TicketSerializer(OrgResourceModelSerializerMixin): type_display = serializers.ReadOnlyField(source='get_type_display', label=_('Type')) action_display = serializers.ReadOnlyField(source='get_action_display', label=_('Action')) status_display = serializers.ReadOnlyField(source='get_status_display', label=_('Status')) meta = MethodSerializer() class Meta: model = Ticket fields = [ 'id', 'title', 'type', 'type_display', 'meta', 'action', 'action_display', 'status', 'status_display', 'applicant', 'applicant_display', 'processor', 'processor_display', 'assignees', 'assignees_display', 'comment', 'date_created', 'date_updated', 'org_id', 'org_name', 'body' ] def get_meta_serializer(self): request = self.context['request'] default_serializer_class = serializers.Serializer if isinstance(self.instance, Ticket): _type = self.instance.type else: _type = request.query_params.get('type') if not _type: return default_serializer_class() action_serializer_classes_mapping = type_serializer_classes_mapping.get(_type) if not action_serializer_classes_mapping: return default_serializer_class() query_action = request.query_params.get('action') _action = query_action if query_action else self.context['view'].action serializer_class = action_serializer_classes_mapping.get(_action) if serializer_class: return serializer_class() serializer_class = action_serializer_classes_mapping.get('default') if serializer_class: return serializer_class() return default_serializer_class() class TicketDisplaySerializer(TicketSerializer): class Meta: model = Ticket fields = TicketSerializer.Meta.fields read_only_fields = fields class TicketApplySerializer(TicketSerializer): org_id = serializers.CharField( required=True, max_length=36, allow_blank=True, label=_("Organization") ) class Meta: model = Ticket fields = TicketSerializer.Meta.fields writeable_fields = [ 'id', 'title', 'type', 'meta', 'assignees', 'comment', 'org_id' ] read_only_fields = list(set(fields) - set(writeable_fields)) extra_kwargs = { 'type': {'required': True}, } def validate_type(self, tp): request_type = self.context['request'].query_params.get('type') if tp != request_type: error = _( 'The `type` in the submission data (`{}`) is different from the type ' 'in the request url (`{}`)'.format(tp, request_type) ) raise serializers.ValidationError(error) return tp @staticmethod def validate_org_id(org_id): org = get_org_by_id(org_id) if not org: error = _('The organization `{}` does not exist'.format(org_id)) raise serializers.ValidationError(error) return org_id def validate_assignees(self, assignees): org_id = self.initial_data.get('org_id') self.validate_org_id(org_id) org = get_org_by_id(org_id) admins = User.get_super_and_org_admins(org) valid_assignees = list(set(assignees) & set(admins)) if not valid_assignees: error = _('None of the assignees belong to Organization `{}` admins'.format(org.name)) raise serializers.ValidationError(error) return valid_assignees class TicketApproveSerializer(TicketSerializer): class Meta: model = Ticket fields = TicketSerializer.Meta.fields writeable_fields = ['meta'] read_only_fields = list(set(fields) - set(writeable_fields)) def validate_meta(self, meta): _meta = self.instance.meta if self.instance else {} _meta.update(meta) return _meta