diff --git a/devices/serializers.py b/devices/serializers.py index 5ef6471..c9f3631 100644 --- a/devices/serializers.py +++ b/devices/serializers.py @@ -2,7 +2,6 @@ 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): @@ -33,6 +32,15 @@ class BlockDeviceSerializer(serializers.ModelSerializer): class DeviceSerializer(serializers.ModelSerializer): user = serializers.SerializerMethodField() + pending_payment_id = serializers.SerializerMethodField() + + def get_pending_payment_id(self, obj): + unpaid_payment = ( + Payment.objects.filter(devices=obj, paid=False) + .order_by("-created_at") + .first() + ) + return unpaid_payment.id if unpaid_payment else None def get_user(self, obj): user = obj.user