Files
sarlink-portal-api/devices/serializers.py
i701 8657435fbf
All checks were successful
Build and Push Docker Images / Build and Push Docker Images (push) Successful in 5m59s
fix(payment): refine unpaid payment query to include status and expiration checks 🐛
2025-07-08 20:53:05 +05:00

76 lines
2.2 KiB
Python

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):
pending_payment_id = serializers.SerializerMethodField()
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
def get_pending_payment_id(self, obj):
# Query the last unpaid payment for the device
unpaid_payment = (
Payment.objects.filter(
devices=obj,
paid=False,
status="PENDING",
expires_at__gt=timezone.now(),
expires_at__isnull=True,
)
.order_by("-created_at")
.first()
)
return unpaid_payment.id if unpaid_payment else 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__"