from django.db import models
from django.utils.text import slugify
from status.models import Status # Import the Status model
from organization.models import BaseTenantModel

class Department(BaseTenantModel):
    department_id = models.AutoField(primary_key=True)
    department_name = models.CharField(max_length=255)
    slug = models.SlugField(unique=True, blank=True, max_length=255)
    department_code = models.CharField(max_length=10, unique=True)
    head_teacher = models.ForeignKey(
        'teacher_management.Teacher', # Assuming a 'teachers' app with a 'Teacher' model
        on_delete=models.SET_NULL,
        null=True,
        blank=True,
        related_name='headed_departments'
    )
    created_at = models.DateTimeField(auto_now_add=True)
    status = models.ForeignKey(Status, on_delete=models.SET_NULL, null=True, blank=True, limit_choices_to={'type': 'department'})

    def save(self, *args, **kwargs):
        if not self.slug:
            self.slug = slugify(self.department_name)
        super(Department, self).save(*args, **kwargs)

    def __str__(self):
        return self.department_name