diff --git a/backend/ticket/serializers.py b/backend/ticket/serializers.py index d28077c..7a82e83 100644 --- a/backend/ticket/serializers.py +++ b/backend/ticket/serializers.py @@ -30,6 +30,7 @@ class MessageForTicketSerializer(serializers.ModelSerializer): exclude = ('sender', 'ticket') class TicketSerializer(serializers.ModelSerializer): status = serializers.SerializerMethodField() + ticket_category = serializers.SerializerMethodField() messages = MessageSerializer(many=True, read_only=True) message = MessageForTicketSerializer(write_only=True) class Meta: @@ -51,8 +52,12 @@ class TicketSerializer(serializers.ModelSerializer): def get_status(self, obj): return obj.get_status_display() + def get_ticket_category(self, obj): + return obj.get_ticket_category_display() + class TicketListSerializer(serializers.ModelSerializer): status = serializers.SerializerMethodField() + ticket_category = serializers.SerializerMethodField() class Meta: model = Ticket exclude = ('customer', 'admin', ) @@ -60,3 +65,5 @@ class TicketListSerializer(serializers.ModelSerializer): def get_status(self, obj): return obj.get_status_display() + def get_ticket_category(self, obj): + return obj.get_ticket_category_display() \ No newline at end of file