from django.contrib.auth import get_user_model
from rest_framework import serializers

from employees.models import Department
from procurement.models import EquipmentCategory

Employee = get_user_model()


class CreateEquipmentFromItemSerializer(serializers.Serializer):
    inventory_number = serializers.CharField(max_length=50)
    category = serializers.PrimaryKeyRelatedField(
        queryset=EquipmentCategory.objects.all()
    )
    department = serializers.PrimaryKeyRelatedField(queryset=Department.objects.all())
    serial_number = serializers.CharField(
        max_length=100,
        required=False,
        allow_blank=True,
        default="",
    )
    location = serializers.CharField(
        max_length=255,
        required=False,
        allow_blank=True,
        default="",
    )
    warranty_until = serializers.DateField(required=False, allow_null=True)
    responsible_person = serializers.PrimaryKeyRelatedField(
        queryset=Employee.objects.visible_in_ui().filter(is_active=True),
        required=False,
        allow_null=True,
    )