from django.db.models import Count
from django.utils import timezone
from datetime import timedelta

from ..models import SubscriptionRequest

def get_latest_requests(limit=50):
    requests = SubscriptionRequest.objects.order_by("-created_at")[:limit]
    requests_dict = []
    for request in requests:
        requests_dict.append({
            "request_id": request.request_id,
            "pages": request.pages,
            "doc_type": request.doc_type,
            # "predict_result": request.predict_result,
            "created_at": request.created_at,
        })
    return requests_dict

def count_requests_by_date(days_limit=5):
    today = timezone.now().date()
    start_date = today - timedelta(days=days_limit)
    requests_by_date = SubscriptionRequest.objects.filter(created_at__gte=start_date).values('created_at__date').annotate(count=Count('id')).values('created_at__date', 'count').order_by('created_at__date')
    count_dict = []
    for rbd in requests_by_date:
        count_dict.append({
            "date": rbd["created_at__date"],
            "count": rbd["count"],
        })
    return count_dict


def get_health_report():
    return {
        "latest_requests": get_latest_requests(),
        "count_by_date": count_requests_by_date(),
    }