shihaam
9d8b5cd26f
Build and Push Docker Images / Build and Push Docker Images (push) Failing after 5s
87 lines
2.5 KiB
Python
87 lines
2.5 KiB
Python
from rest_framework import serializers
|
|
from .models import Device, normalize_mac
|
|
from api.serializers import CustomReadOnlyUserSerializer
|
|
from billing.models import Payment # Import the Payment model
|
|
|
|
|
|
class CreateDeviceSerializer(serializers.ModelSerializer):
|
|
name = serializers.CharField(required=True)
|
|
mac = serializers.CharField(required=True)
|
|
|
|
def validate_mac(self, value):
|
|
return normalize_mac(value)
|
|
|
|
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()
|
|
pending_payment_id = serializers.SerializerMethodField()
|
|
|
|
def get_pending_payment_id(self, obj):
|
|
unpaid_payment = (
|
|
obj.unpaid_payments[0]
|
|
if hasattr(obj, "unpaid_payments") and obj.unpaid_payments
|
|
else None
|
|
)
|
|
return unpaid_payment.id if unpaid_payment else None
|
|
|
|
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 AdminDeviceSerializer(serializers.ModelSerializer):
|
|
pending_payment_id = serializers.SerializerMethodField()
|
|
|
|
def get_pending_payment_id(self, obj):
|
|
unpaid_payment = (
|
|
obj.unpaid_payments[0]
|
|
if hasattr(obj, "unpaid_payments") and obj.unpaid_payments
|
|
else None
|
|
)
|
|
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__"
|