from django.db import models
from django.db.models import Q
from django.utils import timezone
 
class ActiveManager(models.Manager):
    def get_queryset(self):
        return super().get_queryset().filter(is_active=True)
 
class PublishedManager(ActiveManager):
    def get_queryset(self):
        return super().get_queryset().filter(
            Q(published_at__isnull=True) | Q(published_at__lte=timezone.now())
        )
 
class Article(models.Model):
    title = models.CharField(max_length=200)
    slug = models.SlugField(unique=True)
    content = models.TextField()
    is_active = models.BooleanField(default=True)
    published_at = models.DateTimeField(null=True, blank=True)
    author = models.ForeignKey('auth.User', on_delete=models.CASCADE)
    created_at = models.DateTimeField(auto_now_add=True)
 
    objects = models.Manager()
    active = ActiveManager()
    published = PublishedManager()
 
    class Meta:
        indexes = [
            models.Index(fields=['slug']),
            models.Index(fields=['-published_at', 'is_active']),
        ]
        constraints = [
            models.UniqueConstraint(
                fields=['author', 'slug'],
                name='unique_author_slug'
            )
        ]
        ordering = ['-created_at']
        verbose_name = 'Article'
        verbose_name_plural = 'Articles'
 
    def __str__(self):
        return self.title
 
    @classmethod
    def get_visible_for_user(cls, user):
        if user.is_staff:
            return cls.objects.all()
        return cls.published.all()