Misago/misago/users/search.py

from django.contrib.auth import get_user_model
from django.core.exceptions import PermissionDenied
from django.utils.translation import ugettext as _
from django.utils.translation import ugettext_lazy

from misago.search import SearchProvider

from .serializers import UserCardSerializer


HEAD_RESULTS = 8
TAIL_RESULTS = 8

UserModel = get_user_model()


class SearchUsers(SearchProvider):
    name = ugettext_lazy("Users")
    url = 'users'

    def allow_search(self):
        if not self.request.user.acl_cache['can_search_users']:
            raise PermissionDenied(
                _("You don't have permission to search users."))

    def search(self, query, page=1):
        if query:
            results = search_users(
                search_disabled=self.request.user.is_staff,
                username=query
            )
        else:
            results = []

        return {
            'results': UserCardSerializer(results, many=True).data,
            'count': len(results)
        }


def search_users(**filters):
    queryset = UserModel.objects.order_by('slug').select_related(
        'rank', 'ban_cache', 'online_tracker')

    if not filters.get('search_disabled', False):
        queryset = queryset.filter(is_active=True)

    username = filters.get('username').lower()

    results = []

    # lets grab head and tail results:
    results += list(queryset.filter(slug__startswith=username)[:HEAD_RESULTS])
    results += list(queryset.filter(
        slug__contains=username
    ).exclude(pk__in=[r.pk for r in results])[:TAIL_RESULTS])

    return results