from django.db import models

from django.utils.translation import gettext_lazy as _


class Subject(models.Model):
    class Meta:
        verbose_name = _("Subject")
        verbose_name_plural = _("Subjects")

    id = models.CharField(
        _("CIN"),
        max_length=8,
        primary_key=True
    )

    name = models.CharField(
        _("Name"),
        max_length=128
    )

    vat_id = models.CharField(
        _("VAT ID"),
        max_length=12,
        null=True,
        blank=True
    )

    street = models.CharField(
        _("Street"),
        max_length=64,
    )

    zip_code = models.CharField(
        _("Zip Code"),
        max_length=6,
    )

    city = models.CharField(
        _("City"),
        max_length=64,
    )

    city_part = models.CharField(
        _("City part"),
        max_length=64,
        null=True,
        blank=True
    )

    def __str__(self):
        return self.name

    def get_linked_users(self) -> list[int]:
        return list(self.user_set.values_list("id", flat=True))