---

from django.db import models


class Category(models.Model):
    name = models.CharField(max_length=50, unique=True)

    def __str__(self):
        return self.name


class Employee(models.Model):
    first_name = models.CharField(max_length=50)
    last_name = models.CharField(max_length=50)
    department = models.CharField(max_length=100)

    def __str__(self):
        return f"{self.first_name} {self.last_name} ({self.department})"


class Equipment(models.Model):
    STATUS_CHOICES = [
        ('active', 'Активно'),
        ('in_repair', 'На ремонте'),
        ('decommissioned', 'Списано'),
    ]

    name = models.CharField(max_length=100)
    category = models.ForeignKey(Category, on_delete=models.CASCADE)
    serial_number = models.CharField(max_length=50, unique=True)

    employee = models.ForeignKey(
        Employee,
        on_delete=models.SET_NULL,
        null=True,
        blank=True
    )

    status = models.CharField(
        max_length=20,
        choices=STATUS_CHOICES,
        default='active'
    )

    purchase_date = models.DateField()

    def __str__(self):
        return f"{self.name} ({self.serial_number})"