from rest_framework import serializers
from apps.account.models import Notification
from apps.account.models import User




class NotificationSerializer(serializers.ModelSerializer):
    user_type = serializers.ChoiceField(choices=[('user', 'User'), ('merchant', 'Merchant')], default='user')

    class Meta:
        model = Notification
        fields = ['id', 'title', 'message', 'is_read', 'user_type', 'created_at', 'updated_at']



class NotificationSendSerializer(serializers.Serializer):
    title = serializers.CharField()
    body = serializers.CharField()
    data = serializers.DictField(required=False)
    account_id = serializers.CharField(required=True)
    user_type = serializers.CharField(required=True)