Misago/misago/users/api/userendpoints/list.py

from datetime import timedelta

from django.contrib.auth import get_user_model
from django.db.models import Count
from django.utils import timezone

from rest_framework.pagination import PageNumberPagination

from misago.conf import settings
from misago.core.cache import cache
from misago.core.shortcuts import get_object_or_404
from misago.forums.models import Forum

from misago.users.models import Rank
from misago.users.serializers import ScoredUserSerializer


def active(request, queryset):
    cache_key = 'misago_active_posters_ranking'
    ranking = cache.get(cache_key, False)
    if ranking is False:
        ranking = real_active()
        cache.set(cache_key, ranking, 18*3600)
    return ranking


def real_active():
    tracked_period = settings.MISAGO_RANKING_LENGTH
    tracked_since = timezone.now() - timedelta(days=tracked_period)

    ranked_forums = [forum.pk for forum in Forum.objects.all_forums()]

    User = get_user_model()
    queryset = User.objects.filter(posts__gt=0)
    queryset = queryset.filter(post__posted_on__gte=tracked_since,
                               post__forum__in=ranked_forums)
    queryset = queryset.annotate(score=Count('post'))
    queryset = queryset.select_related('user__rank')
    queryset = queryset.order_by('-score', 'slug')

    users_ranking = []
    for result in queryset[:settings.MISAGO_RANKING_SIZE]:
        result.score = result.num_posts
        users_ranking.append(result)
    return {
        'results': ScoredUserSerializer(users_ranking, many=True).data,
        'meta': {
            'ranking_length': settings.MISAGO_RANKING_LENGTH
        }
    }


def rank(request, queryset):
    rank_slug = request.query_params.get('rank')
    if not rank_slug:
        return

    rank = get_object_or_404(Rank.objects.filter(is_tab=True), slug=rank_slug)
    queryset = queryset.filter(rank=rank).order_by('slug')

    return {'queryset': queryset, 'paginate': True}


LISTS = {
    'active': active,
    'rank': rank,
}


def list_endpoint(request, queryset):
    list_type = request.query_params.get('list')
    list_handler = LISTS.get(list_type)

    if list_handler:
        return list_handler(request, queryset)
    else:
        return