40 lines
1.8 KiB
Python
40 lines
1.8 KiB
Python
|
|
from rest_framework import serializers
|
|
from .models import Device, Sistema, sistemas_por_cliente
|
|
from Clientes.models import Clientes
|
|
|
|
class SistemaPrimaryKeyRelatedField(serializers.PrimaryKeyRelatedField):
|
|
def to_internal_value(self,data):
|
|
try:
|
|
print(Sistema.objects.get(nombre_sistema=data))
|
|
return Sistema.objects.get(nombre_sistema=data)
|
|
except Sistema.DoesNotExist:
|
|
raise serializers.ValidationError("Sistema no existe")
|
|
|
|
class ClientPrimaryKeyRelatedField(serializers.PrimaryKeyRelatedField):
|
|
def to_internal_value(self,data):
|
|
try:
|
|
return Clientes.objects.get(RFC=data)
|
|
except Clientes.DoesNotExist:
|
|
raise serializers.ValidationError("No existe Cliente")
|
|
|
|
class DeviceSerializer(serializers.ModelSerializer):
|
|
client = ClientPrimaryKeyRelatedField(queryset=Clientes.objects.all())
|
|
sistema = SistemaPrimaryKeyRelatedField(queryset=Sistema.objects.all())
|
|
token = serializers.CharField(read_only=True)
|
|
class Meta:
|
|
model =Device
|
|
fields = ('client','sistema','device_name','device_os', 'ip_address','token')
|
|
|
|
|
|
def validate(self, data):
|
|
sistema = data.get('sistema', None)
|
|
client = data.get('client', None)
|
|
try:
|
|
sistemaxCli = sistemas_por_cliente.objects.get(id_sistema=sistema,cliente=client)
|
|
print('validate spc', sistemaxCli)
|
|
except sistemas_por_cliente.DoesNotExist:
|
|
raise serializers.ValidationError('No existe licencia para este sistmea y/o cliente')
|
|
if sistemaxCli.num_licencias <= Device.objects.filter(sistema=sistemaxCli.id_sistema).count():
|
|
raise serializers.ValidationError(f"No hay licencias disponibles para este sistema:{sistema} y cliente:{client}")
|
|
return data |