from django.db import models
from django.contrib.postgres.fields import ArrayField
from django.utils.translation import gettext_lazy as _

class StructureType(models.IntegerChoices):
    MATERNELLE = 1, _('Maternelle')
    PRIMAIRE = 2, _('Primaire')
    SECONDAIRE = 3, _('Secondaire')

class Establishment(models.Model):
    name = models.CharField(max_length=255, unique=True)
    address = models.CharField(max_length=255)
    total_capacity = models.IntegerField()
    establishment_type = ArrayField(models.IntegerField(choices=StructureType.choices))
    licence_code = models.CharField(max_length=100, blank=True)
    is_active = models.BooleanField(default=True)
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.name