serializers.py

from rest_framework import serializers

from api.serializers import (
    MiniCountrySerializer,
    MiniEventSerializer,
    SurgeEventSerializer,
)
from deployments.serializers import MolnixTagSerializer
from lang.serializers import ModelSerializer

from .models import Subscription, SurgeAlert


class SurgeAlertSerializer(ModelSerializer):
    event = SurgeEventSerializer(required=False)
    country = MiniCountrySerializer(required=False)
    atype_display = serializers.CharField(source="get_atype_display", read_only=True)
    molnix_status_display = serializers.CharField(source="get_molnix_status_display", read_only=True)
    category_display = serializers.CharField(source="get_category_display", read_only=True)
    molnix_tags = MolnixTagSerializer(many=True, read_only=True)

    class Meta:
        model = SurgeAlert
        fields = (
            "operation",
            "country",
            "message",
            "deployment_needed",
            "is_private",
            "event",
            "created_at",
            "id",
            "atype",
            "atype_display",
            "category",
            "category_display",
            "molnix_id",
            "molnix_tags",
            "molnix_status",
            "opens",
            "closes",
            "start",
            "end",
            "molnix_status",
            "molnix_status_display",
        )


# class UnauthenticatedSurgeAlertSerializer(ModelSerializer):
#     event = MiniEventSerializer()
#     atype_display = serializers.CharField(source='get_atype_display', read_only=True)
#     category_display = serializers.CharField(source='get_category_display', read_only=True)
#
#     class Meta:
#         model = SurgeAlert
#         fields = (
#             'operation', 'deployment_needed', 'is_private', 'event', 'created_at', 'id',
#             'atype', 'atype_display', 'category', 'category_display', 'is_active',
#         )


class SubscriptionSerializer(ModelSerializer):
    country = MiniCountrySerializer()
    event = MiniEventSerializer()
    stype_display = serializers.CharField(source="get_stype_display", read_only=True)
    rtype_display = serializers.CharField(source="get_rtype_display", read_only=True)

    class Meta:
        model = Subscription
        fields = (
            "user",
            "country",
            "region",
            "event",
            "dtype",
            "lookup_id",
            "stype",
            "stype_display",
            "rtype",
            "rtype_display",
        )