from rest_framework import serializers
from manage_notifications.models import IAmPrincipalNotificationSettings, NotificationCategoryChoices


class IAmPrincipalNotificationSettingsSerializer(serializers.ModelSerializer):
    notification_category_display = serializers.SerializerMethodField()

    def get_notification_category_display(self, obj):
        return obj.get_notification_category_display()

    class Meta:
        model = IAmPrincipalNotificationSettings
        fields = ['id', 'notification_category', 'notification_category_display', 'is_enabled']