from django.db.models import Q, Count, Avg
from datetime import datetime, timedelta
 
class ArticleManager(models.Manager):
    def search_articles(self, query, filters=None):
        base_query = self.filter(
            Q(title__icontains=query) | 
            Q(content__icontains=query) |
            Q(tags__name__icontains=query)
        )
        
        if filters:
            query_filter = Q()
            
            if filters.get('published'):
                query_filter &= Q(is_published=True)
            
            if filters.get('date_range'):
                start, end = filters['date_range']
                query_filter &= Q(published_at__range=(start, end))
            
            if filters.get('authors'):
                query_filter &= Q(author__in=filters['authors'])
            
            if filters.get('excluded_categories'):
                query_filter &= ~Q(category__in=filters['excluded_categories'])
            
            base_query = base_query.filter(query_filter)
        
        return base_query.annotate(
            comment_count=Count('comments'),
            avg_rating=Avg('ratings__value')
        ).select_related('author', 'category').order_by('-published_at')
 
    def get_trending(self, days=7):
        cutoff = datetime.now() - timedelta(days=days)
        return self.filter(
            published_at__gte=cutoff
        ).annotate(
            engagement=Count('views') + Count('comments') * 3
        ).order_by('-engagement')[:10]