sarlink-portal-api/devices/serializers.py

from rest_framework import serializers
from .models import Device
from api.serializers import CustomReadOnlyUserSerializer
from billing.models import Payment  # Import the Payment model
from django.utils import timezone


class CreateDeviceSerializer(serializers.ModelSerializer):
    name = serializers.CharField(required=True)
    mac = serializers.CharField(required=True)

    class Meta:  # type: ignore
        model = Device
        fields = [
            "name",
            "mac",
            "blocked_by",
        ]
        depth = 2


class BlockDeviceSerializer(serializers.ModelSerializer):
    blocked = serializers.BooleanField(required=True)
    reason_for_blocking = serializers.CharField(required=True)
    blocked_by = serializers.ChoiceField(
        choices=[("ADMIN", "Admin"), ("PARENT", "Parent")], required=True
    )

    class Meta:  # type: ignore
        model = Device
        fields = ["blocked", "reason_for_blocking", "blocked_by"]


class DeviceSerializer(serializers.ModelSerializer):
    user = serializers.SerializerMethodField()

    def get_user(self, obj):
        user = obj.user
        if user:
            return {
                "id": user.id,
                "name": user.first_name + " " + user.last_name,
                "id_card": user.id_card,
                "mobile": user.mobile,
            }
        return None

    class Meta:  # type: ignore
        model = Device
        fields = "__all__"


class ReadOnlyDeviceSerializer(serializers.ModelSerializer):
    user = CustomReadOnlyUserSerializer(read_only=True)

    class Meta:  # type: ignore
        depth = 2
        model = Device
        fields = "__all__"